Update app.py

app.py

@@ -2,16 +2,19 @@ import gradio as gr
 from pathlib import Path
 import subprocess
 
-# Paths
-FRAME1 = Path("demo/frame1.jpg")
-FRAME2 = Path("demo/frame2.jpg")
 OUTPUT_GIF = Path("demo/demo.gif")
 
-def generate_demo_gif():
+def generate_demo_gif(img1, img2):
+    # Save uploaded images
+    frame1_path = Path("demo/frame1.jpg")
+    frame2_path = Path("demo/frame2.jpg")
+    img1.save(frame1_path)
+    img2.save(frame2_path)
+
     try:
         result = subprocess.run([
             "python", "inference_img.py",
-            "--img", str(FRAME1), str(FRAME2),
+            "--img", str(frame1_path), str(frame2_path),
             "--exp=4"
         ], capture_output=True, text=True)
 
@@ -21,20 +24,24 @@ def generate_demo_gif():
         if result.returncode == 0 and OUTPUT_GIF.exists():
             return OUTPUT_GIF
         else:
-            return None
+            return "GIF generation failed."
     except Exception as e:
         print(f"Error: {e}")
-        return None
-
-def demo_tab():
-    gif_path = generate_demo_gif()
-    if gif_path:
-        return gif_path
-    else:
-        return "GIF generation failed."
+        return "Error during GIF generation."
 
 with gr.Blocks() as demo_ui:
     with gr.Tab("Demo"):
-        gr.Image(value=demo_tab(), label="Generated GIF")
+        gr.Markdown("### Upload two frames to generate a GIF")
+        with gr.Row():
+            img1_input = gr.Image(type="pil", label="Frame 1")
+            img2_input = gr.Image(type="pil", label="Frame 2")
+        generate_btn = gr.Button("Generate GIF")
+        output_gif = gr.Image(label="Generated GIF")
+
+        generate_btn.click(
+            fn=generate_demo_gif,
+            inputs=[img1_input, img2_input],
+            outputs=output_gif
+        )
 
 demo_ui.launch()