Update app.py

app.py

@@ -10,8 +10,10 @@ 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
 import tempfile
-from huggingface_hub import login, hf_hub_download, upload_file, HfApi
+from huggingface_hub import login, hf_hub_download, upload_file, HfApi,upload_folder
 from datetime import datetime
+import shutil
+
 
 # --- Config ---
 REPO_ID = os.environ.get("HFPATH")
@@ -47,6 +49,19 @@ def upload_to_hf(filepath):
         repo_type="dataset"
     )
     return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{path_in_repo}"
+    # --- Upload function ---
+def upload_to_fd(path):
+    #filename = os.path.basename(filepath)
+    SUBFOLDER = datetime.now().strftime("%Y%m%d")
+    path_in_repo = f"{SUBFOLDER}"
+    upload_folder(
+        folder_path=path,
+        path_in_repo=path_in_repo,
+        repo_id=REPO_ID,
+        repo_type="dataset"
+    )
+    return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{path_in_repo}"
+# ------------------------------
 # ------------------------------ ARGS ------------------------------
 parser = argparse.ArgumentParser(description="Face Swapper (Multi-target, Male+Female Sources)")
 parser.add_argument("--out_dir", default=os.getcwd())
@@ -166,7 +181,27 @@ def swap_faces(target_files, male_file, female_file, enhancer_name="NONE"):
                 output_path = temp_file.name
                 cv2.imwrite(output_path, out_frame)  # Convert BGR to RGB before saving
                 output_files.append(output_path)
-                upload_to_hf(output_path)
+                #upload_to_hf(output_path)
+
+
+
+
+    # Tạo thư mục tạm thời
+    with tempfile.TemporaryDirectory() as temp_dir:
+        print(f"Thư mục tạm thời được tạo: {temp_dir}")
+    
+    # Lặp qua từng đường dẫn file trong danh sách
+    for source_file in output_files:
+        # Tạo đường dẫn đầy đủ cho file đích trong thư mục tạm
+        destination_file = os.path.join(temp_dir, os.path.basename(source_file))
+        
+        # Sao chép file vào thư mục tạm
+        shutil.copy(source_file, destination_file)
+        print(f"Đã sao chép {source_file} vào {destination_file}")
+
+    # Tải lên thư mục tạm lên Hugging Face
+    upload_to_fd(temp_dir)
+
     print(f"✔ Hoàn tất tất cả trong {time.time() - start_time:.2f}s")
     return output_files