Update app.py

app.py

@@ -2,10 +2,22 @@ import streamlit as st
 import base64
 import os
 import requests
+from PIL import Image
+from io import BytesIO
+
+# Function to compress and resize the image before base64 encoding
+def compress_and_resize_image(image, max_size=(1024, 1024), quality=85):
+    img = Image.open(image)
+    img.thumbnail(max_size)  # Resize image while maintaining aspect ratio
+    with BytesIO() as byte_io:
+        img.save(byte_io, format="JPEG", quality=quality)  # Save with reduced quality
+        byte_io.seek(0)
+        return byte_io
 
 # Function to convert uploaded image to base64
 def convert_image_to_base64(image):
-    image_bytes = image.read()
+    compressed_image = compress_and_resize_image(image)
+    image_bytes = compressed_image.read()
     encoded_image = base64.b64encode(image_bytes).decode("utf-8")
     return encoded_image
 
@@ -29,12 +41,12 @@ def generate_caption(encoded_image):
             {
                 "role": "user",
                 "content": "Write a caption for this image"
+            },
+            {
+                "role": "user",
+                "content": f"data:image/png;base64,{encoded_image}"  # This is where the image is passed as base64 directly
             }
         ],
-        "image": {
-            "type": "image_url",
-            "image_url": f"data:image/png;base64,{encoded_image}"
-        },
         "temperature": 0
     }