Update app.py

app.py

@@ -1,37 +1,42 @@
 import gradio as gr
 import subprocess
-import shutil
 import os
 from datetime import datetime
 from PIL import Image
 
 def run_scripts(target, source, use_face_enhancer):
-    output_images = []
-    for target_file in target :
-        target_extension = os.path.splitext(target_file.name)[-1]
-        target_name = os.path.splitext(target_file.name)[1]
+    output_images = []  # Danh sách để lưu các ảnh đã xử lý
+    for target_file in target:
+        target_extension = os.path.splitext(target_file.name)[-1]  # Lấy phần mở rộng của tệp
         timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
-        output_path1 = "output" + timestamp + target_extension
+        output_path1 = f"output_{timestamp}{target_extension}"  # Đường dẫn tệp đầu ra
 
-        cmd1 = ["python3", "run.py", "-s", source.name, "-t", target_file.name, "-o", output_path1, "--frame-processor", "face_swapper","face_enhancer",'--many-faces']
+        # Chạy lệnh subprocess để xử lý ảnh
+        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)
+
         # Mở ảnh đã được xử lý và thêm vào danh sách
         output_image = Image.open(output_path1)
         output_images.append(output_image)
-        #outputfile.append(output_path1)
 
-    return output_images
+    return output_images  # Trả về danh sách ảnh đã xử lý
 
+# Giao diện Gradio
 iface = gr.Interface(
     fn=run_scripts,
     inputs=[
-        "files",
-        "file"
+        gr.Files(label="Target Files"),  # Đầu vào là tệp tin mục tiêu (ảnh/video)
+        gr.File(label="Source File"),    # Đầu vào là tệp nguồn (ảnh)
+        gr.Checkbox(label="Use Face Enhancer")  # Tùy chọn cho face enhancer
     ],
-    outputs="image",
-    title="Face swapper",
+    outputs=gr.Images(label="Output Images"),  # Đầu ra là các ảnh đã được xử lý
+    title="Face Swapper",
     description="Upload a target image/video and a source image to swap faces.",
     live=False
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()