Update app.py

app.py

@@ -1,14 +1,56 @@
 import gradio as gr
+import numpy as np
+from PIL import Image
+import tempfile
+import shutil
+import os
+from zipfile import ZipFile
+import io
 
-def dynamic_file_download():
-    # This could be a dynamically generated file content
-    file_content = "Dynamically generated content."
-    return "dynamic_file.txt", file_content
+def sepia(input_img, num_copies):
+    sepia_filter = np.array([
+        [0.393, 0.769, 0.189], 
+        [0.349, 0.686, 0.168], 
+        [0.272, 0.534, 0.131]
+    ])
+    input_img = np.array(input_img) / 255.0
+    sepia_imgs = []
+    for _ in range(num_copies):
+        sepia_img = np.dot(input_img[..., :3], sepia_filter.T)
+        sepia_img = np.clip(sepia_img, 0, 1) * 255
+        sepia_imgs.append(Image.fromarray(sepia_img.astype(np.uint8)))
+    return sepia_imgs
 
-iface = gr.Interface(
-    fn=lambda x: x,  # Placeholder function for the interface
-    inputs=gr.Textbox(),  # Example input component
-    outputs=gr.DownloadButton(label="📂 Click to download dynamically generated file", value=dynamic_file_download)
-)
+def zip_sepia_images(sepia_imgs):
+    # Use BytesIO to avoid actual file system operations for ZIP file creation
+    zip_bytes = io.BytesIO()
+    with ZipFile(zip_bytes, 'w') as zipf:
+        for i, img in enumerate(sepia_imgs):
+            # Convert PIL Image to bytes
+            img_byte_arr = io.BytesIO()
+            img.save(img_byte_arr, format='PNG')
+            img_byte_arr.seek(0)
+            zipf.writestr(f"sepia_image_{i}.png", img_byte_arr.getvalue())
+    zip_bytes.seek(0)
+    return zip_bytes, "sepia_images.zip"
 
-iface.launch()
+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")
+
+    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)
+
+    generate_btn = gr.Button("Generate Sepia Images")
+    generate_btn.click(fn=update_output, 
+                       inputs=[input_img, num_copies], 
+                       outputs=[gallery, download_btn])
+
+if __name__ == "__main__":
+    demo.launch()