Update app.py

app.py

@@ -21,18 +21,25 @@ def sepia(input_img, num_copies):
         sepia_imgs.append(Image.fromarray(sepia_img.astype(np.uint8)))
     return sepia_imgs
 
+
 def zip_sepia_images(sepia_imgs):
-    # Use BytesIO to avoid actual file system operations for ZIP file creation
+    # Initialize a BytesIO object to hold the ZIP file in memory
     zip_bytes = io.BytesIO()
     with ZipFile(zip_bytes, 'w') as zipf:
         for i, img in enumerate(sepia_imgs):
-            # Convert PIL Image to bytes
+            # Convert the PIL Image to a bytes object
             img_byte_arr = io.BytesIO()
             img.save(img_byte_arr, format='PNG')
             img_byte_arr.seek(0)
+            # Write the image bytes to the ZIP file
             zipf.writestr(f"sepia_image_{i}.png", img_byte_arr.getvalue())
+    # IMPORTANT: Seek back to the start of the BytesIO object
     zip_bytes.seek(0)
-    return zip_bytes, "sepia_images.zip"
+    
+    # Return the ZIP file as a tuple: (filename, bytes)
+    # Note: Ensure the filename has the correct file extension (.zip)
+    return "sepia_images.zip", zip_bytes.getvalue()
+
 
 with gr.Blocks() as demo:
     with gr.Row():
@@ -42,10 +49,10 @@ with gr.Blocks() as demo:
         gallery = gr.Gallery(label="Sepia Images")
         download_btn = gr.File(label="Download ZIP")
 
-    def update_output(input_img, num_copies):
-        sepia_imgs = sepia(input_img, num_copies)
-        zip_file, zip_name = zip_sepia_images(sepia_imgs)
-        return sepia_imgs, (zip_name, zip_file)
+def update_output(input_img, num_copies):
+    sepia_imgs = sepia(input_img, num_copies)
+    zip_name, zip_bytes = zip_sepia_images(sepia_imgs)
+    return sepia_imgs, (zip_name, zip_bytes)  # This tuple matches Gradio's expectation
 
     generate_btn = gr.Button("Generate Sepia Images")
     generate_btn.click(fn=update_output,