Update app.py

app.py

@@ -343,8 +343,22 @@ async def add_arxiv(arxiv_id: str = Form(...), user: User = Depends(get_current_
 
 @app.post("/add_references/")
 async def add_references(index: int = Form(...), user: User = Depends(get_current_user)):
-    refs = user.add_reference_papers(index)
-    return {"message": "References added", "references": refs, "context_papers": user.context_papers}
+    print(f"Received request to add references for index: {index}")
+    print(f"User has {len(user.papers)} main papers")
+    print(f"Paper titles: {[paper.title for paper in user.papers]}")
+    
+    if index < 0 or index >= len(user.papers):
+        raise HTTPException(
+            status_code=400, 
+            detail=f"Invalid paper index: {index}. User has {len(user.papers)} papers (valid indices: 0-{len(user.papers)-1})"
+        )
+    
+    try:
+        refs = user.add_reference_papers(index)
+        return {"message": "References added", "references": refs or [], "context_papers": user.context_papers}
+    except Exception as e:
+        print(f"Error adding references: {str(e)}")
+        raise HTTPException(status_code=500, detail=f"Failed to add references: {str(e)}")
 
 @app.get("/ask/")
 async def ask_question(q: str, user: User = Depends(get_current_user)):