Update app.py

app.py

@@ -12,8 +12,23 @@ MODELS = [
     ("Llama 3.2 Vision", "meta-llama/llama-3.2-11b-vision-instruct:free")
 ]
 
+def format_to_message_dict(history):
+    """Convert history to proper message format"""
+    messages = []
+    for pair in history:
+        if len(pair) == 2:
+            human, ai = pair
+            if human:
+                messages.append({"role": "user", "content": human})
+            if ai:
+                messages.append({"role": "assistant", "content": ai})
+    return messages
+
 def ask_ai(message, chatbot, model_choice):
     """Basic AI query function"""
+    if not message.strip():
+        return chatbot, ""
+    
     # Get model ID
     model_id = MODELS[0][1]  # Default
     for name, model_id_value in MODELS:
@@ -22,11 +37,7 @@ def ask_ai(message, chatbot, model_choice):
             break
     
     # Create messages from chatbot history
-    messages = []
-    for human_msg, ai_msg in chatbot:
-        messages.append({"role": "user", "content": human_msg})
-        if ai_msg:
-            messages.append({"role": "assistant", "content": ai_msg})
+    messages = format_to_message_dict(chatbot)
     
     # Add current message
     messages.append({"role": "user", "content": message})
@@ -52,11 +63,11 @@ def ask_ai(message, chatbot, model_choice):
         if response.status_code == 200:
             result = response.json()
             ai_response = result.get("choices", [{}])[0].get("message", {}).get("content", "")
-            chatbot.append((message, ai_response))
+            chatbot = chatbot + [[message, ai_response]]
         else:
-            chatbot.append((message, f"Error: Status code {response.status_code}"))
+            chatbot = chatbot + [[message, f"Error: Status code {response.status_code}"]]
     except Exception as e:
-        chatbot.append((message, f"Error: {str(e)}"))
+        chatbot = chatbot + [[message, f"Error: {str(e)}"]]
     
     return chatbot, ""
 
@@ -67,27 +78,25 @@ def clear_chat():
 with gr.Blocks() as demo:
     gr.Markdown("# Simple AI Chat")
     
+    chatbot = gr.Chatbot(height=400)
+    
+    with gr.Row():
+        message = gr.Textbox(
+            placeholder="Type your message here...",
+            label="Message",
+            lines=2
+        )
+    
+    with gr.Row():
+        model_choice = gr.Radio(
+            [name for name, _ in MODELS],
+            value=MODELS[0][0],
+            label="Model"
+        )
+        
     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")
+        submit_btn = gr.Button("Send")
+        clear_btn = gr.Button("Clear Chat")
     
     # Set up events
     submit_btn.click(