Update app.py

app.py

@@ -77,6 +77,22 @@ def save_image_and_show(image, folder_name="output_images"):
     # Show the image to the user (not the path)
     st.image(image, caption="Saved Image", channels="RGB")
 
+    zip_filename = f"{folder_name}.zip"
+    with zipfile.ZipFile(zip_filename, 'w') as zipf:
+        for root, dirs, files in os.walk(folder_name):
+            for file in files:
+                filepath = os.path.join(root, file)
+                arcname = os.path.relpath(filepath, folder_name)
+                zipf.write(filepath, arcname)
+
+    # Read zip for download
+    with open(zip_filename, "rb") as f:
+        st.download_button(
+            label="Download All Images as ZIP",
+            data=f,
+            file_name=zip_filename,
+            mime="application/zip"
+        )