Create app.py

app.py

@@ -0,0 +1,45 @@
+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([
+        [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)]
+    return sepia_imgs
+
+def zip_sepia_images(sepia_imgs):
+    temp_dir = tempfile.mkdtemp()
+    zip_path = os.path.join(temp_dir, "sepia_images.zip")
+    with ZipFile(zip_path, 'w') as zipf:
+        for i, img in enumerate(sepia_imgs):
+            img_path = os.path.join(temp_dir, f"sepia_image_{i}.png")
+            gr.utils.save_image(img_path, img)
+            zipf.write(img_path, os.path.basename(img_path))
+    return zip_path
+
+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=False)
+        
+    input_img.change(fn=lambda x: x, inputs=input_img, 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()