Update app.py

app.py

@@ -8,65 +8,53 @@ except Exception as e:
     chimera = None
     print(f"Startup Error: {e}")
 
-def chat_logic(message, history, mode_selection):
-    if history is None:
-        history = []
-        
+def generate_response(message, history, mode_selection):
+    """
+    ChatInterface automatically handles history. 
+    We just need to return the string response.
+    """
     if not chimera:
-        # Universal Format: [User Message, Bot Message]
-        history.append([message, "❌ Error: API Keys missing. Check Settings -> Secrets."])
-        return history, ""
+        return "❌ Error: API Keys missing. Check Settings -> Secrets."
     
+    # Map friendly names
     role_map = {
         "Auto (Router)": "Auto",
         "⚡ ASM (Qwen + Llama)": "ASM",
         "🔬 SFE (Data/Science)": "SFE",
         "🎨 CSM (Story/Creative)": "CSM"
     }
+    # If mode_selection is None (default), use Auto
     selected_role = role_map.get(mode_selection, "Auto")
 
-    # Get response
+    # Get response (Chimera handles the logic)
     response_text, active_module = chimera.process_request(message, history, selected_role)
     
-    # Format output
-    final_response = f"**[{active_module} Active]**\n\n{response_text}"
-    
-    # Universal List Format (Crash-Proof)
-    history.append([message, final_response])
-    
-    return history, ""
+    # Return formatted string
+    return f"**[{active_module} Active]**\n\n{response_text}"
 
 # --- Sci-Fi Theme CSS ---
 custom_css = """
 body {background-color: #0b0f19; color: #c9d1d9;}
 .gradio-container {font-family: 'IBM Plex Mono', monospace;}
-header {display: none !important;}
-#chatbot {
-    height: 600px; 
-    border: 1px solid #30363d; 
-    background-color: #0d1117;
-    border-radius: 12px;
-}
 """
 
-with gr.Blocks(title="Axon Trinity") as demo:
+# We use ChatInterface for maximum stability
+with gr.Blocks(css=custom_css, title="Axon Trinity") as demo:
     gr.Markdown("# ⚔️ AXON: QWEN TRINITY")
-    gr.Markdown("*> Pipeline: Gemini (Plan) ➔ Qwen 3 (Code) ➔ Llama 3.3 (Refine)*")
     
-    with gr.Row():
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(elem_id="chatbot")
-        
-        with gr.Column(scale=1):
-            mode = gr.Dropdown(
-                choices=["Auto (Router)", "⚡ ASM (Qwen + Llama)", "🔬 SFE (Data/Science)", "🎨 CSM (Story/Creative)"],
-                value="Auto (Router)",
-                label="Persona Mode"
-            )
+    # The Mode Selector
+    mode_picker = gr.Dropdown(
+        choices=["Auto (Router)", "⚡ ASM (Qwen + Llama)", "🔬 SFE (Data/Science)", "🎨 CSM (Story/Creative)"],
+        value="Auto (Router)",
+        label="Persona Mode"
+    )
 
-    msg = gr.Textbox(placeholder="Enter coding task...", autofocus=True)
-    msg.submit(chat_logic, [msg, chatbot, mode], [chatbot, msg])
+    # The Chat Interface (Handles all UI logic automatically)
+    chat = gr.ChatInterface(
+        fn=generate_response,
+        additional_inputs=[mode_picker],
+        type="messages" # Trying the modern format one last time, safely
+    )
 
 if __name__ == "__main__":
-    # FIX: Disable SSR to prevent the asyncio crash
-    demo.launch(css=custom_css, ssr_mode=False)
+    demo.launch(ssr_mode=False)