Update app.py

app.py

@@ -73,15 +73,15 @@ def generate_image(prompt, output_filename="generated_image"):
 # Function to handle chat interaction
 def chat_handler(user_input, chat_history):
     # Add user message to chat history
-    chat_history.append([user_input, None])
+    chat_history.append({"role": "user", "content": user_input})
     
     # Generate image based on user input
     img, status = generate_image(user_input)
     
     # Add AI response to chat history
     if img:
-        chat_history.append([None, img])
-    chat_history.append([None, status])
+        chat_history.append({"role": "assistant", "content": img})
+    chat_history.append({"role": "assistant", "content": status})
     
     return chat_history, ""
 
@@ -94,7 +94,7 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
     chatbot = gr.Chatbot(
         label="Chat",
         height=400,
-        bubble_full_width=False,
+        type="messages",  # Explicitly set to 'messages' format
         avatar_images=(None, None)  # No avatars for simplicity
     )
     
@@ -116,8 +116,7 @@ with gr.Blocks(title="Image Editing Chatbot") as demo:
     run_btn.click(
         fn=chat_handler,
         inputs=[prompt_input, chat_state],
-        outputs=[chatbot, prompt_input],
-        _js="() => {return [document.querySelector('input').value, null]}"
+        outputs=[chatbot, prompt_input]
     )
 
     # Also allow Enter key to submit