Update app.py

app.py

@@ -5,56 +5,56 @@ import gradio as gr
 from PIL import Image
 import io
 
-# Paths to pre-existing demo images
-FRAME1_PATH = Path("demo/frame1.png")
-FRAME2_PATH = Path("demo/frame2.png")
-OUTPUT_GIF = Path("/tmp/output.gif")  # Use /tmp for fast access
+# Fixed input paths and output location
+FRAME1_PATH = "demo/frame1.png"
+FRAME2_PATH = "demo/frame2.png"
+OUTPUT_GIF = "/tmp/output.gif"
 
 def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
-    progress(0.05, desc="Starting...")
+    progress(0.1, desc="Starting inference...")
 
-    # Clean previous output if any
-    if OUTPUT_GIF.exists():
-        OUTPUT_GIF.unlink(missing_ok=True)
+    # Delete old output if exists
+    try:
+        os.remove(OUTPUT_GIF)
+    except FileNotFoundError:
+        pass
 
-    # Build command using demo images directly
+    # Build and run command
     cmd = [
         "python", "inference_img.py",
-        "--img", str(FRAME1_PATH), str(FRAME2_PATH),
-        f"--exp={int(exp)}",
+        "--img", FRAME1_PATH, FRAME2_PATH,
+        "--exp", str(exp),
         "--model", "train_log/"
     ]
     print("Running:", " ".join(cmd))
-    progress(0.25, desc="Running model")
-
     result = subprocess.run(cmd, capture_output=True, text=True)
+
     print("STDOUT:", result.stdout)
     print("STDERR:", result.stderr)
 
-    # Check result
-    if result.returncode == 0 and OUTPUT_GIF.exists():
-        progress(0.85, desc="Loading GIF")
+    # Check and display result
+    if result.returncode == 0 and os.path.exists(OUTPUT_GIF):
         with open(OUTPUT_GIF, "rb") as f:
-            gif_bytes = f.read()
-        image = Image.open(io.BytesIO(gif_bytes))
-        progress(1.0, desc="Done")
-        return image, "✅ GIF generated successfully!"
+            gif = Image.open(io.BytesIO(f.read()))
+        progress(1.0, desc="GIF created!")
+        return gif, "✅ Done!"
     else:
-        return None, "❌ GIF generation failed. Check model and paths."
+        return None, "❌ Inference failed or output missing"
 
-# Build Gradio UI (no uploads)
+# UI setup
 with gr.Blocks() as demo_ui:
-    gr.Markdown("## 🌀 Img_InterpolationCutoff — Demo GIF from Preloaded Frames")
+    gr.Markdown("## 🎞️ Demo GIF Generator — Interpolate Two Frames")
 
     with gr.Row():
-        frame_preview_1 = gr.Image(value=str(FRAME1_PATH), interactive=False, label="Frame 1")
-        frame_preview_2 = gr.Image(value=str(FRAME2_PATH), interactive=False, label="Frame 2")
+        gr.Image(value=FRAME1_PATH, label="Frame 1", interactive=False)
+        gr.Image(value=FRAME2_PATH, label="Frame 2", interactive=False)
 
     exp = gr.Slider(1, 4, value=2, step=1, label="Interpolation Exponent")
     run_btn = gr.Button("Generate GIF")
-    out_gif = gr.Image(label="Animated GIF")
+    out_gif = gr.Image(label="Output GIF")
     status = gr.Markdown()
 
     run_btn.click(fn=generate_demo_gif, inputs=[exp], outputs=[out_gif, status])
 
+# Launch the app
 demo_ui.launch()