Update app.py

app.py

@@ -1,65 +1,59 @@
-# app.py
-
 import subprocess
 from pathlib import Path
 import gradio as gr
 
-# --- Static demo image paths ---
-FRAME1 = Path("demo/frame1.jpg")  # First image
-FRAME2 = Path("demo/frame2.jpg")  # Second image
-RESULT_GIF = Path("output/retro.gif")  # Expected result path
+# --- Paths ---
+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")
 
 def interpolate(img_a_path, img_b_path):
-    """
-    Run the interpolation pipeline from the Space root:
-    1. Run img_inference.py to generate intermediate frames
-    2. Create color palette with ffmpeg
-    3. Build final interpolated GIF with palette
-    """
     try:
-        # Step 1: Run the inference script
+        # Step 1: Run inference
         subprocess.run([
             "python3", "inference_img.py",
             "--img", str(img_a_path), str(img_b_path)
         ], check=True)
 
-        # Step 2: Generate optimized palette
+        # Step 2: Generate color palette
         subprocess.run([
-            "ffmpeg", "-r", "14", "-f", "image2", "-i", "output/img%d.png",
-            "-vf", "palettegen=stats_mode=single", "palette.png"
+            "ffmpeg", "-r", "14", "-f", "image2", "-i", f"{TARGET_DIR}img%d.png",
+            "-vf", "palettegen=stats_mode=single", f"{TMP_PREFIX}palette.png"
         ], check=True)
 
-        # Step 3: Create final GIF with palette
+        # Step 3: Build final GIF
         subprocess.run([
-            "ffmpeg", "-r", "14", "-f", "image2", "-i", "output/img%d.png",
-            "-i", "palette.png", "-lavfi", "paletteuse", str(RESULT_GIF)
+            "ffmpeg", "-r", "14", "-f", "image2", "-i", f"{TARGET_DIR}img%d.png",
+            "-i", f"{TMP_PREFIX}palette.png", "-lavfi", "paletteuse", str(RESULT_GIF)
         ], check=True)
 
         return str(RESULT_GIF)
 
     except subprocess.CalledProcessError:
-        raise gr.Error("Interpolation failed — check your script and paths.")
+        raise gr.Error("Interpolation failed — check 'inference_img.py' and output files.")
 
 def reset_demo():
-    """
-    Return paths to static demo images.
-    """
     return str(FRAME1), str(FRAME2)
 
-# --- Gradio Interface ---
-with gr.Blocks(title="RIFE Image Interpolation") as demo:
+with gr.Blocks(title="RIFE Interpolation") as demo:
     with gr.Tab("Demo"):
-        gr.Markdown("### Interpolate between two images → Retro GIF")
-
+        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])
 
+    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)
+
 demo.launch()