Update app.py

app.py

@@ -1,91 +1,32 @@
 import gradio as gr
 import subprocess
+import shutil
 import os
 from datetime import datetime
-from huggingface_hub import login, upload_file
 
-# --- Config ---
-REPO_ID = "mr2along/Picture"
-HF_TOKEN = os.environ.get("MAGIC")
-if not HF_TOKEN:
-    print("⚠️ Lưu ý: HF_TOKEN không được tìm thấy trong biến môi trường 'MAGIC'.")
-
-# --- Hàm upload file ---
-def upload_to_huggingface(filepath):
-    try:
-        login(HF_TOKEN)
-        filename = os.path.basename(filepath)
-        upload_file(
-            path_or_fileobj=filepath,
-            path_in_repo=filename,
-            repo_id=REPO_ID,
-            repo_type="dataset"
-        )
-        return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{filename}"
-    except Exception as e:
-        return f"❌ Lỗi upload: {str(e)}"
-
-# --- Hàm xử lý chính ---
-def run_scripts(target_files, source_file):
-    output_files, logs = [], ""
-    if not target_files or not source_file:
-        return [], "⚠️ Vui lòng chọn đủ source và target!"
-
-    for tgt in target_files:
-        tgt_path = tgt.name if hasattr(tgt, "name") else str(tgt)
-        tgt_ext = os.path.splitext(tgt_path)[-1]
-        ts = datetime.now().strftime("%Y%m%d%H%M%S")
-        out_path = f"output_{ts}{tgt_ext}"
-
-        cmd = [
-            "python3", "run.py",
-            "-s", source_file.name if hasattr(source_file, "name") else str(source_file),
-            "-t", tgt_path,
-            "-o", out_path,
-            "--frame-processor", "face_swapper", "face_enhancer",
-            "--many-faces"
-        ]
-
-        try:
-            res = subprocess.run(cmd, capture_output=True, text=True)
-            if res.returncode != 0:
-                logs += f"❌ Lỗi xử lý {tgt_path}:\n{res.stderr}\n"
-                continue
-            output_files.append(out_path)
-            url = upload_to_huggingface(out_path)
-            logs += f"✅ {out_path} → [Uploaded]({url})\n"
-        except Exception as e:
-            logs += f"❌ Exception khi xử lý {tgt_path}: {str(e)}\n"
-
-    return output_files, logs
-
-# --- UI Gradio v4.44.1 ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    demo.title = "Face Swapper Tool"
-
-    gr.Markdown("# 🤖 Face Swapper")
-    gr.Markdown("Upload ảnh/video cần swap và ảnh khuôn mặt để ghép")
-
-    with gr.Row():
-        target_input = gr.File(
-            file_types=[("image",), ("video",)],
-            file_count="multiple",
-            label="Ảnh/Video Target"
-        )
-        source_input = gr.File(
-            file_types=[("image",)],
-            label="Ảnh Source"
-        )
-
-    run_btn = gr.Button("🔄 Chạy Face Swap")
-    result_gallery = gr.Files(label="📂 File kết quả")
-    logs_output = gr.Markdown()  # Không có label
-
-    run_btn.click(fn=run_scripts, inputs=[target_input, source_input], outputs=[result_gallery, logs_output])
-
-demo.launch(
-    share=True,
-    server_name="0.0.0.0",
-    server_port=7860,
-    ssr_mode=False
-)
+def run_scripts(target, source, use_face_enhancer):
+    outputfile=[]
+    for target_file in target :
+        target_extension = os.path.splitext(target_file.name)[-1]
+        target_name = os.path.splitext(target_file.name)[1]
+        timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
+        output_path1 = "output" + timestamp + target_extension
+        cmd1 = ["python3", "run.py", "-s", source.name, "-t", target_file.name, "-o", output_path1, "--frame-processor", "face_swapper","face_enhancer",'--many-faces']
+        subprocess.run(cmd1)
+        outputfile.append(output_path1)
+        print(output_path1)
+    return outputfile
+
+iface = gr.Interface(
+    fn=run_scripts,
+    inputs=[
+        "files",
+        "file"
+    ],
+    outputs="files",
+    title="Face swapper",
+    description="Upload a target image/video and a source image to swap faces.",
+    live=False
+)
+
+iface.launch()