Update app.py

app.py

@@ -1,87 +1,65 @@
+# app.py
 import os
-from pathlib import Path
+import uuid
 import subprocess
+from pathlib import Path
 import gradio as gr
-from PIL import Image
-import io
 
-# Fixed input paths and output location
+FRAME1 = Path("demo/frame1.png")
+FRAME2 = Path("demo/frame2.png")
 
-FRAME1_PATH = "demo/frame1.png"
-FRAME2_PATH = "demo/frame2.png"
-TARGET_DIR = "/home/user/app/output/"
-#TARGET_DIR = "/home/user/app/output/"
-OUTPUT_GIF = "output/output.gif"
+TARGET_DIR = "/home/user/app/output/"     # absolute path that inference writes to
 PALETTE_PNG = "demo/palette.png"
-
-
-def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
-    progress(0.1, desc="Starting inference...")
-
-    # Delete old output if exists
-    try:
-        os.remove(OUTPUT_GIF)
-    except FileNotFoundError:
-        pass
-
-    # Build and run command
-    cmd = [
-        "python", "inference_img.py",
-        "--img", FRAME1_PATH, FRAME2_PATH,
-        "--exp", str(exp),
-        "--model", "train_log/"
-    ]
-    print("Running:", " ".join(cmd))
-    result = subprocess.run(cmd, capture_output=True, text=True)
-
-    print("STDOUT:", result.stdout)
-    print("STDERR:", result.stderr)
-    print("Exists?", os.path.exists("output/img0.png"))  # ⬅️ Add it here
-
-
-# Generate palette
-subprocess.run([
-    "ffmpeg", "-y", "-r", "14", "-f", "image2",
-    "-i", f"{TARGET_DIR}img%d.png",
-    "-vf", "palettegen=stats_mode=single",
-    PALETTE_PNG
-], check=True)
-
-#!ffmpeg -r 14 -f image2 -i output/img%d.png -vf "palettegen=stats_mode=single" /content/demo/palette.png
-# Step 2: Generate GIF using palette
-subprocess.run([
-    "ffmpeg", "-y", "-r", "14", "-f", "image2",
-    "-i", f"{TARGET_DIR}img%d.png",
-    "-i", PALETTE_PNG,
-    "-lavfi", "paletteuse",
-    OUTPUT_GIF
-], check=True)
-
-#!ffmpeg -r 14 -f image2 -i output/img%d.png -i palette.png -lavfi "paletteuse" output/retro.gif
-subprocess.run([
-    "ffmpeg", "-y", "-r", "14", "-f", "image2",
-    "-i", f"{TARGET_DIR}img%d.png",
-    "-i", PALETTE_PNG,
-    "-lavfi", "paletteuse",
-    OUTPUT_GIF
-], check=True)
-
-# UI setup
-with gr.Blocks() as demo_ui:
-    gr.Markdown("## 🎞️ Demo GIF Generator — Interpolate Two Frames")
-
-    with gr.Row():
-        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="Output GIF")
-    status = gr.Markdown()
-
-    run_btn.click(fn=generate_demo_gif, inputs=[exp], outputs=[out_gif, status])
-
-
-# Launch the app
-demo_ui.launch(ssr_mode=False)
-
+OUTPUT_GIF_DIR = "output"
+os.makedirs("demo", exist_ok=True)
+os.makedirs(TARGET_DIR, exist_ok=True)
+os.makedirs(OUTPUT_GIF_DIR, exist_ok=True)
+
+def interpolate_image(img_a_path: str, img_b_path: str) -> tuple[str, str]:
+    # 1) Run inference to generate frames into TARGET_DIR/img%d.png
+    subprocess.run([
+        "python3", "inference_img.py",
+        "--img", str(img_a_path), str(img_b_path),
+        "--exp", "4"
+    ], check=True)
+
+    # 2) Generate palette from frames
+    subprocess.run([
+        "ffmpeg", "-y", "-r", "14", "-f", "image2",
+        "-i", f"{TARGET_DIR}img%d.png",
+        "-vf", "palettegen=stats_mode=single",
+        PALETTE_PNG
+    ], check=True)
+
+    # 3) Generate final GIF using palette
+    out_gif = os.path.join(OUTPUT_GIF_DIR, f"{uuid.uuid4()}.gif")
+    subprocess.run([
+        "ffmpeg", "-y", "-r", "14", "-f", "image2",
+        "-i", f"{TARGET_DIR}img%d.png",
+        "-i", PALETTE_PNG,
+        "-lavfi", "paletteuse",
+        out_gif
+    ], check=True)
+
+    return out_gif, out_gif
+
+with gr.Blocks(title="RIFE Interpolation") as demo:
+    with gr.Tab("Demo"):
+        gr.Markdown("### Demo: Preloaded images")
+        input_imageA = gr.Image(type="filepath", value=str(FRAME1), label="Image A")
+        input_imageB = gr.Image(type="filepath", value=str(FRAME2), label="Image B")
+        run_btn = gr.Button("Interpolate")
+        result_img = gr.Image(type="filepath", label="Interpolated GIF")
+        result_path = gr.Textbox(label="Output path", interactive=False)
+        run_btn.click(interpolate_image, [input_imageA, input_imageB], [result_img, result_path])
+
+    with gr.Tab("Upload your images"):
+        gr.Markdown("### Upload any two images")
+        user_A = gr.Image(type="filepath", label="Image A")
+        user_B = gr.Image(type="filepath", label="Image B")
+        run_btn2 = gr.Button("Interpolate")
+        user_img = gr.Image(type="filepath", label="Interpolated GIF")
+        user_path = gr.Textbox(label="Output path", interactive=False)
+        run_btn2.click(interpolate_image, [user_A, user_B], [user_img, user_path])
+
+demo.launch()