| import gradio as gr |
| import numpy as np |
| from io import BytesIO |
| from PIL import Image |
| import zipfile |
| import os |
|
|
| class NamedBytesIO(BytesIO): |
| def __init__(self, *args, name="output.zip", **kwargs): |
| super(NamedBytesIO, self).__init__(*args, **kwargs) |
| self.name = name |
|
|
| def split_image_grid(image, grid_cols, grid_rows): |
| if isinstance(image, np.ndarray): |
| image = Image.fromarray(image) |
| |
| width, height = image.width, image.height |
| cell_width = width // grid_cols |
| cell_height = height // grid_rows |
| frames = [] |
| for i in range(grid_rows): |
| for j in range(grid_cols): |
| left = j * cell_width |
| upper = i * cell_height |
| right = left + cell_width |
| lower = upper + cell_height |
| frame = image.crop((left, upper, right, lower)) |
| frames.append(np.array(frame)) |
| return frames |
|
|
| def zip_images(images): |
| zip_buffer = NamedBytesIO(name="output.zip") |
| with zipfile.ZipFile(zip_buffer, 'w') as zipf: |
| for idx, img in enumerate(images): |
| img_buffer = BytesIO() |
| img = Image.fromarray(img) |
| img.save(img_buffer, format='PNG') |
| img_buffer.seek(0) |
| zipf.writestr(f'image_{idx}.png', img_buffer.getvalue()) |
| zip_buffer.seek(0) |
| return zip_buffer |
|
|
| def create_gif(images): |
| gif_buffer = NamedBytesIO(name="output.gif") |
| images_pil = [Image.fromarray(img) for img in images] |
| images_pil[0].save(gif_buffer, format='GIF', save_all=True, append_images=images_pil[1:], duration=100, loop=0) |
| gif_buffer.seek(0) |
| return gif_buffer |
|
|
| def process_image(image, grid_cols_input, grid_rows_input): |
| frames = split_image_grid(image, grid_cols_input, grid_rows_input) |
| zip_file = zip_images(frames) |
| return zip_file |
|
|
| def process_image_to_gif(image, grid_cols_input, grid_rows_input): |
| frames = split_image_grid(image, grid_cols_input, grid_rows_input) |
| gif_file = create_gif(frames) |
| return gif_file |
|
|
| with gr.Blocks() as demo: |
| with gr.Row(): |
| image_input = gr.Image(label="Input Image", type="pil") |
| grid_cols_input = gr.Slider(1, 10, value=2, step=1, label="Grid Columns") |
| grid_rows_input = gr.Slider(1, 10, value=2, step=1, label="Grid Rows") |
| with gr.Row(): |
| zip_button = gr.Button("Create Zip File") |
| gif_button = gr.Button("Create GIF") |
| with gr.Row(): |
| zip_output = gr.File(label="Download Zip File") |
| gif_output = gr.File(label="Download GIF") |
| zip_button.click(process_image, inputs=[image_input, grid_cols_input, grid_rows_input], outputs=zip_output) |
| gif_button.click(process_image_to_gif, inputs=[image_input, grid_cols_input, grid_rows_input], outputs=gif_output) |
|
|
| demo.launch(show_error=True) |
