Send TTS audio message for frontend playback

enhanced_websocket_handler.py

@@ -1,5 +1,5 @@
 """
-Enhanced WebSocket handler with hybrid LLM and optional voice features
+Enhanced WebSocket handler with hybrid LLM and voice features
 """
 
 from fastapi import WebSocket, WebSocketDisconnect
@@ -399,22 +399,21 @@ async def handle_voice_message(websocket: WebSocket, data: dict, session_data: d
 
 async def get_hybrid_response(user_message: str, context: str, config: dict, knowledge_base: str):
     """Get response using hybrid LLM with document search"""
-    
     # Search for relevant documents directly using LanceDB
     try:
         logger.info(f"🔍 Searching documents for: {user_message}")
-        
         # Use the search_documents function instead of LangChain tool
         from rag_service import search_documents
-        
         # Search for relevant documents
         docs = search_documents(user_message, limit=3)
-        
         if docs:
-                    if audio_response:
-                        await websocket.send_bytes(audio_response)
-                    else:
-                        logger.warning("⚠️ Could not generate audio response")
+            context = "\n\n".join([doc["content"] for doc in docs])
+            sources = [doc["source"] for doc in docs]
+            logger.info(f"📚 Found {len(docs)} documents from sources: {sources}")
+        else:
+            logger.info("📚 No documents found, using existing context")
+    except Exception as e:
+        logger.warning(f"❌ Document search failed: {e}, using existing context")
     logger.info(f"🤖 Getting LLM response...")
     response_text = await hybrid_llm_service.get_response(
         user_message, 
@@ -422,11 +421,9 @@ async def get_hybrid_response(user_message: str, context: str, config: dict, kno
         system_prompt="""You are a helpful government document assistant. Provide accurate, helpful responses based on the context provided. When appropriate, suggest additional resources or redirect users to relevant departments for more assistance."""
     )
     logger.info(f"✅ LLM response received, length: {len(response_text)}")
-    
     # Determine which provider was used
     provider = hybrid_llm_service.choose_llm_provider(user_message)
     provider_used = provider.value if provider else "unknown"
-    
     return response_text, provider_used
 
 async def send_text_response(websocket: WebSocket, response_text: str, provider_used: str, session_data: dict):