Update app.py

app.py

@@ -154,46 +154,56 @@ def parse_srt(content: str):
     return parsed
 
 
-
 logger = logging.getLogger(__name__)
 
-def compress_advanced(image_bytes, max_width=800, use_webp=True):
+def compress_image(image_bytes, max_width=800, quality=80):
+    """
+    Compresses an image to WebP (best) or optimized JPEG.
+    Renamed back to 'compress_image' to fix your error.
+    """
     try:
         img = Image.open(io.BytesIO(image_bytes))
 
-        # 1. Resize efficiently (using thumbnail is faster/safer than manual math)
+        # 1. Efficient Resize (Using thumbnail prevents upscaling artifacts)
         img.thumbnail((max_width, max_width), Image.Resampling.LANCZOS)
 
         buffer = io.BytesIO()
 
+        # 2. Try WebP first (Best quality/size ratio)
+        # If you strictly need JPEG, change use_webp to False
+        use_webp = True 
+        
         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)
+            # method=6 is the strongest compression algo for WebP
+            img.save(buffer, format="WEBP", quality=quality, method=6)
         else:
-            # OPTION B: Optimized JPEG (If you strictly need JPG)
-            # Handle transparency (paste on white) only for JPEG
+            # Fallback: Optimized JPEG
+            # Handle transparency (paste on white)
             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)
+                # Handle paletted images with transparency
+                if img.mode == 'P':
+                    img = img.convert('RGBA')
+                background.paste(img, mask=img.split()[3])
                 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)
+                quality=quality, 
+                optimize=True, 
+                progressive=True,
+                subsampling=0
             )
 
         return buffer.getvalue()
 
     except Exception as e:
-        logger.error(f"Compression failed: {e}")
+        logger.error(f"Image compression failed: {e}")
+        # If logging isn't setup, print the error so you can see it
+        print(f"Error: {e}")
         return None
 
 def process_batch_gemini(api_key, items, model_name):