Update app.py

app.py

@@ -118,4 +118,59 @@ def generate(
         return None, f"Output file was not created: {out_path}"
 
     progress(1.0, desc="Done.")
-    return out_path, "Genera
+    return out_path, "Generation complete."
+
+
+with gr.Blocks() as demo:
+    gr.Markdown("# BFS Best Face Swap Video")
+
+    with gr.Row():
+        guide_video = gr.Video(
+            label="Guide Video",
+            sources=["upload"],
+            type="filepath",
+        )
+        face_image = gr.Image(
+            label="Reference Face Image",
+            type="pil",
+        )
+
+    prompt = gr.Textbox(
+        label="Prompt",
+        lines=4,
+        placeholder="headswap FACE ... ACTION ...",
+    )
+
+    with gr.Row():
+        duration = gr.Slider(1, 10, value=5, step=0.5, label="Duration (seconds)")
+        fps = gr.Slider(8, 30, value=24, step=1, label="FPS")
+
+    with gr.Row():
+        lora_strength = gr.Slider(0.0, 2.0, value=1.0, step=0.05, label="LoRA Strength")
+        seed = gr.Number(value=42, precision=0, label="Seed")
+
+    hf_token = gr.Textbox(
+        label="Hugging Face Token (optional)",
+        type="password",
+    )
+
+    run_btn = gr.Button("Generate", variant="primary")
+    output_video = gr.Video(label="Output Video")
+    status = gr.Textbox(label="Status", interactive=False)
+
+    run_btn.click(
+        fn=generate,
+        inputs=[
+            guide_video,
+            face_image,
+            prompt,
+            duration,
+            fps,
+            lora_strength,
+            seed,
+            hf_token,
+        ],
+        outputs=[output_video, status],
+    )
+
+demo.launch()