Update app.py

app.py

@@ -7,6 +7,8 @@ import shutil
 import logging
 import base64
 from concurrent.futures import ThreadPoolExecutor
+from PIL import Image, ImageOps
+
 
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.staticfiles import StaticFiles
@@ -151,24 +153,47 @@ def parse_srt(content: str):
 
     return parsed
 
-def compress_image(image_bytes, quality=70, max_width=800):
+
+
+logger = logging.getLogger(__name__)
+
+def compress_advanced(image_bytes, max_width=800, use_webp=True):
     try:
         img = Image.open(io.BytesIO(image_bytes))
-        if img.mode != 'RGB':
-            img = img.convert('RGB')
-        
-        width, height = img.size
-        if width > max_width:
-            height = int((height * max_width) / width)
-            width = max_width
-            img = img.resize((width, height), Image.Resampling.LANCZOS)
-        
+
+        # 1. Resize efficiently (using thumbnail is faster/safer than manual math)
+        img.thumbnail((max_width, max_width), Image.Resampling.LANCZOS)
+
         buffer = io.BytesIO()
-        # Quality expects integer 1-95
-        img.save(buffer, format="JPEG", quality=int(quality))
+
+        if use_webp:
+            # OPTION A: WebP (Best Balance)
+            # quality=80 in WebP looks like quality=90+ in JPEG but is much smaller
+            img.save(buffer, format="WEBP", quality=80, method=6)
+        else:
+            # OPTION B: Optimized JPEG (If you strictly need JPG)
+            # Handle transparency (paste on white) only for JPEG
+            if img.mode in ('RGBA', 'LA') or (img.mode == 'P' and 'transparency' in img.info):
+                background = Image.new('RGB', img.size, (255, 255, 255))
+                background.paste(img, mask=img.split()[3] if len(img.split()) > 3 else None)
+                img = background
+            elif img.mode != 'RGB':
+                img = img.convert('RGB')
+
+            # Advanced JPEG Optimization
+            img.save(
+                buffer, 
+                format="JPEG", 
+                quality=80, 
+                optimize=True,   # Extra pass to shrink file size
+                progressive=True, # Loads gradually on slow nets & usually smaller
+                subsampling=0    # Keeps edges sharp (4:4:4)
+            )
+
         return buffer.getvalue()
+
     except Exception as e:
-        logger.error(f"Image compression failed: {e}")
+        logger.error(f"Compression failed: {e}")
         return None
 
 def process_batch_gemini(api_key, items, model_name):