Update app.py

app.py

@@ -1,8 +1,7 @@
+from pathlib import Path
 import gradio as gr
 import numpy as np
-import tempfile
-import shutil
-import os
+from zipfile import ZipFile
 
 def sepia(input_img, num_copies):
     sepia_filter = np.array([
@@ -16,20 +15,29 @@ def sepia(input_img, num_copies):
     sepia_imgs = [(sepia_img).astype(np.uint8) for _ in range(num_copies)]
     return sepia_imgs
 
+def zip_sepia_images(sepia_imgs):
+    temp_dir = Path(tempfile.mkdtemp())  # Use Path for easier file handling
+    zip_path = temp_dir / "sepia_images.zip"
+    with ZipFile(zip_path, 'w') as zipf:
+        for i, img in enumerate(sepia_imgs):
+            img_path = temp_dir / f"sepia_image_{i}.png"
+            gr.utils.save_image(str(img_path), img)  # Convert Path object to string for compatibility
+            zipf.write(img_path, img_path.name)
+    return str(zip_path)  # Convert Path object to string for return
 
 with gr.Blocks() as demo:
     with gr.Row():
         input_img = gr.Image()
         num_copies = gr.Number(label="Number of Copies", value=1)
         gallery = gr.Gallery(label="Sepia Images")
-       
+        download_btn = gr.File(label="Download ZIP", visible=True)
         
     input_img.change(fn=lambda x: x, inputs=input_img, outputs=gallery)
-    #num_copies.change(fn=lambda x: x, inputs=num_copies, outputs=gallery)
+    num_copies.change(fn=lambda x: x, inputs=num_copies, outputs=gallery)
     
     generate_btn = gr.Button("Generate Sepia Images")
     generate_btn.click(fn=sepia, inputs=[input_img, num_copies], outputs=gallery)
-  
+    generate_btn.click(fn=zip_sepia_images, inputs=gallery, outputs=download_btn)
 
 if __name__ == "__main__":
     demo.launch()