Upload upload_to_hf.py

upload_to_hf.py

@@ -1,47 +1,108 @@
 #!/usr/bin/env python3
 """
-AIQalam HuggingFace Uploader
-راهنما: توکن HuggingFace خودت رو جایگزین HF_TOKEN کن و اسکریپت رو اجرا کن
+Qalam HuggingFace Upload Script
+Upload Qalam agent configuration to HuggingFace Hub
 """
 
 import os
-
-# توکن HuggingFace — از huggingface.co/settings/tokens بگیر
-HF_TOKEN = "YOUR_HF_TOKEN_HERE"
-REPO_ID = "Qalam/AIQalam"
-
-files_to_upload = [
-    "AIQalam_agent.json",
-    "AIQalam_agent_config.md",
-]
-
-if HF_TOKEN == "YOUR_HF_TOKEN_HERE":
-    print("⚠️  لطفاً HF_TOKEN رو در خط 8 اسکریپت وارد کن!")
-    exit(1)
+import argparse
+from pathlib import Path
 
 try:
-    from huggingface_hub import HfApi
+    from huggingface_hub import HfApi, login
 except ImportError:
-    print("نصب huggingface_hub...")
+    print("Installing huggingface_hub...")
     os.system("pip install huggingface_hub")
-    from huggingface_hub import HfApi
+    from huggingface_hub import HfApi, login
 
-api = HfApi(token=HF_TOKEN)
 
-print(f"📤 آپلود به {REPO_ID}...")
+def get_token():
+    """Get token from environment or user input"""
+    token = os.environ.get("HF_TOKEN")
+    if not token:
+        token = input("Enter your HuggingFace token: ").strip()
+    return token
 
-for fname in files_to_upload:
-    if os.path.exists(fname):
-        print(f"  ⬆️  {fname}")
-        api.upload_file(
-            path_or_fileobj=fname,
-            path_in_repo=fname,
-            repo_id=REPO_ID,
-            repo_type="space",
-        )
-        print(f"  ✅ {fname} آپلود شد!")
+
+def upload_to_hf(token: str, repo_id: str = "Qalam/AIQalam", local_dir: str = None):
+    """Upload Qalam files to HuggingFace"""
+    
+    api = HfApi()
+    
+    print("Logging in to HuggingFace...")
+    login(token=token)
+    
+    if local_dir is None:
+        local_dir = Path(__file__).parent.absolute()
     else:
-        print(f"  ⚠️  {fname} پیدا نشد!")
+        local_dir = Path(local_dir).absolute()
+    
+    print(f"Local directory: {local_dir}")
+    
+    files_to_upload = [
+        "README.md",
+        "Qalam_agent_v3.json",
+        "Qalam_agent_v2.json",
+        "Qalam_agent.json",
+        "Qalam_Al-Mutlaq.json",
+        "Qalam_agent_config.md",
+        "upload_to_hf.py",
+    ]
+    
+    existing_files = []
+    for f in files_to_upload:
+        path = local_dir / f
+        if path.exists():
+            existing_files.append(str(path))
+            print(f"  Found: {f}")
+    
+    if not existing_files:
+        print("No files found to upload!")
+        return
+    
+    print(f"\nUploading {len(existing_files)} files to https://huggingface.co/{repo_id}")
+    
+    try:
+        api.create_repo(
+            repo_id=repo_id,
+            repo_type="model",
+            exist_ok=True,
+            private=False
+        )
+        print(f"Repository ready: {repo_id}")
+    except Exception as e:
+        print(f"Error: {e}")
+        return
+    
+    for file_path in existing_files:
+        file_name = Path(file_path).name
+        print(f"  Uploading {file_name}...")
+        try:
+            api.upload_file(
+                path_or_fileobj=file_path,
+                path_in_repo=file_name,
+                repo_id=repo_id,
+                repo_type="model",
+                commit_message=f"Upload {file_name}"
+            )
+            print(f"  Done: {file_name}")
+        except Exception as e:
+            print(f"  Error uploading {file_name}: {e}")
+    
+    print(f"\nDone! View at: https://huggingface.co/{repo_id}")
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Upload Qalam to HuggingFace")
+    parser.add_argument("--token", "-t", help="HuggingFace token (or set HF_TOKEN env var)")
+    parser.add_argument("--repo", "-r", default="Qalam/AIQalam", help="Repository ID")
+    parser.add_argument("--dir", "-d", help="Local directory")
+    
+    args = parser.parse_args()
+    
+    token = args.token or get_token()
+    upload_to_hf(token, args.repo, args.dir)
+
 
-print("\n🎉 همه فایل‌ها آپلود شدند!")
-print(f"🔗 https://huggingface.co/{REPO_ID}")
+if __name__ == "__main__":
+    main()