Update app.py

app.py

@@ -18,14 +18,14 @@ def chat_with_image(message, image, history):
     messages = [{"role": "system", "content": "You are a helpful assistant that can analyze images and text."}]
     
     for human, assistant in history:
-        if isinstance(human, list) and len(human) == 2 and human[1] is not None:
+        if image is not None and human.startswith("Image uploaded: "):
             # This is an image message
-            encoded_image = encode_image(Image.open(human[1]))
+            encoded_image = encode_image(Image.open(image))
             messages.append({
                 "role": "user",
                 "content": [
                     {"type": "image_url", "image_url": {"url": f"data:image/png;base64,{encoded_image}"}},
-                    {"type": "text", "text": human[0]}
+                    {"type": "text", "text": message}
                 ]
             })
         else:
@@ -77,13 +77,18 @@ with gr.Blocks() as demo:
 
     def user(user_message, image, history):
         if image is not None:
-            return "", None, history + [[user_message, image], None]
+            return "", None, history + [["Image uploaded: " + user_message, None]]
         else:
-            return "", None, history + [[user_message, None], None]
+            return "", None, history + [[user_message, None]]
 
     def bot(history):
-        user_message, user_image = history[-1][0]
-        bot_message = chat_with_image(user_message, user_image, history[:-1])
+        user_message = history[-1][0]
+        image = None
+        if user_message.startswith("Image uploaded: "):
+            image = history[-2][0].split(": ", 1)[1]  # Get the image path from the previous message
+            user_message = user_message.split(": ", 1)[1]  # Get the actual message
+        
+        bot_message = chat_with_image(user_message, image, history[:-1])
         history[-1][1] = ""
         for character in bot_message:
             history[-1][1] += character