Spaces:
Runtime error
Runtime error
File size: 6,305 Bytes
aa85975 f647b33 aa85975 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 f647b33 d8c4344 |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 |
import gradio as gr
from PIL import Image, ImageDraw, ImageFont, ImageSequence
import numpy as np
def create_grid(image_file, grid_x, grid_y, font_size, font_color, position, border_size, border_color):
image_file.file.seek(0) # Seek to the beginning of the file
# Open the image from bytes
img = Image.open(image_file.name)
total_frames = img.n_frames
selected_frames_count = grid_x * grid_y
# Select evenly spaced frames
selected_frames_indices = np.linspace(0, total_frames - 1, selected_frames_count).astype(int)
selected_frames = [img.seek(i) or img.copy() for i in selected_frames_indices]
# Modify frames by adding border and number
modified_frames = []
try:
font = ImageFont.truetype("Lato-Regular.ttf", font_size)
except IOError:
print("Font not found, using default font.")
font = ImageFont.load_default()
positions = {
"Top Left": (20, 20),
"Top Right": (img.width - 20 - font_size, 20),
"Bottom Left": (20, img.height - 20 - font_size),
"Bottom Right": (img.width - 20 - font_size, img.height - 20 - font_size)
}
for i, frame in enumerate(selected_frames):
# Add border
border_width = border_size
frame_with_border = Image.new('RGB', (frame.width + 2*border_width, frame.height + 2*border_width), border_color.lower())
frame_with_border.paste(frame, (border_width, border_width))
# Add number
draw = ImageDraw.Draw(frame_with_border)
text = str(i + 1)
text_position = (border_width + positions[position][0], border_width + positions[position][1])
draw.text(text_position, text, font=font, fill=font_color)
modified_frames.append(frame_with_border)
# Combine modified frames into a grid
grid_width = modified_frames[0].width * grid_x
grid_height = modified_frames[0].height * grid_y
grid_img = Image.new('RGB', (grid_width, grid_height), border_color.lower())
for i, frame in enumerate(modified_frames):
x_offset = (i % grid_x) * frame.width
y_offset = (i // grid_x) * frame.height
grid_img.paste(frame, (x_offset, y_offset))
output_info = f"Grid size: {grid_x} x {grid_y}\n\nSelected Frames: {selected_frames_count} / {total_frames} ({selected_frames_count / total_frames * 100:.2f}%)"
return grid_img, output_info
def gif_info(image_file, grid_x, grid_y, font_size, font_color, position, border_size, border_color):
image_file.file.seek(0)
img = Image.open(image_file.name)
total_frames = img.n_frames
frame_rate = img.info.get('duration', 100) / 1000.0 # Convert to seconds
# Generate grid using create_grid function
grid_img, output_info = create_grid(image_file, grid_x, grid_y, font_size, font_color, position, border_size, border_color)
gif_details = f"\n\n**Total Frames:** {total_frames}\n\n**Frame Rate:** {frame_rate} sec/frame\n\n{output_info}"
return grid_img, gif_details
with gr.Blocks() as app:
gr.Markdown('## GIF Grid Generator')
gr.Markdown('Upload a GIF to generate a grid from its frames. Use the sliders to adjust the grid size and text settings.')
with gr.Row():
with gr.Column():
control_image = gr.File(label="Upload a GIF", type="file", elem_id="file_upload", file_types=[".gif"])
gif_details = gr.Markdown("Upload a gif")
grid_x_slider = gr.Slider(minimum=1, maximum=10, step=1, value=3, label="Grid X Size")
grid_y_slider = gr.Slider(minimum=1, maximum=10, step=1, value=3, label="Grid Y Size")
font_color_dropdown = gr.Dropdown(choices=["Black", "White", "Red", "Green", "Blue"], value="White", label="Numbering Color")
position_radio = gr.Radio(choices=["Top Left", "Top Right", "Bottom Left", "Bottom Right"], value="Top Left", label="Numbering Position")
font_size_slider = gr.Slider(minimum=10, maximum=100, step=5, value=40, label="Font Size")
border_color_dropdown = gr.Dropdown(choices=["Black", "White", "Red", "Green", "Blue"], value="White", label="Border Color")
border_size_slider = gr.Slider(minimum=0, maximum=100, step=5, value=10, label="Border Size")
with gr.Column():
result_image = gr.Image(label="Generated Grid")
# Use .change() method to listen for changes in any of the controls
control_image.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
grid_x_slider.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
grid_y_slider.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
font_size_slider.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
font_color_dropdown.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
position_radio.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
border_size_slider.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
border_color_dropdown.change(gif_info, inputs=[control_image, grid_x_slider, grid_y_slider, font_size_slider, font_color_dropdown, position_radio, border_size_slider, border_color_dropdown], outputs=[result_image, gif_details])
if __name__ == "__main__":
app.launch()
|