raannakasturi commited on
Commit
398744d
·
verified ·
1 Parent(s): 3294248

Update summarize_paper.py

Browse files
Files changed (1) hide show
  1. summarize_paper.py +23 -22
summarize_paper.py CHANGED
@@ -1,23 +1,24 @@
1
- import json
2
- from gradio_client import Client
3
-
4
- def summarize_paper(pdf_url, paper_id, access_key):
5
- mindmap = None
6
- summary = None
7
- try:
8
- summarizer_client = Client("raannakasturi/ReXploreAPI")
9
- result = summarizer_client.predict(
10
- url=pdf_url,
11
- id=paper_id,
12
- access_key=access_key,
13
- api_name="/rexplore_summarizer"
14
- )
15
- if result:
16
- data = json.loads(result[0])
17
- if data["mindmap_status"] == "success":
18
- mindmap = data["mindmap"]
19
- if data["summary_status"] == "success":
20
- summary = data["summary"]
21
- except Exception as e:
22
- print(f"Error summarizing paper: {e}")
 
23
  return summary, mindmap
 
1
+ import json
2
+ from gradio_client import Client
3
+
4
+ def summarize_paper(pdf_url, paper_id, access_key):
5
+ mindmap = None
6
+ summary = None
7
+ try:
8
+ summarizer_client = Client("raannakasturi/ReXploreAPI")
9
+ result = summarizer_client.predict(
10
+ url=pdf_url,
11
+ id=paper_id,
12
+ access_key=access_key,
13
+ api_name="/rexplore_summarizer"
14
+ )
15
+ print(result)
16
+ if result:
17
+ data = json.loads(result[0])
18
+ if data["mindmap_status"] == "success":
19
+ mindmap = data["mindmap"]
20
+ if data["summary_status"] == "success":
21
+ summary = data["summary"]
22
+ except Exception as e:
23
+ print(f"Error summarizing paper: {e}")
24
  return summary, mindmap