Update app.py

app.py

@@ -2,8 +2,9 @@ import gradio as gr
 import subprocess
 import os
 from datetime import datetime
-from huggingface_hub import login, hf_hub_download, upload_file, HfApi
-
+from huggingface_hub import login, hf_hub_download, upload_file, HfApi, upload_folder
+import tempfile
+import shutil
 # --- Config ---
 REPO_ID = os.environ.get("HFPATH")
 HF_TOKEN = os.environ.get("MAGIC")
@@ -39,11 +40,24 @@ def upload_to_hf(filepath):
         repo_type="dataset"
     )
     return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{path_in_repo}"
+def upload_to_fd(path):
+    SUBFOLDER = datetime.now().strftime("%Y%m%d")
+    upload_folder(folder_path=path,
+                  path_in_repo=SUBFOLDER,
+                  repo_id=REPO_ID,
+                  repo_type="dataset")
+    return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{SUBFOLDER}"
 
 def run_scripts(subfolder, source):
     outputfile = []
     target_files = download_from_hf(subfolder)
-    for target_file in target_files:
+    
+    
+    # Tạo thư mục tạm và lưu file đầu ra vào đó
+    with tempfile.TemporaryDirectory() as temp_dir:
+        print(f"Thư mục tạm thời được tạo: {temp_dir}")
+
+        for target_file in target_files:
         target_extension = os.path.splitext(target_file)[-1]
         #filename = os.path.splitext(target_file)[0]
         timestamp = datetime.now().strftime("%Y%m%d%H%M%S")
@@ -52,9 +66,14 @@ def run_scripts(subfolder, source):
         subprocess.run(cmd1)
         outputfile.append(output_path)
         print(output_path)
-        upload_to_hf(output_path)
-    return outputfile
+    # Sao chép file vào thư mục tạm
+       destination_file = os.path.join(temp_dir, os.path.basename(output_path))
+       shutil.copy(output_path, destination_file)
+       print(f"Đã sao chép {output_path} vào {destination_file}")
 
+        # Tải lên thư mục tạm lên Hugging Face
+        upload_to_fd(temp_dir)
+    return outputfile
 iface = gr.Interface(
     fn=run_scripts,
     inputs=[