Update app.py

app.py

@@ -351,5 +351,34 @@ async def ask_question(q: str, user: User = Depends(get_current_user)):
     answer = user.ask_question(q)
     return {"question": q, "answer": answer}
 
+@app.get("/user_data/")
+async def get_user_data(user: User = Depends(get_current_user)):
+    """Get user's current session data including papers and API key status"""
+    detailed_papers = []
+    for i, paper in enumerate(user.papers):
+        detailed_papers.append({
+            "title": paper.title,
+            "type": "arxiv" if paper.arxiv_id else "pdf",
+            "has_references": bool(paper.references),
+            "references_loaded": bool(paper.references),
+            "references": [ref.title for ref in paper.references] if paper.references else []
+        })
+    
+    return {
+        "papers": user.context_papers,  # Keep for backward compatibility
+        "detailed_papers": detailed_papers,
+        "has_api_key": user.QA_LLM is not None,
+        "paper_count": len(user.papers)
+    }
+
+@app.get("/check_api_key/")
+async def check_api_key(user: User = Depends(get_current_user)):
+    """Check if user has an existing API key"""
+    return {
+        "has_api_key": user.QA_LLM is not None,
+        "message": "API key found" if user.QA_LLM is not None else "No API key found"
+    }
+
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)