Update app.py

app.py

@@ -1,25 +1,26 @@
 import gradio as gr
 import numpy as np
-import os
 from PIL import Image
 import io
 from zipfile import ZipFile
 
-def sepia_and_zip(input_img, num_copies):
-    # Apply the sepia filter to the image
+def sepia_and_zip(input_img_path, num_copies):
+    input_img = Image.open(input_img_path)
+    input_img_np = np.array(input_img)
+    
     sepia_filter = np.array([
         [0.393, 0.769, 0.189],
         [0.349, 0.686, 0.168],
         [0.272, 0.534, 0.131]
     ])
-    input_img = input_img / 255.0
-    sepia_img = np.dot(input_img[..., :3], sepia_filter.T)
-    sepia_img = np.clip(sepia_img, 0, 1) * 255
-    sepia_imgs = [(sepia_img).astype(np.uint8) for _ in range(num_copies)]
-
+    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
     pil_images = [Image.fromarray(img) for img in sepia_imgs]
-
+    
     # Prepare ZIP file
     zip_bytes_io = io.BytesIO()
     with ZipFile(zip_bytes_io, 'w') as zip_file:
@@ -29,21 +30,23 @@ def sepia_and_zip(input_img, num_copies):
             img_byte_arr = img_byte_arr.getvalue()
             zip_file.writestr(f"sepia_image_{i}.png", img_byte_arr)
     zip_bytes_io.seek(0)
-
+    
+    # Return a gallery of images and a ZIP file for download
     return pil_images, zip_bytes_io
 
-with gr.Blocks() as demo:
-    with gr.Row():
-        input_img = gr.Image()
-        num_copies = gr.Number(label="Number of Copies", value=1)
-    with gr.Row():
-        gallery = gr.Gallery(label="Sepia Images")
-        download_btn = gr.File(label="Download ZIP")
-        
-    btn = gr.Button("Generate Sepia Images")
-    btn.click(fn=sepia_and_zip, 
-              inputs=[input_img, num_copies], 
-              outputs=[gallery, download_btn])
+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 = []
+
+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)
 
-if __name__ == "__main__":
-    demo.launch()
+iface.launch(debug=True, enable_queue=False)