Update app.py

app.py

@@ -66,7 +66,7 @@ def create_gif_imageio(images, duration=50, loop=0):
     images_pil = [Image.fromarray(img) for img in images]
     imageio.mimsave(gif_path, images_pil, duration=duration, loop=loop)
     return gif_path
-
+    
 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)
@@ -77,7 +77,13 @@ def process_image_to_gif(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)
-    return gif_file
+    
+    # Preview the first frame of the GIF
+    preview_image = Image.fromarray(enhanced_frames[0])
+    preview_image.save(os.path.join(GENERATED_FILES_DIR, "preview_frame.png"))
+    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")
@@ -99,9 +105,14 @@ with gr.Blocks() as demo:
         zip_button = gr.Button("Create Zip File")
         gif_button = gr.Button("Create GIF")
     with gr.Row():
+        preview_image = gr.Image(label="Preview GIF Frame")
         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)
+    gif_button.click(
+        process_image_to_gif, 
+        inputs=[image_input, grid_cols_input, grid_rows_input], 
+        outputs=[preview_image, gif_output]
+    )
 
 demo.launch(show_error=True)