Update main.py

main.py

@@ -5,32 +5,39 @@ from gradio_client import Client
 import mistune
 import asyncio
 
-app = fastapi.FastAPI(
-    docs_url='/',
-)
+app = fastapi.FastAPI(docs_url='/')
 
 @app.get("/")
 def read_root():
     return {"data": "Hello World"}
 
-# http://localhost:8080/generate?pdfurl=https://nayankasturi.eu.org/assets/docs/latest_resume.pdf&doi=10.1145/3394486.3412865
 @app.get("/generate")
 async def generate(pdfurl: str, doi: str):
+    # Adjust the client initialization if needed
     client = Client("raannakasturi/ScientryAPI")
-    result = await asyncio.to_thread(
-        client.predict,
-        url=pdfurl,
-        id=doi,
-        access_key="scientrypass",
-        api_name="/rexplore_summarizer"
-    )
-    data = json.loads(result[0])
-    summary = html.unescape(mistune.html(data['summary']))
-    mindmap = data['mindmap']
-    returndata = {
+    
+    try:
+        result = await asyncio.to_thread(
+            client.predict,
+            url=pdfurl,
+            id=doi,
+            access_key="scientrypass",
+            api_name="/rexplore_summarizer"
+        )
+    except Exception as e:
+        return {"error": f"Failed to call predict API: {str(e)}"}
+    
+    try:
+        data = json.loads(result[0])
+    except Exception as e:
+        return {"error": f"Failed to parse API response: {str(e)}"}
+    
+    summary = html.unescape(mistune.html(data.get('summary', '')))
+    mindmap = data.get('mindmap', {})
+    
+    return {
         "data": {
             "summary": summary,
             "mindmap": mindmap
         }
     }
-    return json.dumps(returndata)