File size: 1,606 Bytes
16b7d61 e50d873 9b73a59 16b7d61 8579fd1 e50d873 8579fd1 e50d873 0ce663a 9b73a59 e50d873 9b73a59 d7324f8 9b73a59 e50d873 8579fd1 9b73a59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
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=True)
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()
|