Update app.py

app.py

@@ -1,59 +1,95 @@
+# app.py
+import os
+import uuid
 import subprocess
 from pathlib import Path
+from typing import Tuple
+
 import gradio as gr
 
-# --- Paths ---
+# Demo images
 FRAME1 = Path("demo/frame1.jpg")
 FRAME2 = Path("demo/frame2.jpg")
+
 TMP_PREFIX = "/tmp/gradio/"
 TARGET_DIR = f"{TMP_PREFIX}output/"
-RESULT_GIF = Path(f"{TARGET_DIR}retro.gif")
+EXTENSION = "gif"
+
+def enhance_image(img_a_path: str, img_b_path: str, mode: str) -> Tuple[str, str]:
+    """
+    3-step pipeline:
+    1) inference_img.py --img <A> <B> --exp=4  -> writes frames as /tmp/gradio/output/img%d.png
+    2) ffmpeg palettegen
+    3) ffmpeg paletteuse -> final GIF at a unique path
+    Returns: (gif_filepath_for_preview, gif_filepath_for_download)
+    """
+    # Ensure temp dirs exist
+    os.makedirs(TARGET_DIR, exist_ok=True)
+
+    # Unique output path per run
+    interpolated_path = f"{TARGET_DIR}{uuid.uuid4()}.{EXTENSION}"
+    palette_path = f"{TMP_PREFIX}palette.png"
 
-def interpolate(img_a_path, img_b_path):
     try:
-        # Step 1: Run inference
+        # 1) Run interpolation (frames -> TARGET_DIR/img1.png ... imgN.png)
         subprocess.run([
             "python3", "inference_img.py",
-            "--img", str(img_a_path), str(img_b_path)
+            "--img", str(img_a_path), str(img_b_path),
+            "--exp", "4"
         ], check=True)
 
-        # Step 2: Generate color palette
+        # 2) Generate palette
         subprocess.run([
-            "ffmpeg", "-r", "14", "-f", "image2", "-i", f"{TARGET_DIR}img%d.png",
-            "-vf", "palettegen=stats_mode=single", f"{TMP_PREFIX}palette.png"
+            "ffmpeg", "-y", "-r", "14", "-f", "image2",
+            "-i", f"{TARGET_DIR}img%d.png",
+            "-vf", "palettegen=stats_mode=single",
+            palette_path
         ], check=True)
 
-        # Step 3: Build final GIF
+        # 3) Use palette to produce final GIF
         subprocess.run([
-            "ffmpeg", "-r", "14", "-f", "image2", "-i", f"{TARGET_DIR}img%d.png",
-            "-i", f"{TMP_PREFIX}palette.png", "-lavfi", "paletteuse", str(RESULT_GIF)
+            "ffmpeg", "-y", "-r", "14", "-f", "image2",
+            "-i", f"{TARGET_DIR}img%d.png",
+            "-i", palette_path,
+            "-lavfi", "paletteuse",
+            interpolated_path
         ], check=True)
 
-        return str(RESULT_GIF)
+        return interpolated_path, interpolated_path
 
-    except subprocess.CalledProcessError:
-        raise gr.Error("Interpolation failed — check 'inference_img.py' and output files.")
-
-def reset_demo():
-    return str(FRAME1), str(FRAME2)
+    except subprocess.CalledProcessError as e:
+        raise gr.Error(f"Interpolation failed. {e}")
 
+# Gradio UI
 with gr.Blocks(title="RIFE Interpolation") as demo:
     with gr.Tab("Demo"):
-        gr.Markdown("### Demo")
-        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_btn = gr.Button("Interpolate")
-        reset_btn = gr.Button("Reset")
-        result_img = gr.Image(type="filepath", label="Interpolated GIF")
-        run_btn.click(interpolate, inputs=[img_a, img_b], outputs=result_img)
-        reset_btn.click(reset_demo, outputs=[img_a, img_b])
+        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")
+        enhance_mode = gr.Dropdown(choices=["default"], value="default", label="Mode")  # placeholder to match signature
+        g_btn = gr.Button("Interpolate")
+
+        output_image = gr.Image(type="filepath", label="Interpolated GIF")
+        enhance_image_path = gr.Textbox(label="Output path", interactive=False)
+
+        g_btn.click(
+            fn=enhance_image,
+            inputs=[input_imageA, input_imageB, enhance_mode],
+            outputs=[output_image, enhance_image_path],
+        )
 
     with gr.Tab("Upload your images"):
-        gr.Markdown("### Upload two images to interpolate between them")
-        user_img_a = gr.Image(type="filepath", label="Upload Image A")
-        user_img_b = gr.Image(type="filepath", label="Upload Image B")
-        run_custom = gr.Button("Interpolate")
-        user_result = gr.Image(type="filepath", label="Interpolated Result")
-        run_custom.click(interpolate, inputs=[user_img_a, user_img_b], outputs=user_result)
+        gr.Markdown("### Upload any two images")
+        user_A = gr.Image(type="filepath", label="Image A")
+        user_B = gr.Image(type="filepath", label="Image B")
+        user_mode = gr.Dropdown(choices=["default"], value="default", label="Mode")
+        user_btn = gr.Button("Interpolate")
+        user_out = gr.Image(type="filepath", label="Interpolated GIF")
+        user_path = gr.Textbox(label="Output path", interactive=False)
+        user_btn.click(
+            fn=enhance_image,
+            inputs=[user_A, user_B, user_mode],
+            outputs=[user_out, user_path],
+        )
 
 demo.launch()