Spaces:
Running
Running
Update summarize_paper.py
Browse files- 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 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
|
|
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
|