bug fix

multi_llm_chatbot_backend/app/api/routes/chat_sessions.py

@@ -91,6 +91,34 @@ async def get_user_chat_sessions(
             detail="Could not fetch chat sessions"
         )
 
+
+
+@router.get("/chat-sessions/count")
+async def get_chat_sessions_count(
+    current_user: User = Depends(get_current_active_user)
+):
+    """Get count of user's chat sessions"""
+    try:
+        db = get_database()
+        user_object_id = ObjectId(str(current_user.id))
+        
+        count = await db.chat_sessions.count_documents({
+            "user_id": user_object_id,
+            "is_active": {"$ne": False},
+            "deleted_at": {"$exists": False}
+        })
+        
+        return {"count": count}
+        
+    except Exception as e:
+        logger.error(f"Error counting chat sessions for user {current_user.id}: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="Failed to count chat sessions"
+        )
+
+
+
 @router.get("/chat-sessions/{session_id}")
 async def get_chat_session(
     session_id: str,
@@ -223,29 +251,7 @@ async def save_message_to_session(
             detail="Could not save message"
         )
     
-@router.get("/chat-sessions/count")
-async def get_chat_sessions_count(
-    current_user: User = Depends(get_current_active_user)
-):
-    """Get count of user's chat sessions"""
-    try:
-        db = get_database()
-        user_object_id = ObjectId(str(current_user.id))
-        
-        count = await db.chat_sessions.count_documents({
-            "user_id": user_object_id,
-            "is_active": {"$ne": False},
-            "deleted_at": {"$exists": False}
-        })
-        
-        return {"count": count}
-        
-    except Exception as e:
-        logger.error(f"Error counting chat sessions for user {current_user.id}: {e}")
-        raise HTTPException(
-            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
-            detail="Failed to count chat sessions"
-        )
+
 
 @router.delete("/chat-sessions/{session_id}")
 async def delete_chat_session(

multi_llm_chatbot_backend/app/core/canvas_database.py

@@ -23,11 +23,11 @@ async def setup_canvas_collections(db: AsyncIOMotorDatabase):
             if isinstance(index, list):
                 # Compound index
                 await collection.create_index(index)
-                logger.info(f"Created compound index: {index}")
+                logger.info(f"Created compound index: {str(index)}")
             else:
                 # Simple index
                 await collection.create_index(index)
-                logger.info(f"Created index: {index}")
+                logger.info(f"Created index: {str(index)}")
         
         # Ensure TTL index for old canvases (optional cleanup after 2 years)
         await collection.create_index(
@@ -72,5 +72,4 @@ async def cleanup_old_canvas_data(db: AsyncIOMotorDatabase):
             logger.info("No orphaned canvas records found")
             
     except Exception as e:
-        logger.error(f"Error during canvas cleanup: {e}")
-
+        logger.error(f"Error during canvas cleanup: {e}")

multi_llm_chatbot_backend/app/core/database.py

@@ -36,7 +36,7 @@ async def connect_to_mongo():
         await create_indexes()
         try:
             from app.core.canvas_database import setup_canvas_collections
-            await setup_canvas_collections(db)  # Pass db directly
+            await setup_canvas_collections(db.database)  # Pass db directly
             logger.info("Canvas database initialization completed")
         except Exception as canvas_error:
             logger.error(f"Canvas database initialization failed: {canvas_error}")

phd-advisor-frontend/src/pages/CanvasPage.js

@@ -95,18 +95,26 @@ const CanvasPage = ({ user, authToken, onNavigateToChat, onSignOut }) => {
   const [isProcessingFirstTime, setIsProcessingFirstTime] = useState(false);
 
   useEffect(() => {
+    let pollInterval = null;
+    
     const initializeCanvas = async () => {
       await fetchCanvas();
       await fetchStats();
       await triggerAutoUpdate();
       
-      // After initial load, check if we need to populate empty canvas
       setTimeout(() => {
         checkForEmptyCanvasWithChats();
-      }, 2000); // Wait 2 seconds after initial load
+      }, 2000);
     };
     
     initializeCanvas();
+    
+    // Cleanup on unmount
+    return () => {
+      if (pollInterval) {
+        clearInterval(pollInterval);
+      }
+    };
   }, []);
 
   const fetchCanvas = async () => {
@@ -161,11 +169,9 @@ const CanvasPage = ({ user, authToken, onNavigateToChat, onSignOut }) => {
   // Check if user has chats but empty canvas
   const checkForEmptyCanvasWithChats = async () => {
     try {
-      // Check if canvas is empty (no insights)
       const isEmpty = !canvasData || canvasData.total_insights === 0;
       
       if (isEmpty) {
-        // Check if user has any chat sessions
         const response = await fetch(`${process.env.REACT_APP_API_URL}/api/chat-sessions/count`, {
           headers: {
             'Authorization': `Bearer ${authToken}`,
@@ -175,16 +181,20 @@ const CanvasPage = ({ user, authToken, onNavigateToChat, onSignOut }) => {
         
         if (response.ok) {
           const { count } = await response.json();
-          
           if (count > 0) {
-            // User has chats but empty canvas - trigger full refresh
             console.log(`User has ${count} chats but empty canvas. Triggering full refresh.`);
             await handleFullRefresh();
           }
+        } else {
+          console.error('Failed to fetch chat sessions count:', response.status);
+          // Don't trigger refresh if count fails
+          return;
         }
       }
     } catch (error) {
       console.error('Error checking for empty canvas with chats:', error);
+      // Stop the polling if there's an error
+      return;
     }
   };