Update app.py

app.py

@@ -1,17 +1,21 @@
-import zipfile
 import os
+import zipfile
 import tempfile
 from PIL import Image
 import gradio as gr
+import shutil
 
 def resize_icons(zip_file_path, width, height):
+    import uuid
+
     valid_extensions = (".png", ".jpg", ".jpeg", ".bmp", ".gif", ".webp")
 
-    # Create temporary working directory
+    # Use temporary working directory for processing
     with tempfile.TemporaryDirectory() as tmpdir:
-        # Extract uploaded zip
+        # Extract uploaded ZIP
         extract_dir = os.path.join(tmpdir, "extracted")
         os.makedirs(extract_dir, exist_ok=True)
+
         with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
             zip_ref.extractall(extract_dir)
 
@@ -21,17 +25,16 @@ def resize_icons(zip_file_path, width, height):
 
         files_processed = 0
 
-        # Resize valid images
+        # Walk and resize images
         for root, _, files in os.walk(extract_dir):
-            # Skip macOS __MACOSX folders
             if "__MACOSX" in root:
                 continue
 
             for file in files:
                 if not file.lower().endswith(valid_extensions):
                     continue
-                if file.startswith("._"):  # Skip macOS metadata
-                    continue
+                if file.startswith("._"):
+                    continue  # Skip macOS metadata files
 
                 file_path = os.path.join(root, file)
                 try:
@@ -39,7 +42,7 @@ def resize_icons(zip_file_path, width, height):
                     img = img.convert("RGBA")
                     img_resized = img.resize((int(width), int(height)), Image.LANCZOS)
 
-                    # Save resized image keeping subfolder structure
+                    # Preserve folder structure
                     rel_path = os.path.relpath(file_path, extract_dir)
                     output_path = os.path.join(output_dir, rel_path)
                     os.makedirs(os.path.dirname(output_path), exist_ok=True)
@@ -48,22 +51,23 @@ def resize_icons(zip_file_path, width, height):
                 except Exception as e:
                     print(f"Skipping {file_path}: {e}")
 
-        # If nothing processed, raise error
         if files_processed == 0:
             raise FileNotFoundError("No valid image files found to resize.")
 
-        # Zip the resized images
-        output_zip_path = os.path.join(tmpdir, "resized_icons.zip")
-        with zipfile.ZipFile(output_zip_path, 'w') as zipf:
+        # Write ZIP to persistent path
+        unique_id = uuid.uuid4().hex
+        final_zip_path = f"/tmp/resized_icons_{unique_id}.zip"
+
+        with zipfile.ZipFile(final_zip_path, 'w') as zipf:
             for root, _, files in os.walk(output_dir):
                 for file in files:
                     full_path = os.path.join(root, file)
                     arcname = os.path.relpath(full_path, output_dir)
                     zipf.write(full_path, arcname)
 
-        return output_zip_path
+        return final_zip_path
 
-# Gradio interface
+# Gradio UI
 demo = gr.Interface(
     fn=resize_icons,
     inputs=[
@@ -73,7 +77,7 @@ demo = gr.Interface(
     ],
     outputs=gr.File(label="Download Resized ZIP"),
     title="📦 Icon Resizer",
-    description="Upload a ZIP of images (icons). Set a target width and height. Get back a ZIP with resized icons."
+    description="Upload a ZIP of images (icons). Set width and height. Download resized icons as a ZIP."
 )
 
 if __name__ == "__main__":