Update app.py from anycoder

app.py

@@ -18,11 +18,12 @@ def process_message(message, history, image):
     # Prepare messages for the API
     messages = []
     
-    # Add chat history - convert tuple format to messages format
-    for user_msg, assistant_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    # Add chat history - Gradio 6 uses list of dicts format
+    for msg_dict in history:
+        if msg_dict["role"] == "user":
+            messages.append({"role": "user", "content": msg_dict["content"]})
+        elif msg_dict["role"] == "assistant":
+            messages.append({"role": "assistant", "content": msg_dict["content"]})
     
     # Add current message and image
     if message or image:
@@ -39,6 +40,9 @@ def process_message(message, history, image):
         
         messages.append(current_message)
     
+    # Add user message to history immediately
+    new_history = history + [{"role": "user", "content": message if message else ""}]
+    
     # Get response from the model
     response = ""
     try:
@@ -48,17 +52,20 @@ def process_message(message, history, image):
             stream=True,
         )
         
+        # Add assistant message to history
+        new_history.append({"role": "assistant", "content": ""})
+        
         for chunk in stream:
             if chunk.choices[0].delta.content:
                 response += chunk.choices[0].delta.content
-                # Update history with new messages in the correct format
-                new_history = history + [(message, response)]
-                yield new_history, new_history
+                # Update the assistant's message
+                new_history[-1]["content"] = response
+                yield new_history
     
     except Exception as e:
         error_msg = f"Error: {str(e)}"
-        error_history = history + [(message, error_msg)]
-        yield error_history, error_history
+        new_history[-1]["content"] = error_msg
+        yield new_history
 
 # Create Gradio interface
 with gr.Blocks() as demo:
@@ -72,7 +79,6 @@ with gr.Blocks() as demo:
     chatbot = gr.Chatbot(
         label="Conversation",
         height=400,
-        type="messages",  # Set to messages format
         avatar_images=(
             "https://cdn-icons-png.flaticon.com/512/147/147144.png",
             "https://cdn-icons-png.flaticon.com/512/4712/4712025.png"
@@ -90,7 +96,7 @@ with gr.Blocks() as demo:
             img = gr.Image(
                 label="Upload image",
                 type="filepath",
-                height=50
+                height=150
             )
     
     with gr.Row():
@@ -98,18 +104,29 @@ with gr.Blocks() as demo:
         clear_btn = gr.ClearButton(components=[msg, img, chatbot], value="Clear Chat")
     
     # Set up the chat interface
+    def submit_message(message, history, image):
+        # Clear the textbox and image after submission
+        for update in process_message(message, history, image):
+            yield update
+    
     msg.submit(
-        fn=process_message,
+        fn=submit_message,
         inputs=[msg, chatbot, img],
-        outputs=[chatbot, chatbot],
+        outputs=[chatbot],
         api_visibility="public"
+    ).then(
+        lambda: ("", None),
+        outputs=[msg, img]
     )
     
     submit_btn.click(
-        fn=process_message,
+        fn=submit_message,
         inputs=[msg, chatbot, img],
-        outputs=[chatbot, chatbot],
+        outputs=[chatbot],
         api_visibility="public"
+    ).then(
+        lambda: ("", None),
+        outputs=[msg, img]
     )
 
 # Launch the app