Update app.py

app.py

@@ -1,7 +1,19 @@
-import os
+import gradio as gr
+import subprocess
 
-# 直接运行 main.py（如果 main.py 是主脚本）
-if os.path.exists("main.py"):
-    os.system("python main.py")
-else:
-    print("Error: main.py not found. Please check your project structure.")
+# 运行 Face Swap 代码
+def run_script(source_video, target_face):
+    cmd = f"python inference.py --source {source_video} --target {target_face}"
+    result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
+    return result.stdout
+
+# Gradio UI
+iface = gr.Interface(
+    fn=run_script,
+    inputs=["file", "file"],
+    outputs="text",
+    title="Video Face Swap",
+    description="Upload a video and a face image to swap faces."
+)
+
+iface.launch()