Update app.py

app.py

@@ -4,49 +4,43 @@ from PIL import Image
 import io
 from zipfile import ZipFile
 
-def sepia_and_zip(input_img_path, num_copies):
-    input_img = Image.open(input_img_path)
+def sepia_and_zip(input_img, num_copies):
+    # Convert input_img from Image format to NumPy array
     input_img_np = np.array(input_img)
-    
+
+    # Define the sepia filter
     sepia_filter = np.array([
         [0.393, 0.769, 0.189],
         [0.349, 0.686, 0.168],
         [0.272, 0.534, 0.131]
     ])
+
+    # Apply sepia filter
     input_img_np = input_img_np / 255.0
     sepia_img_np = np.dot(input_img_np[..., :3], sepia_filter.T)
     sepia_img_np = np.clip(sepia_img_np, 0, 1) * 255
     sepia_imgs = [(sepia_img_np).astype(np.uint8) for _ in range(num_copies)]
     
-    # Convert numpy images to PIL images for gallery
+    # Convert NumPy images back to PIL images for output
     pil_images = [Image.fromarray(img) for img in sepia_imgs]
     
-    # Prepare ZIP file
+    # Prepare the ZIP file in-memory
     zip_bytes_io = io.BytesIO()
     with ZipFile(zip_bytes_io, 'w') as zip_file:
         for i, img in enumerate(pil_images):
             img_byte_arr = io.BytesIO()
             img.save(img_byte_arr, format='PNG')
-            img_byte_arr = img_byte_arr.getvalue()
-            zip_file.writestr(f"sepia_image_{i}.png", img_byte_arr)
+            img_byte_arr.seek(0)
+            zip_file.writestr(f"sepia_image_{i}.png", img_byte_arr.read())
     zip_bytes_io.seek(0)
-    
-    # Return a gallery of images and a ZIP file for download
-    return pil_images, zip_bytes_io
 
-title = "Sepia Tone Generator with Downloadable ZIP"
-description = "Upload an image and select the number of sepia-toned copies you want. View the images in a gallery and download them all as a ZIP file."
-examples = []
+    # Return values: Gallery images and the ZIP file as a tuple (BytesIO, filename)
+    return pil_images, ("sepia_images.zip", zip_bytes_io)
 
 iface = gr.Interface(fn=sepia_and_zip,
-                     inputs=[gr.Image(type="filepath", label="Input Image"),
-                             gr.Number(label="Number of Copies")],
-                     outputs=[gr.Gallery(label="Sepia Images"),
-                              gr.File(label="Download ZIP")],
-                     title=title,
-                     description=description,
-                     examples=examples,
-                     allow_flagging="never",
-                     cache_examples=False)
+                     inputs=[gr.Image(label="Input Image"), gr.Number(label="Number of Copies", default=1)],
+                     outputs=[gr.Gallery(label="Sepia Images"), gr.File(label="Download ZIP")],
+                     title="Sepia Tone Generator with Downloadable ZIP",
+                     description="Upload an image and select the number of sepia-toned copies you want. View the images in a gallery and download them all as a ZIP file.")
 
-iface.launch(debug=True, enable_queue=False)
+iface.launch(debug=True)