Update app.py

app.py

@@ -5,24 +5,25 @@ import shutil
 
 def download_and_upload(token, source_repo_id, repo_type, target_repo_id, path_in_repo=""):
     try:
-        # Use HF_TOKEN secret from environment if no token is provided
+        # Use HF_TOKEN secret from environment if input is empty
         if token.strip() == "":
             token = os.environ.get("HF_TOKEN")
-
         if not token:
-            return "❌ Token missing. Please provide one or set HF_TOKEN as a secret in Space settings."
+            return "❌ Token missing. Provide one or set HF_TOKEN in secrets."
 
-        # Login
         login(token=token)
         api = HfApi()
 
-        # Download the repo
+        # Try to create target repo (will skip if already exists)
+        api.create_repo(repo_id=target_repo_id, repo_type=repo_type, exist_ok=True)
+
+        # Download
         if repo_type == "model":
             path = snapshot_download(repo_id=source_repo_id)
         else:
             path = snapshot_download(repo_id=source_repo_id, repo_type=repo_type)
 
-        # Upload to target repo
+        # Upload
         api.upload_folder(
             folder_path=path,
             repo_id=target_repo_id,
@@ -30,15 +31,12 @@ def download_and_upload(token, source_repo_id, repo_type, target_repo_id, path_i
             path_in_repo=path_in_repo,
         )
 
-        # Auto delete after upload
-        if os.path.exists(path):
-            shutil.rmtree(path)
-
-        return f"✅ Downloaded from {source_repo_id}, uploaded to {target_repo_id}, and cleaned up local files."
+        return f"✅ Downloaded from {source_repo_id} and uploaded to {target_repo_id} (created if not exist)!"
 
     except Exception as e:
         return f"❌ Failed: {e}"
 
+
 with gr.Blocks() as demo:
     gr.Markdown("## 🤗 Auto Download & Upload HF Repo (With Auto-Cleanup)")