Update app.py

app.py

@@ -161,7 +161,7 @@ def get_image_size_kb(image):
     size_bytes = buffer.tell()
     return size_bytes / 1024  # Convert to KB
 
-def upload_to_s3(image, user_id, folder=""):
+def upload_to_s3(image, user_id, folder="", force_quality=None):
     """
     Upload image to S3 bucket and return the S3 path
     
@@ -169,6 +169,7 @@ def upload_to_s3(image, user_id, folder=""):
         image: PIL Image object
         user_id: User ID for folder structure
         folder: Subfolder to store the image in (e.g., "raw-uploads" or "processed-512x512")
+        force_quality: Override default quality settings if specified
     """
     if st.session_state.get("demo_mode", False):
         return f"demo/{user_id}/demo_image.jpg"
@@ -186,8 +187,14 @@ def upload_to_s3(image, user_id, folder=""):
         
         # Convert PIL image to bytes
         buffer = BytesIO()
-        # Higher quality for raw uploads, compressed for processed
-        quality = 95 if folder == "raw-uploads" else 85
+        
+        # Set quality based on folder or forced value
+        if force_quality is not None:
+            quality = force_quality
+        else:
+            # Higher quality for raw uploads, compressed for processed
+            quality = 95 if folder == "raw-uploads" else 85
+            
         image.save(buffer, format="JPEG", quality=quality, optimize=True)
         buffer.seek(0)