bug fixes

multi_llm_chatbot_backend/app/api/routes/debug.py

@@ -1,10 +1,11 @@
 from fastapi import APIRouter, Request, Query
 from app.core.session_manager import get_session_manager
 from app.core.rag_manager import get_rag_manager
-from app.api.utils import get_or_create_session_for_request
 from app.core.bootstrap import chat_orchestrator
 import logging
 
+from app.api.old_routes import get_or_create_session_for_request
+
 logger = logging.getLogger(__name__)
 
 router = APIRouter()

multi_llm_chatbot_backend/app/api/routes/documents.py

@@ -3,12 +3,11 @@ from fastapi import Query
 from app.utils.document_extractor import extract_text_from_file
 from app.core.session_manager import get_session_manager
 from app.core.rag_manager import get_rag_manager
-from app.api.utils import get_or_create_session_for_request
+from app.api.utils import get_or_create_session_for_request_async
 from fastapi.responses import StreamingResponse
 from app.utils.chat_summary import generate_summary_from_messages, parse_summary_to_blocks, format_summary_for_text_export
 from app.utils.file_export import prepare_export_response, generate_pdf_file_from_blocks
 from app.core.session_manager import get_session_manager
-from app.api.utils import get_or_create_session_for_request
 from app.core.bootstrap import chat_orchestrator
 from app.core.auth import get_current_active_user
 from app.core.database import get_database
@@ -167,7 +166,7 @@ async def upload_document(
             logger.info(f"Uploading document to specific chat session: {session_id}")
         else:
             # For new/temporary chats, use regular session management
-            session_id = get_or_create_session_for_request(request)
+            session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
             logger.info(f"Uploading document to new session: {session_id}")
         
         session = session_manager.get_session(session_id)
@@ -235,7 +234,7 @@ async def upload_document(
 @router.post("/search-documents")
 async def search_documents(request: Request, query: str = Body(..., embed=True), persona: str = Body("", embed=True)):
     try:
-        session_id = get_or_create_session_for_request(request)
+        session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
         rag_manager = get_rag_manager()
 
         persona_contexts = {
@@ -267,7 +266,7 @@ async def search_documents(request: Request, query: str = Body(..., embed=True),
 @router.get("/document-stats")
 async def get_document_stats(request: Request):
     try:
-        session_id = get_or_create_session_for_request(request)
+        session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
         rag_manager = get_rag_manager()
         return rag_manager.get_document_stats(session_id)
     except Exception as e:
@@ -278,7 +277,7 @@ async def get_document_stats(request: Request):
 @router.get("/uploaded-files")
 async def get_uploaded_filenames(request: Request):
     try:
-        session_id = get_or_create_session_for_request(request)
+        session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
         session = session_manager.get_session(session_id)
         return {"files": session.uploaded_files}
     except Exception as e:
@@ -289,7 +288,7 @@ async def get_uploaded_filenames(request: Request):
 @router.get("/document-insights/{filename}")
 async def get_document_insights(filename: str, request: Request):
     try:
-        session_id = get_or_create_session_for_request(request)
+        session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
         rag_manager = get_rag_manager()
         stats = rag_manager.get_document_stats(session_id)
         document_info = next((doc for doc in stats.get("documents", []) if doc["filename"] == filename), None)
@@ -370,7 +369,7 @@ async def export_chat(
             messages = convert_messages_for_export(raw_messages)
         else:
             # Export current in-memory session (existing behavior)
-            session_id = get_or_create_session_for_request(request)
+            session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
             session = session_manager.get_session(session_id)
             # In-memory messages might already be in the right format, but convert to be safe
             messages = convert_messages_for_export(session.messages)
@@ -450,7 +449,7 @@ async def chat_summary(
             messages = convert_messages_for_export(raw_messages)
         else:
             # Summarize current in-memory session (existing behavior)
-            session_id = get_or_create_session_for_request(request)
+            session_id = await get_or_create_session_for_request_async(request)  # FIXED: Added await
             session = session_manager.get_session(session_id)
             # Convert in-memory messages
             messages = convert_messages_for_export(session.messages)
@@ -521,5 +520,4 @@ async def chat_summary(
         raise HTTPException(
             status_code=500,
             detail=f"Summary generation failed: {str(e)}"
-        )
-
+        )

multi_llm_chatbot_backend/app/api/utils.py

@@ -82,48 +82,6 @@ async def load_chat_session_into_context(chat_session_id: str, user_id: str) ->
         logger.error(f"Error loading chat session into context: {e}")
         return None
 
-def get_or_create_session_for_request(
-    request: Request, 
-    session_id_override: Optional[str] = None,
-    chat_session_id: Optional[str] = None,
-    user_id: Optional[str] = None
-) -> str:
-    """
-    Enhanced session management that properly handles chat switching
-    
-    Args:
-        request: FastAPI request object
-        session_id_override: Explicit session ID to use
-        chat_session_id: MongoDB chat session ID (for existing chats)
-        user_id: User ID (for loading existing chats)
-    """
-    
-    # Case 1: Loading an existing chat session
-    if chat_session_id and user_id:
-        try:
-            # We need to handle this in the calling function since this isn't async
-            # Return a special identifier that the calling function can handle
-            return f"load_chat_{chat_session_id}"
-            
-        except Exception as e:
-            logger.error(f"Error handling chat session load: {e}")
-            # Fall through to create new session
-    
-    # Case 2: Explicit session ID provided
-    if session_id_override:
-        return session_id_override
-
-    # Case 3: Check for session header
-    session_header = request.headers.get("X-Session-ID")
-    if session_header:
-        return session_header
-
-    # Case 4: Create a truly new session (don't fall back to IP)
-    # This ensures each new chat gets its own context
-    new_session_id = session_manager.create_session()
-    logger.info(f"Created new session: {new_session_id}")
-    return new_session_id
-
 async def get_or_create_session_for_request_async(
     request: Request, 
     session_id_override: Optional[str] = None,

phd-advisor-frontend/src/components/EnhancedChatInput.js

@@ -100,8 +100,8 @@ const EnhancedChatInput = ({
       {showUpload && (
         <div className="floating-upload-section">
           <FileUpload 
-            onFileUploaded={onFileUploaded}
-            isUploading={isLoading}
+            onFileUploaded={handleFileUploaded}
+            isUploading={isUploading}
             currentChatSessionId={currentChatSessionId}  
             authToken={authToken}
             onUploadStart={handleUploadStart}