Update app.py

app.py

@@ -1,91 +1,58 @@
-import os
-import shutil
-import tempfile
-import threading
+# app.py
 import subprocess
 from pathlib import Path
-
 import gradio as gr
-from PIL import Image
-
-RIFE_DIR = Path("rife")  # Local bundled repo
-OUTPUT_DIR = RIFE_DIR / "output"
-LOCK = threading.Lock()
-
-def interpolate(a: Image.Image, b: Image.Image, fps: int = 14, exp: int = 2):
-    with LOCK:
-        # Normalize inputs
-        a = a.convert("RGB")
-        b = b.convert("RGB")
-        if b.size != a.size:
-            b = b.resize(a.size, Image.BICUBIC)
-
-        work_dir = Path(tempfile.mkdtemp(prefix="rife_run_"))
-        p1 = work_dir / "a.png"
-        p2 = work_dir / "b.png"
-        a.save(p1, "PNG")
-        b.save(p2, "PNG")
-
-        # Clean previous outputs
-        if OUTPUT_DIR.exists():
-            shutil.rmtree(OUTPUT_DIR)
-        OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
-
-        # Run RIFE inference
-        cmd = ["python3", str(RIFE_DIR / "inference_img.py"), "--img", str(p1), str(p2)]
-        if isinstance(exp, int) and exp >= 1:
-            cmd += ["--exp", str(exp)]
-        subprocess.run(cmd, cwd=str(RIFE_DIR), check=True)
-
-        # Collect interpolated frames
-        frames = []
-        i = 1
-        while True:
-            fp = OUTPUT_DIR / f"img{i}.png"
-            if not fp.exists():
-                break
-            frames.append(fp)
-            i += 1
-        if not frames:
-            raise RuntimeError("No frames generated.")
-
-        # Build GIF
-        images = [Image.open(p).convert("RGBA") for p in frames]
-        duration_ms = max(1, int(1000 / max(1, fps)))
-        gif_path = work_dir / "interpolation.gif"
-        images[0].save(
-            gif_path,
-            save_all=True,
-            append_images=images[1:],
-            optimize=False,
-            duration=duration_ms,
-            loop=0,
-            disposal=2,
-        )
-
-        # Optional cleanup
-        try:
-            shutil.rmtree(OUTPUT_DIR)
-        except Exception:
-            pass
-
-        return str(gif_path)
-
-# Gradio UI
-TITLE = "🔥 RIFE Interpolation Demo (PyTorch, Local)"
-with gr.Blocks(title=TITLE, analytics_enabled=False) as demo:
-    gr.Markdown(f"# {TITLE}")
-    with gr.Row():
-        with gr.Column():
-            img_a = gr.Image(type="pil", label="Image A")
-            img_b = gr.Image(type="pil", label="Image B")
-        with gr.Column():
-            fps = gr.Slider(6, 30, value=14, step=1, label="FPS")
-            exp = gr.Slider(1, 4, value=2, step=1, label="Interpolation exponent")
-            run = gr.Button("Interpolate", variant="primary")
-    gif_out = gr.Image(type="filepath", label="Result GIF")
-    run.click(interpolate, inputs=[img_a, img_b, fps, exp], outputs=[gif_out])
-    demo.queue(concurrency_count=1, max_size=8)
 
-if __name__ == "__main__":
-    demo.launch()
+# --- Paths ---
+DEMO_DIR = Path("demo")                         # Folder with input frames
+FRAME1 = DEMO_DIR / "frame1.png"                # First image
+FRAME2 = DEMO_DIR / "frame2.png"                # Second image
+RETRO_GIF = DEMO_DIR / "demo.gif"               # Final result
+
+# --- Interpolation function ---
+def interpolate(img_a_path, img_b_path):
+    """
+    Run interpolation pipeline using existing shell scripts and commands.
+    Expects two input image paths. Outputs interpolated GIF.
+    """
+    try:
+        # Step 1: Run your inference script to generate intermediate frames
+        subprocess.run([
+            "python3", "img_inference.py",
+            "--img", str(img_a_path), str(img_b_path)
+        ], check=True)
+
+        # Step 2: Generate optimized palette using ffmpeg
+        subprocess.run([
+            "ffmpeg", "-r", "14", "-f", "image2", "-i", "output/img%d.png",
+            "-vf", "palettegen=stats_mode=single", "palette.png"
+        ], check=True)
+
+        # Step 3: Apply palette to produce final gif
+        subprocess.run([
+            "ffmpeg", "-r", "14", "-f", "image2", "-i", "output/img%d.png",
+            "-i", "palette.png", "-lavfi", "paletteuse", str(RETRO_GIF)
+        ], check=True)
+
+        return str(RETRO_GIF)
+    except subprocess.CalledProcessError:
+        raise gr.Error("Interpolation failed. Please check script and inputs.")
+
+# --- Demo reset: returns static frames ---
+def reset_demo():
+    return str(FRAME1), str(FRAME2)
+
+# --- Gradio UI ---
+with gr.Blocks(title="RIFE Image Interpolation") as demo:
+    with gr.Tab("Demo"):
+        gr.Markdown("### Interpolate between two images")
+        img_a = gr.Image(type="filepath", value=str(FRAME1), label="Image A")
+        img_b = gr.Image(type="filepath", value=str(FRAME2), label="Image B")
+        run = gr.Button("Interpolate")
+        reset = gr.Button("Reset")
+
+        result = gr.Image(type="filepath", label="Interpolated GIF")
+        run.click(interpolate, inputs=[img_a, img_b], outputs=result)
+        reset.click(reset_demo, outputs=[img_a, img_b])
+
+demo.launch()