Update app.py

app.py

@@ -6,13 +6,10 @@ import argparse
 import insightface
 import onnxruntime
 import gradio as gr
-from tqdm import tqdm
-from moviepy.editor import VideoFileClip
-
 from face_swapper import Inswapper, paste_to_whole
 from face_analyser import analyse_face
 from face_enhancer import load_face_enhancer_model, get_available_enhancer_names
-from utils import merge_img_sequence_from_ref
+import tempfile
 
 # ------------------------------ ARGS ------------------------------
 parser = argparse.ArgumentParser(description="Face Swapper (Multi-target, Male+Female Sources)")
@@ -59,14 +56,11 @@ def swap_on_frame(frame_bgr, analysed_source_male, analysed_source_female, enhan
         gender = analysed_face.get("gender", 1)  # 1 = male, 0 = female
         src_face = None
 
-        # Có cả nam + nữ → swap theo giới tính
         if analysed_source_male is not None and analysed_source_female is not None:
             src_face = analysed_source_male if gender == 1 else analysed_source_female
-        # Chỉ có nam → chỉ swap nam
         elif analysed_source_male is not None and analysed_source_female is None:
             if gender == 1:
                 src_face = analysed_source_male
-        # Chỉ có nữ → chỉ swap nữ
         elif analysed_source_female is not None and analysed_source_male is None:
             if gender == 0:
                 src_face = analysed_source_female
@@ -96,23 +90,22 @@ def swap_on_frame(frame_bgr, analysed_source_male, analysed_source_female, enhan
             model, runner = load_face_enhancer_model(name=enhancer_name, device=device)
             frame_bgr = runner(frame_bgr, model)
         except Exception as e:
-            print(f"[Enhancer] Lỗi khi chạy {enhancer_name}: {e}")
+            print(f"[Enhancer] Error while running {enhancer_name}: {e}")
 
     return frame_bgr
 
-
 # ------------------------------ PROCESS ------------------------------
 def swap_faces(target_files, male_file, female_file, enhancer_name="NONE"):
     start_time = time.time()
 
     analysed_source_male, analysed_source_female = None, None
 
-    # Nguồn nam
+    # Source male
     if male_file is not None:
         male_source_path = male_file.name
         analysed_source_male = analyse_face(cv2.imread(male_source_path), FACE_ANALYSER)
 
-    # Nguồn nữ
+    # Source female
     if female_file is not None:
         female_source_path = female_file.name
         analysed_source_female = analyse_face(cv2.imread(female_source_path), FACE_ANALYSER)
@@ -120,7 +113,7 @@ def swap_faces(target_files, male_file, female_file, enhancer_name="NONE"):
     if analysed_source_male is None and analysed_source_female is None:
         raise ValueError("❌ Cần ít nhất 1 khuôn mặt nguồn (Nam hoặc Nữ).")
 
-    out_images, out_videos = [], []
+    output_files = []
 
     for f in target_files:
         target_path = f.name
@@ -130,18 +123,15 @@ def swap_faces(target_files, male_file, female_file, enhancer_name="NONE"):
         if ext in [".jpg", ".jpeg", ".png"]:
             frame_bgr = cv2.imread(target_path)
             out_frame = swap_on_frame(frame_bgr, analysed_source_male, analysed_source_female, enhancer_name)
-            out_images.append(out_frame[:, :, ::-1])  # RGB cho Gradio
-            out_videos=f">>> File ok"
-
-        # -------------------- VIDEO --------------------
-        elif ext in [".mp4", ".avi", ".mov"]:
-            out_videos=f">>> File errored"
-
 
+            # Save the output image to a temporary file
+            with tempfile.NamedTemporaryFile(suffix=ext, delete=False) as temp_file:
+                output_path = temp_file.name
+                cv2.imwrite(output_path, out_frame[:, :, ::-1])  # Convert BGR to RGB before saving
+                output_files.append(output_path)
 
     print(f"✔ Hoàn tất tất cả trong {time.time() - start_time:.2f}s")
-    return out_images, out_videos
-
+    return output_files
 
 # ------------------------------ UI ------------------------------
 with gr.Blocks() as demo:
@@ -149,27 +139,24 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            target_input = gr.Files(label="Files đích (ảnh/video)", file_types=[".jpg", ".png", ".mp4"])
+            target_input = gr.Files(label="Files đích (ảnh)", file_types=[".jpg", ".png"])
             male_input = gr.File(label="File nguồn Nam (ảnh)", file_types=[".jpg", ".png"])
             female_input = gr.File(label="File nguồn Nữ (ảnh)", file_types=[".jpg", ".png"])
             enhancer = gr.Dropdown(ENHANCER_CHOICES, label="Face Enhancer", value="NONE")
             run_btn = gr.Button("✨ Swap")
 
         with gr.Column():
-            output_gallery = gr.Files(label="Kết quả ảnh")
-            output_videos = gr.TextArea(label="Kết quả")
+            output_files = gr.Files(label="Kết quả ảnh")
 
     def run_wrapper(target_files, male_file, female_file, enhancer_name):
-        out_imgs, out_vids = swap_faces(target_files, male_file, female_file, enhancer_name)
-        imgs = [img for img in out_imgs if img is not None]
-        vids = [None]
-        return imgs,vids
+        out_files = swap_faces(target_files, male_file, female_file, enhancer_name)
+        return out_files
 
     run_btn.click(
         fn=run_wrapper,
         inputs=[target_input, male_input, female_input, enhancer],
-        outputs=[output_gallery, output_videos],
+        outputs=[output_files],
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()