Update app.py

app.py

@@ -8,54 +8,69 @@ except Exception as e:
     chimera = None
     print(f"Startup Error: {e}")
 
-def generate_response(message, history, mode_selection):
-    """
-    Standard ChatInterface logic.
-    Accepts: message (str), history (list of lists), mode_selection (str)
-    Returns: response (str)
-    """
-    if not chimera:
-        return "❌ Error: API Keys missing. Check Settings -> Secrets."
+def chat_logic(message, image_file, history, mode):
+    if history is None: history = []
     
-    # Map friendly names
-    role_map = {
-        "Auto (Router)": "Auto",
-        "⚡ ASM (Qwen + Llama)": "ASM",
-        "🔬 SFE (Data/Science)": "SFE",
-        "🎨 CSM (Story/Creative)": "CSM"
-    }
-    # Handle case where mode is None
-    selected_role = role_map.get(mode_selection, "Auto")
+    # 1. Handle Empty Input
+    if not message and not image_file:
+        return history, "", None
+        
+    if not chimera:
+        history.append([message, "❌ Error: API Keys missing."])
+        return history, "", None
 
-    # Get response from Chimera Brain
-    response_text, active_module = chimera.process_request(message, history, selected_role)
+    # 2. Process
+    response_text, active_module = chimera.process_request(message, history, mode, image_file)
+    
+    # 3. Format Output
+    final_response = f"**[{active_module} Active]**\n\n{response_text}"
+    
+    # 4. Append to History (Universal List Format)
+    if image_file:
+        # If user uploaded an image, show a placeholder text for their side
+        history.append([f"[Uploaded Image] {message}", final_response])
+    else:
+        history.append([message, final_response])
     
-    # Return formatted string
-    return f"**[{active_module} Active]**\n\n{response_text}"
+    return history, "", None
 
-# --- Sci-Fi Theme CSS ---
+# --- UI Layout ---
 custom_css = """
 body {background-color: #0b0f19; color: #c9d1d9;}
 .gradio-container {font-family: 'IBM Plex Mono', monospace;}
 """
 
-# 1. Removed 'css' from Blocks to fix the Warning
-with gr.Blocks(title="Axon Trinity") as demo:
-    gr.Markdown("# ⚔️ AXON: QWEN TRINITY")
+with gr.Blocks(title="Axon God Mode", css=custom_css) as demo:
+    gr.Markdown("# ⚡ AXON: GOD MODE")
+    gr.Markdown("*> Modules: VIM (Vision) | NET (Web) | IGM (Art) | ASM (Code)*")
     
-    mode_picker = gr.Dropdown(
-        choices=["Auto (Router)", "⚡ ASM (Qwen + Llama)", "🔬 SFE (Data/Science)", "🎨 CSM (Story/Creative)"],
-        value="Auto (Router)",
-        label="Persona Mode"
-    )
+    with gr.Row():
+        chatbot = gr.Chatbot(height=500, elem_id="chatbot")
+    
+    with gr.Row():
+        with gr.Column(scale=4):
+            msg = gr.Textbox(placeholder="Ask anything, or upload an image...", show_label=False)
+            # The Magic Upload Button
+            btn_upload = gr.Image(type="filepath", label="Upload for Vision (VIM)", height=70)
+        
+        with gr.Column(scale=1):
+            mode = gr.Dropdown(
+                choices=["Auto", "ASM (Code)", "IGM (Generate Image)", "NET (Search)", "VIM (Vision)"],
+                value="Auto", show_label=False
+            )
+            submit = gr.Button("🚀 EXECUTE", variant="primary")
 
-    # 2. REMOVED 'type="messages"' completely.
-    # This forces it to use the default 'List of Lists' format which never crashes.
-    chat = gr.ChatInterface(
-        fn=generate_response,
-        additional_inputs=[mode_picker]
+    # Event Handlers
+    submit.click(
+        chat_logic, 
+        inputs=[msg, btn_upload, chatbot, mode], 
+        outputs=[chatbot, msg, btn_upload]
+    )
+    msg.submit(
+        chat_logic, 
+        inputs=[msg, btn_upload, chatbot, mode], 
+        outputs=[chatbot, msg, btn_upload]
     )
 
 if __name__ == "__main__":
-    # 3. Moved CSS here and disabled SSR for safety
-    demo.launch(ssr_mode=False, css=custom_css)
+    demo.launch(ssr_mode=False)