Update app.py

app.py

@@ -11,7 +11,7 @@ def encode_image(img):
     buffered = BytesIO()
     img.save(buffered, format="PNG")
     encoded_string = base64.b64encode(buffered.getvalue()).decode("utf-8")
-    return encoded_string
+    return f"data:image/png;base64,{encoded_string}"
 
 def get_api_response(api_key, user_inputs):
     """
@@ -38,11 +38,11 @@ def get_api_response(api_key, user_inputs):
             })
         
         if user_inputs.get("image"):
-            # Open the uploaded image (already a PIL Image)
+            # Encode the uploaded image to a base64 string
             base64_img = encode_image(user_inputs["image"])
             messages.append({
                 "type": "image_url",
-                "image_url": {"url": f"data:image/png;base64,{base64_img}"}
+                "image_url": {"url": base64_img}
             })
         
         payload = {
@@ -85,7 +85,9 @@ def chatbot_response(api_key, user_inputs, history):
     if user_text:
         user_content.append(user_text)
     if user_image:
-        user_content.append(user_image)
+        # Convert the PIL Image to a base64 string
+        img_data = encode_image(user_image)
+        user_content.append(img_data)
     
     # Append the user's message to the history
     if user_content: