Update app.py

app.py

@@ -61,7 +61,8 @@ def enhance_gif(images):
         enhanced_images.append(np.array(img))
     return enhanced_images
 
-def create_gif_imageio(images, duration=50, loop=0):
+def create_gif_imageio(images, fps=10, loop=0):
+    duration = 1000 / fps  # Convert FPS to milliseconds
     gif_path = os.path.join(GENERATED_FILES_DIR, "output_enhanced.gif")
     images_pil = [Image.fromarray(img) for img in images]
     imageio.mimsave(gif_path, images_pil, duration=duration, loop=loop)
@@ -72,11 +73,11 @@ def process_image(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):
+def process_image_to_gif(image, grid_cols_input, grid_rows_input, fps_input):
     frames = split_image_grid(image, grid_cols_input, grid_rows_input)
     interpolated_frames = interpolate_frames(frames, factor=2)
     enhanced_frames = enhance_gif(interpolated_frames)
-    gif_file = create_gif_imageio(enhanced_frames, duration=50, loop=0)
+    gif_file = create_gif_imageio(enhanced_frames, fps=fps_input, loop=0)
     
     # Preview the first frame of the GIF
     preview_image = Image.fromarray(enhanced_frames[0])
@@ -84,7 +85,7 @@ def process_image_to_gif(image, grid_cols_input, grid_rows_input):
     preview_path = os.path.join(GENERATED_FILES_DIR, "preview_frame.png")
     
     return preview_path, gif_file
-
+    
 def zip_images(images):
     zip_path = os.path.join(GENERATED_FILES_DIR, "output.zip")
     with zipfile.ZipFile(zip_path, 'w') as zipf:
@@ -101,6 +102,7 @@ with gr.Blocks() as demo:
         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")
+        fps_input = gr.Slider(1, 30, value=10, step=1, label="FPS (Frames per Second)")
     with gr.Row():
         zip_button = gr.Button("Create Zip File")
         gif_button = gr.Button("Create GIF")
@@ -111,7 +113,7 @@ with gr.Blocks() as demo:
     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], 
+        inputs=[image_input, grid_cols_input, grid_rows_input, fps_input], 
         outputs=[preview_image, gif_output]
     )