Update app.py

app.py

@@ -5,36 +5,38 @@ import zipfile
 import tempfile
 import shutil
 
-def make_icons_white_in_folder(folder_path):
-    for root, _, files in os.walk(folder_path):
-        for filename in files:
-            if filename.lower().endswith('.png'):
-                filepath = os.path.join(root, filename)
-                image = Image.open(filepath).convert("RGBA")
-                pixels = image.load()
-
-                for y in range(image.height):
-                    for x in range(image.width):
-                        r, g, b, a = pixels[x, y]
-                        if a != 0:
-                            pixels[x, y] = (255, 255, 255, a)
-
-                image.save(filepath)
-
-def process_zip(zip_file):
-    # Create temp directories
+def recolor_icons(zip_file, hex_color):
+    # Convert hex color to RGB tuple
+    hex_color = hex_color.lstrip("#")
+    target_rgb = tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
+
+    # Set up temp directories
     temp_input_dir = tempfile.mkdtemp()
     temp_output_zip = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
+    temp_output_zip.close()
 
     try:
-        # Extract uploaded zip
+        # Extract zip file
         with zipfile.ZipFile(zip_file, 'r') as zip_ref:
             zip_ref.extractall(temp_input_dir)
 
-        # Process PNGs in the folder
-        make_icons_white_in_folder(temp_input_dir)
+        # Process PNGs
+        for root, _, files in os.walk(temp_input_dir):
+            for filename in files:
+                if filename.lower().endswith('.png'):
+                    filepath = os.path.join(root, filename)
+                    image = Image.open(filepath).convert("RGBA")
+                    pixels = image.load()
+
+                    for y in range(image.height):
+                        for x in range(image.width):
+                            r, g, b, a = pixels[x, y]
+                            if a != 0:  # If not transparent
+                                pixels[x, y] = (*target_rgb, a)
 
-        # Repack processed files into a new zip
+                    image.save(filepath)
+
+        # Create new zip
         with zipfile.ZipFile(temp_output_zip.name, 'w') as zip_out:
             for root, _, files in os.walk(temp_input_dir):
                 for file in files:
@@ -45,17 +47,25 @@ def process_zip(zip_file):
         return temp_output_zip.name
 
     finally:
-        # Clean up the temp directory
         shutil.rmtree(temp_input_dir)
 
-# Gradio interface
-demo = gr.Interface(
-    fn=process_zip,
-    inputs=gr.File(label="Upload ZIP of PNGs", file_types=[".zip"]),
-    outputs=gr.File(label="Download Processed ZIP"),
-    title="PNG Icon Whitenifier",
-    description="Upload a ZIP file containing PNG images. The tool will turn all visible parts of each PNG white (preserving transparency) and give you a new ZIP for download."
-)
+# Gradio UI
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎨 PNG Recoloring Tool")
+    gr.Markdown(
+        "Upload a **ZIP** file of `.png` icons and choose a color to apply to all visible pixels.\n"
+        "Transparency is preserved — only the visible parts are recolored."
+    )
+
+    with gr.Row():
+        zip_input = gr.File(label="📁 Upload ZIP of PNGs", file_types=[".zip"])
+        color_picker = gr.ColorPicker(label="🎨 Choose Color", value="#ffffff")
+
+    output_zip = gr.File(label="⬇️ Download Recolored ZIP")
+
+    submit_btn = gr.Button("🔄 Recolor Icons")
+
+    submit_btn.click(fn=recolor_icons, inputs=[zip_input, color_picker], outputs=output_zip)
 
 if __name__ == "__main__":
     demo.launch()