Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import base64
 from PIL import Image
 from io import BytesIO
+import zlib  # Compression library to reduce image size
 
 # Set page title and layout
 st.set_page_config(page_title="Image Caption Generator", layout="centered")
@@ -24,14 +25,21 @@ def resize_image(image, max_width=500):
     resized_image = image.resize((max_width, new_height), Image.Resampling.LANCZOS)
     return resized_image
 
+# Function to compress the image before converting to Base64
+def compress_image(image, quality=70):
+    """
+    Compress the image to reduce the size before Base64 encoding.
+    """
+    buffered = BytesIO()
+    image.save(buffered, format="JPEG", quality=quality)  # Save as JPEG with compression
+    return buffered.getvalue()
+
 # Function to convert image to Base64
-def convert_to_base64(image):
+def convert_to_base64(image_data):
     """
-    Convert the image to Base64 encoding.
+    Convert the compressed image to Base64 encoding.
     """
-    buffer = BytesIO()
-    image.save(buffer, format="PNG")
-    image_base64 = base64.b64encode(buffer.getvalue()).decode()
+    image_base64 = base64.b64encode(image_data).decode()
     return image_base64
 
 # Function to call Nebius API
@@ -91,8 +99,11 @@ if uploaded_image:
     # Display the resized image
     st.image(resized_image, caption="Resized Image", use_container_width=True)
 
-    # Convert resized image to Base64
-    base64_string = convert_to_base64(resized_image)
+    # Compress the image before converting to Base64
+    compressed_image_data = compress_image(resized_image)
+
+    # Convert compressed image to Base64
+    base64_string = convert_to_base64(compressed_image_data)
 
     # Display Base64 output
     st.subheader("Base64 Encoded String")