Update app.py

app.py

@@ -8,7 +8,10 @@ import io
 # Fixed input paths and output location
 FRAME1_PATH = "demo/frame1.png"
 FRAME2_PATH = "demo/frame2.png"
-OUTPUT_GIF = "/tmp/output.gif"
+TARGET_DIR = "/home/user/app/output/"
+OUTPUT_GIF = "output.gif"
+PALETTE_PATH = "palette.png"
+
 
 def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
     progress(0.1, desc="Starting inference...")
@@ -33,14 +36,32 @@ def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
     print("STDERR:", result.stderr)
     print("Exists?", os.path.exists("output/img0.png"))  # ⬅️ Add it here
 
-    # Check and display result
-    if result.returncode == 0 and os.path.exists(OUTPUT_GIF):
-        with open(OUTPUT_GIF, "rb") as f:
-            gif = Image.open(io.BytesIO(f.read()))
-        progress(1.0, desc="GIF created!")
-        return gif, "✅ Done!"
-    else:
-        return None, "❌ Inference failed or output missing"
+
+# Step 1: Generate palette
+subprocess.run([
+    "ffmpeg", "-y", "-r", "14", "-f", "image2",
+    "-i", f"{TARGET_DIR}img%d.png",
+    "-vf", "palettegen=stats_mode=single",
+    PALETTE_PATH
+], check=True)
+
+# Step 2: Generate GIF using palette
+subprocess.run([
+    "ffmpeg", "-y", "-r", "14", "-f", "image2",
+    "-i", f"{TARGET_DIR}img%d.png",
+    "-i", PALETTE_PATH,
+    "-lavfi", "paletteuse",
+    OUTPUT_GIF
+], check=True)
+
+# Step 3: Check and display result
+if os.path.exists(OUTPUT_GIF):
+    with open(OUTPUT_GIF, "rb") as f:
+        gif = Image.open(io.BytesIO(f.read()))
+    print("✅ GIF created!")
+    # You can return or display the gif here
+else:
+    print("❌ Inference failed or output missing")
 
 # UI setup
 with gr.Blocks() as demo_ui: