Update app.py

app.py

@@ -9,9 +9,10 @@ import io
 
 FRAME1_PATH = "demo/frame1.png"
 FRAME2_PATH = "demo/frame2.png"
-TARGET_DIR = "/home/user/app/output/"
-OUTPUT_GIF = "output.gif"
-PALETTE_PNG = "palette.png"
+TARGET_DIR = "output/"
+#TARGET_DIR = "/home/user/app/output/"
+OUTPUT_GIF = "output/output.gif"
+PALETTE_FULL_PATH = "/demo/palette.png"
 
 
 def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
@@ -38,7 +39,7 @@ def generate_demo_gif(exp=2, progress=gr.Progress(track_tqdm=True)):
     print("Exists?", os.path.exists("output/img0.png"))  # ⬅️ Add it here
 
 
-# Step 2: Generate palette
+# Generate palette
 subprocess.run([
     "ffmpeg", "-y", "-r", "14", "-f", "image2",
     "-i", f"{TARGET_DIR}img%d.png",
@@ -55,14 +56,13 @@ subprocess.run([
     OUTPUT_GIF
 ], check=True)
 
-# Step 3: Check and display result
-if os.path.exists(TARGET_DIR/OUTPUT_GIF):
-    with open(TARGET_DIR/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")
+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: