Update app.py

app.py

@@ -23,8 +23,7 @@ def chat_logic(message, image_file, history, mode):
         history.append({"role": "assistant", "content": "❌ Error: API Keys missing."})
         return history, "", None
     
-    # 2. Convert history to simple format for chimera (it doesn't use the full history currently)
-    # But we'll keep it available for future use
+    # 2. Convert history to simple format for chimera
     simple_history = []
     for msg in history:
         if isinstance(msg, dict):
@@ -53,7 +52,7 @@ def chat_logic(message, image_file, history, mode):
     else:
         user_msg = message
     
-    # 6. Append to History (Dictionary Format for modern Gradio)
+    # 6. Append to History (Dictionary Format for Gradio 6.x)
     history.append({"role": "user", "content": user_msg})
     history.append({"role": "assistant", "content": final_response})
     
@@ -70,12 +69,12 @@ body {
 }
 """
 
-with gr.Blocks(title="Axon God Mode", css=custom_css) as demo:
+with gr.Blocks(title="Axon God Mode") as demo:
     gr.Markdown("# ⚡ AXON: GOD MODE")
     gr.Markdown("*> Modules: VIM (Vision) | NET (Web) | IGM (Art) | ASM (Code)*")
     
     with gr.Row():
-        chatbot = gr.Chatbot(height=500, elem_id="chatbot", type="messages")
+        chatbot = gr.Chatbot(height=500, elem_id="chatbot")
     
     with gr.Row():
         with gr.Column(scale=4):
@@ -103,4 +102,4 @@ with gr.Blocks(title="Axon God Mode", css=custom_css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(ssr_mode=False)
+    demo.launch(ssr_mode=False, css=custom_css)