Update app.py

app.py

@@ -33,6 +33,16 @@ def show_gallery():
     files = sorted(IMAGE_DIR.iterdir(), key=os.path.getmtime, reverse=True)
     return [str(f) for f in files]
 
+def clear_gallery():
+    """Deletes all files in the image directory."""
+    if IMAGE_DIR.exists():
+        for file_path in IMAGE_DIR.iterdir():
+            if file_path.is_file():
+                file_path.unlink() # Deletes the file
+        print("Gallery cleared successfully.")
+    # Return an empty list to update the gallery UI
+    return []
+
 def upload_via_ui(image_from_ui):
     """
     Handles image uploads directly from the Gradio UI interface.
@@ -104,6 +114,9 @@ with gr.Blocks(title="Shekar's Portfolio Gallery") as demo:
             outputs=gallery
         )
 
+    with gr.Accordion("Admin Controls", open=False):
+        clear_btn = gr.Button("⚠️ Clear All Gallery Images")
+
     # --- Hidden components to create the API endpoints ---
     with gr.Row(visible=False):
         # Textbox to receive the base64 data URL from the JS client
@@ -117,6 +130,8 @@ with gr.Blocks(title="Shekar's Portfolio Gallery") as demo:
             outputs=gallery, 
             api_name="upload_via_data_url"
         )
+
+    clear_btn.click(fn=clear_gallery, inputs=None, outputs=gallery)
         
     # Load and refresh the gallery when the app starts.
     # This also creates the /show_gallery endpoint for the JS client.