Update app.py

app.py

@@ -5,36 +5,54 @@ from PIL import Image
 import gradio as gr
 
 def resize_icons(zip_file_path, width, height):
-    # Create a temporary working directory
+    valid_extensions = (".png", ".jpg", ".jpeg", ".bmp", ".gif", ".webp")
+
+    # Create temporary working directory
     with tempfile.TemporaryDirectory() as tmpdir:
-        # Extract ZIP contents
+        # 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)
 
-        # Prepare resized image output folder
+        # Prepare output folder for resized images
         output_dir = os.path.join(tmpdir, "resized")
         os.makedirs(output_dir, exist_ok=True)
 
-        # Resize all valid images
+        files_processed = 0
+
+        # Resize valid 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
+
                 file_path = os.path.join(root, file)
                 try:
                     img = Image.open(file_path)
                     img = img.convert("RGBA")
                     img_resized = img.resize((int(width), int(height)), Image.LANCZOS)
 
-                    # Maintain relative path
+                    # Save resized image keeping subfolder 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)
                     img_resized.save(output_path)
+                    files_processed += 1
                 except Exception as e:
-                    print(f"Skipping non-image file {file_path}: {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 resized files
+        # Zip the resized images
         output_zip_path = os.path.join(tmpdir, "resized_icons.zip")
         with zipfile.ZipFile(output_zip_path, 'w') as zipf:
             for root, _, files in os.walk(output_dir):
@@ -54,8 +72,8 @@ demo = gr.Interface(
         gr.Number(label="Height", value=64)
     ],
     outputs=gr.File(label="Download Resized ZIP"),
-    title="Icon Resizer",
-    description="Upload a ZIP of icons/images. Set target width and height. Download a ZIP of resized icons."
+    title="📦 Icon Resizer",
+    description="Upload a ZIP of images (icons). Set a target width and height. Get back a ZIP with resized icons."
 )
 
 if __name__ == "__main__":