Update app.py

app.py

@@ -12,7 +12,7 @@ MODELS = [
     ("Llama 3.2 Vision", "meta-llama/llama-3.2-11b-vision-instruct:free")
 ]
 
-def ask_ai(message, history, model_choice):
+def ask_ai(message, chatbot, model_choice):
     """Basic AI query function"""
     # Get model ID
     model_id = MODELS[0][1]  # Default
@@ -21,9 +21,9 @@ def ask_ai(message, history, model_choice):
             model_id = model_id_value
             break
     
-    # Create messages from history
+    # Create messages from chatbot history
     messages = []
-    for human_msg, ai_msg in history:
+    for human_msg, ai_msg in chatbot:
         messages.append({"role": "user", "content": human_msg})
         if ai_msg:
             messages.append({"role": "assistant", "content": ai_msg})
@@ -52,30 +52,61 @@ def ask_ai(message, history, model_choice):
         if response.status_code == 200:
             result = response.json()
             ai_response = result.get("choices", [{}])[0].get("message", {}).get("content", "")
-            history.append((message, ai_response))
+            chatbot.append((message, ai_response))
         else:
-            history.append((message, f"Error: Status code {response.status_code}"))
+            chatbot.append((message, f"Error: Status code {response.status_code}"))
     except Exception as e:
-        history.append((message, f"Error: {str(e)}"))
+        chatbot.append((message, f"Error: {str(e)}"))
     
-    return history, ""
+    return chatbot, ""
+
+def clear_chat():
+    return [], ""
 
 # Create minimal interface
-demo = gr.Interface(
-    fn=ask_ai,
-    inputs=[
-        gr.Textbox(placeholder="Type your message here...", label="Message"),
-        gr.State([]),
-        gr.Radio([name for name, _ in MODELS], value=MODELS[0][0], label="Model")
-    ],
-    outputs=[
-        gr.Chatbot(label="Conversation", height=500, type="messages"),
-        gr.Textbox(visible=False)
-    ],
-    title="Simple AI Chat",
-    description="A minimalist chat interface for OpenRouter AI models",
-    allow_flagging="never"
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# Simple AI Chat")
+    
+    with gr.Row():
+        with gr.Column():
+            chatbot = gr.Chatbot(height=400, type="messages")
+            
+            with gr.Row():
+                message = gr.Textbox(
+                    placeholder="Type your message here...",
+                    label="Message",
+                    lines=3
+                )
+            
+            with gr.Row():
+                model_choice = gr.Radio(
+                    [name for name, _ in MODELS],
+                    value=MODELS[0][0],
+                    label="Model"
+                )
+                
+            with gr.Row():
+                submit_btn = gr.Button("Send")
+                clear_btn = gr.Button("Clear Chat")
+    
+    # Set up events
+    submit_btn.click(
+        fn=ask_ai,
+        inputs=[message, chatbot, model_choice],
+        outputs=[chatbot, message]
+    )
+    
+    message.submit(
+        fn=ask_ai,
+        inputs=[message, chatbot, model_choice],
+        outputs=[chatbot, message]
+    )
+    
+    clear_btn.click(
+        fn=clear_chat,
+        inputs=[],
+        outputs=[chatbot, message]
+    )
 
 # Launch directly with Gradio's built-in server
 if __name__ == "__main__":