Update app.py

app.py

@@ -46,6 +46,7 @@ logger = logging.getLogger(__name__)
 MONGO_CLIENT = None
 DB = None
 RAG_MODELS_INITIALIZED = False
+SESSION_LAST_ACCESS = {}  # Track last access time for each session
 APP_STATE = {
     "startup_time": None,
     "mongodb_connected": False,
@@ -185,21 +186,6 @@ def load_session_from_mongodb(session_id: str) -> Dict[str, Any]:
     session_logger.info("Session loaded from MongoDB with existing embeddings")
     return session_store
 
-def build_indices_for_session(session_id: str) -> Dict[str, Any]:
-    """Builds all indices for a session using its SessionRAG instance."""
-    session_logger = create_session_logger(session_id)
-    with STORE_LOCK:
-        store = SESSION_STORES[session_id]
-        session_rag = store["session_rag"]
-
-    session_logger.info(f"Building indices for {store['metadata']['chunk_count']} chunks...")
-    session_rag.build_all_indices(session_rag.chunks_data) # Correctly passing chunks
-    
-    with STORE_LOCK:
-        SESSION_STORES[session_id]["indexed"] = True
-    
-    session_logger.info("All session-specific indices built successfully")
-    return store["metadata"]
 
 def get_chat_history_safely(session_id: str, limit: int = 50) -> List[Dict[str, Any]]:
     """Get chat history with error handling."""
@@ -323,6 +309,33 @@ async def chat_with_document(session_id: str, request: ChatRequest):
             # Update last access time
             SESSION_LAST_ACCESS[session_id] = datetime.utcnow()
             session_rag = SESSION_STORES[session_id]["session_rag"]
+            # Process the query
+        session_logger.info(f"Processing query: {request.message[:100]}...")
+        result = await asyncio.to_thread(session_rag.query_documents, request.message, top_k=5)
+        APP_STATE["total_queries"] += 1
+        
+        answer = result.get('answer', 'Unable to generate an answer.')
+        
+        # Save chat messages asynchronously
+        asyncio.create_task(save_chat_message_safely(session_id, "user", request.message))
+        asyncio.create_task(save_chat_message_safely(session_id, "assistant", answer))
+        
+        processing_time = time.time() - start_time
+        session_logger.info(f"Query processed in {processing_time:.2f}s.")
+        
+        return ChatResponse(
+            success=True,
+            answer=answer,
+            sources=result.get('sources', []),
+            chat_history=[],
+            processing_time=processing_time,
+            session_id=session_id,
+            query_analysis=result.get('query_analysis'),
+            confidence=result.get('confidence')
+        )
+    except Exception as e:
+        session_logger.error(f"Chat processing failed: {e}", exc_info=True)
+        raise HTTPException(status_code=500, detail=f"Chat processing error: {e}")
 
 @app.get("/history/{session_id}")
 async def get_session_history(session_id: str):