Update app.py

app.py

@@ -1,3 +1,4 @@
+# Create updated app.py file with fixes
 import gradio as gr
 import os
 import json
@@ -11,6 +12,7 @@ api_key = None
 client = None
 chat_history = []
 current_response_id = None
+current_image = None
 
 def validate_api_key(key):
     """Validate OpenAI API key by making a simple request"""
@@ -41,19 +43,22 @@ def encode_image(image_path):
     with open(image_path, 'rb') as image_file:
         return base64.b64encode(image_file.read()).decode("utf-8")
 
-def chat(message, history, image=None):
+def chat(message, chat_history):
     """Process chat messages with optional image"""
-    global client, current_response_id, chat_history
+    global client, current_response_id, current_image
     
     if not client:
-        return history + [("You", "Please set your OpenAI API key first.")]
+        return chat_history + [{"role": "assistant", "content": "Please set your OpenAI API key first."}]
     
     try:
-        if image is not None:
+        # Add user message to history
+        chat_history.append({"role": "user", "content": message})
+        
+        if current_image is not None:
             # Save the uploaded image to a temporary file
             temp_file = NamedTemporaryFile(delete=False, suffix=".png")
             image_path = temp_file.name
-            image.save(image_path)
+            current_image.save(image_path)
             
             # Encode the image
             base64_image = encode_image(image_path)
@@ -83,6 +88,8 @@ def chat(message, history, image=None):
             
             # Clean up the temporary file
             os.unlink(image_path)
+            # Reset the image
+            current_image = None
         else:
             # Text-only message
             response = client.responses.create(
@@ -93,14 +100,14 @@ def chat(message, history, image=None):
         
         current_response_id = response.id
         
-        # Update chat history for export
-        chat_history.append({"role": "user", "content": message})
+        # Add assistant response to history
         chat_history.append({"role": "assistant", "content": response.output_text})
         
-        # Return updated history
-        return history + [("You", message), ("Bot", response.output_text)]
+        return chat_history
     except Exception as e:
-        return history + [("You", message), ("Bot", f"Error: {str(e)}")]
+        error_message = f"Error: {str(e)}"
+        chat_history.append({"role": "assistant", "content": error_message})
+        return chat_history
 
 def export_chat():
     """Export chat history to JSON file"""
@@ -119,9 +126,10 @@ def export_chat():
 
 def clear_chat():
     """Clear chat history and reset response ID"""
-    global chat_history, current_response_id
+    global chat_history, current_response_id, current_image
     chat_history = []
     current_response_id = None
+    current_image = None
     return []
 
 def generate_iframe_code(request: gr.Request):
@@ -140,9 +148,11 @@ def generate_iframe_code(request: gr.Request):
     
     return iframe_code
 
-def process_image_upload(img, text):
-    """Process image upload and update message text"""
-    return gr.update(value=f"{text if text else ''}[Image attached]")
+def save_image(img):
+    """Save uploaded image to global variable"""
+    global current_image
+    current_image = img
+    return "[Image attached]"
 
 def create_ui():
     """Create the Gradio UI"""
@@ -190,7 +200,7 @@ def create_ui():
             with gr.Group(visible=False) as chat_interface:
                 with gr.Tabs() as tabs:
                     with gr.TabItem("Chat", elem_classes="tab-content"):
-                        chatbot = gr.Chatbot(height=400, type="tuples")
+                        chatbot = gr.Chatbot(height=400, type="messages")
                         
                         with gr.Row():
                             with gr.Column(scale=4):
@@ -224,21 +234,16 @@ def create_ui():
         )
         
         # Handle image uploads
-        image_upload = None
         image_btn.upload(
-            process_image_upload,
-            inputs=[image_btn, msg],
-            outputs=[msg]
-        ).then(
-            lambda x: x,
+            save_image,
             inputs=[image_btn],
-            outputs=[image_upload]
+            outputs=[msg]
         )
         
         # Handle chat submission
         msg.submit(
             chat,
-            inputs=[msg, chatbot, image_btn],
+            inputs=[msg, chatbot],
             outputs=[chatbot]
         ).then(
             lambda: gr.update(value=""),
@@ -273,4 +278,4 @@ def create_ui():
 
 if __name__ == "__main__":
     demo = create_ui()
-    demo.launch()
+    demo.launch()