Upload setup_hf_spaces_secrets.py with huggingface_hub

setup_hf_spaces_secrets.py

@@ -0,0 +1,168 @@
+#!/usr/bin/env python3
+"""
+Setup HF Spaces Secrets via API
+Easily configure sensitive credentials for your Space
+"""
+
+import os
+import sys
+import requests
+import json
+from typing import Dict, Tuple
+
+def add_space_secret(hf_token: str, space_id: str, key: str, value: str) -> Tuple[bool, str]:
+    """
+    Add a secret to HF Space via API.
+    
+    Args:
+        hf_token: Hugging Face API token
+        space_id: Space ID (username/space-name)
+        key: Secret key name
+        value: Secret value
+    
+    Returns:
+        Tuple of (success: bool, message: str)
+    """
+    
+    headers = {"Authorization": f"Bearer {hf_token}"}
+    url = f"https://huggingface.co/api/spaces/{space_id}/secrets"
+    data = {"key": key, "value": value}
+    
+    try:
+        response = requests.post(url, json=data, headers=headers)
+        
+        if response.status_code == 200:
+            return True, f"✓ Secret '{key}' added successfully"
+        elif response.status_code == 409:
+            return False, f"⚠️  Secret '{key}' already exists (remove it first to update)"
+        else:
+            return False, f"✗ Error ({response.status_code}): {response.text[:100]}"
+    
+    except Exception as e:
+        return False, f"✗ Connection error: {str(e)}"
+
+def list_space_secrets(hf_token: str, space_id: str) -> Dict:
+    """List all secrets in a Space."""
+    headers = {"Authorization": f"Bearer {hf_token}"}
+    url = f"https://huggingface.co/api/spaces/{space_id}/secrets"
+    
+    try:
+        response = requests.get(url, headers=headers)
+        if response.status_code == 200:
+            return response.json()
+        else:
+            return {"error": f"Error ({response.status_code})"}
+    except Exception as e:
+        return {"error": str(e)}
+
+def delete_space_secret(hf_token: str, space_id: str, key: str) -> Tuple[bool, str]:
+    """Delete a secret from a Space."""
+    headers = {"Authorization": f"Bearer {hf_token}"}
+    url = f"https://huggingface.co/api/spaces/{space_id}/secrets/{key}"
+    
+    try:
+        response = requests.delete(url, headers=headers)
+        if response.status_code == 204:
+            return True, f"✓ Secret '{key}' deleted"
+        else:
+            return False, f"✗ Error ({response.status_code})"
+    except Exception as e:
+        return False, f"✗ Error: {str(e)}"
+
+def main():
+    print("\n" + "="*70)
+    print("🔐 HF Spaces Secrets Manager")
+    print("="*70)
+    
+    # Get HF token
+    hf_token = os.environ.get("HF_TOKEN")
+    if not hf_token:
+        print("\n❌ ERROR: HF_TOKEN not set")
+        print("   Set it with: export HF_TOKEN='hf_...'")
+        print("   Get your token: https://huggingface.co/settings/tokens")
+        sys.exit(1)
+    
+    # Get Space ID
+    import argparse
+    parser = argparse.ArgumentParser(description="Configure HF Spaces secrets")
+    parser.add_argument("--space", required=True, help="Space ID (username/space-name)")
+    parser.add_argument("--qwen-key", help="Qwen API key")
+    parser.add_argument("--qwen-url", help="Qwen base URL")
+    parser.add_argument("--qwen-model", help="Qwen model name")
+    parser.add_argument("--list", action="store_true", help="List all secrets")
+    parser.add_argument("--delete", help="Delete a secret by key")
+    
+    args = parser.parse_args()
+    space_id = args.space
+    
+    print(f"\n📍 Space: {space_id}")
+    print(f"🔑 Using HF Token: {hf_token[:20]}...")
+    
+    # List secrets
+    if args.list:
+        print("\n[Listing secrets...]")
+        secrets = list_space_secrets(hf_token, space_id)
+        if "error" in secrets:
+            print(f"✗ Error: {secrets['error']}")
+        else:
+            for s in secrets:
+                print(f"  • {s.get('name', 'unknown')}")
+        return
+    
+    # Delete secret
+    if args.delete:
+        print(f"\n[Deleting secret: {args.delete}]")
+        success, message = delete_space_secret(hf_token, space_id, args.delete)
+        print(f"  {message}")
+        return
+    
+    # Add secrets
+    secrets_to_add = []
+    
+    if args.qwen_key:
+        secrets_to_add.append(("QWEN_API_KEY", args.qwen_key))
+    
+    if args.qwen_url:
+        secrets_to_add.append(("QWEN_BASE_URL", args.qwen_url))
+    
+    if args.qwen_model:
+        secrets_to_add.append(("QWEN_MODEL", args.qwen_model))
+    
+    if not secrets_to_add:
+        print("\n[Interactive mode]")
+        print("\nEnter secrets (leave blank to skip):")
+        
+        qwen_key = input("  QWEN_API_KEY [sk-...]: ").strip()
+        if qwen_key:
+            secrets_to_add.append(("QWEN_API_KEY", qwen_key))
+        
+        qwen_url = input("  QWEN_BASE_URL [leave blank]: ").strip()
+        if qwen_url:
+            secrets_to_add.append(("QWEN_BASE_URL", qwen_url))
+        
+        qwen_model = input("  QWEN_MODEL [leave blank]: ").strip()
+        if qwen_model:
+            secrets_to_add.append(("QWEN_MODEL", qwen_model))
+    
+    if not secrets_to_add:
+        print("\n❌ No secrets to add")
+        return
+    
+    # Add secrets
+    print("\n[Adding secrets]")
+    success_count = 0
+    
+    for key, value in secrets_to_add:
+        success, message = add_space_secret(hf_token, space_id, key, value)
+        print(f"  {message}")
+        if success:
+            success_count += 1
+    
+    # Summary
+    print("\n" + "="*70)
+    print(f"✓ Configured {success_count}/{len(secrets_to_add)} secrets")
+    print(f"🚀 Visit: https://huggingface.co/spaces/{space_id}")
+    print("="*70 + "\n")
+
+if __name__ == "__main__":
+    main()