Fix: Remove asyncio event loop policy, fix MistralClient references, simplify chat message format

ai_investigation.py

@@ -634,13 +634,12 @@ def ask_investigation_question(question, conversation_history=None):
 # =====================================================
 
 def get_mistral_client():
-    """Get Mistral client"""
+    """Get Mistral client - uses HuggingFace InferenceClient"""
     try:
-        api_key = "any"
-        client = MistralClient(api_key=api_key)
-        return client
+        from database import client as hf_client
+        return hf_client
     except Exception as e:
-        print(f"⚠️ Mistral error: {e}")
+        print(f"Warning: LLM error: {e}")
         return None
 
 

app.py

@@ -44,22 +44,8 @@ from ai_investigation import (
 # ASYNC FIX & EVENT LOOP MANAGEMENT
 # =========================================================
 
-# Set appropriate event loop policy based on platform
-if platform.system() == "Windows":
-    try:
-        asyncio.set_event_loop_policy(asyncio.WindowsProactorEventLoopPolicy())
-    except Exception as e:
-        print(f"⚠️ Could not set ProactorEventLoop: {e}")
-else:
-    try:
-        asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
-    except Exception:
-        pass
-
-# Suppress asyncio and other warnings
+# Suppress all warnings to prevent asyncio event loop cleanup messages
 import warnings
-warnings.filterwarnings('ignore', category=ResourceWarning)
-warnings.filterwarnings('ignore', category=RuntimeWarning)
 warnings.filterwarnings('ignore')
 
 # =========================================================
@@ -662,6 +648,8 @@ with gr.Blocks(
             """Process investigation with chat-like conversational flow"""
             
             if not message or len(message.strip()) < 2:
+                if not chat_history:
+                    chat_history = []
                 return chat_history, conv_state, inv_results
             
             # Ensure chat_history is a list
@@ -674,33 +662,35 @@ with gr.Blocks(
             # Format response for chat
             if result.get("status") == "error":
                 error_msg = result.get("message", "Investigation failed")
-                ai_response = f"❌ **Investigation Failed**\n\n{error_msg}"
+                ai_response = f"Error: {error_msg}"
             else:
                 # Build comprehensive response for chat
                 answer = result.get("answer", "")
                 analysis = result.get("analysis", {})
                 
-                ai_response = f"""**🔍 Investigation Results**
-
-**Summary:**
-{result.get('total_records', 0)} records retrieved | {analysis.get('unique_vehicles', 0)} vehicles | {analysis.get('unique_locations', 0)} locations
+                ai_response = f"""Investigation Results
 
-**Analysis:**
-{answer}
+Summary: {result.get('total_records', 0)} records retrieved | {analysis.get('unique_vehicles', 0)} vehicles | {analysis.get('unique_locations', 0)} locations
 
+Analysis: {answer}
 """
                 
                 # Add key findings
                 if analysis.get("key_findings"):
-                    ai_response += "**🚨 Key Findings:**\n"
+                    ai_response += "\nKey Findings:\n"
                     for finding in analysis.get("key_findings", [])[:3]:
-                        ai_response += f"• {finding}\n"
+                        ai_response += f"- {finding}\n"
                 
                 # Store full results for detailed tabs
                 inv_results = result
             
             # Add complete [user, assistant] pair to chat history
-            chat_history = chat_history + [[message, ai_response]]
+            try:
+                msg_str = str(message).strip()
+                resp_str = str(ai_response).strip()
+                chat_history = list(chat_history) + [[msg_str, resp_str]]
+            except Exception as e:
+                print(f"Chat formatting error: {e}")
             
             return chat_history, conv_state + [message], inv_results