Update main.py

main.py

@@ -157,8 +157,13 @@ def process_receipt_endpoint():
 
         # Process the image with Gemini
         image = Image.open(io.BytesIO(image_bytes))
+        # --- Image Compression ---
+        compressed_buffer = io.BytesIO()
+        image.save(compressed_buffer, format="JPEG", optimize=True, quality=90) # Adjust quality as needed
+        compressed_image_bytes = compressed_buffer.getvalue()
+        compressed_image = Image.open(io.BytesIO(compressed_image_bytes)
         model = configure_gemini()
-        result_text = process_receipt(model, image)
+        result_text = process_receipt(model, compressed_image)
 
         try:
             json_str = result_text[result_text.find('{'):result_text.rfind('}')+1]