Spaces:
Running
Running
File size: 3,707 Bytes
815d041 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 |
#!/usr/bin/env python3
"""
Safe Upload Script for Hugging Face Spaces
==========================================
This script safely uploads only necessary files to Hugging Face Spaces,
excluding sensitive files like .env, __pycache__, etc.
Usage:
python safe_upload.py your_username/your_space_name
Requirements:
pip install huggingface_hub
"""
import os
import sys
from pathlib import Path
from huggingface_hub import HfApi
# Files to upload (whitelist approach - safer)
ALLOWED_FILES = [
"AI_Talk_Gradio.py",
"requirements.txt",
".gitignore",
"README_Deploy.md"
]
# Files/folders to NEVER upload (blacklist - extra safety)
BLOCKED_PATTERNS = [
".env",
"__pycache__",
".gradio",
"*.pyc",
"*.pyo",
"*.pyd",
"*.pem",
"*.key",
"*.crt",
"flagged",
".vscode",
".idea"
]
def is_safe_file(file_path):
"""Check if file is safe to upload"""
file_name = os.path.basename(file_path)
# Check whitelist
if file_name not in ALLOWED_FILES:
return False
# Extra check - ensure no blocked patterns
for pattern in BLOCKED_PATTERNS:
if pattern in file_path.lower():
return False
return True
def safe_upload_to_hf(repo_id):
"""Safely upload files to Hugging Face Space"""
api = HfApi()
print(f"🚀 Starting safe upload to {repo_id}")
print("📋 Files to be uploaded:")
current_dir = Path(".")
files_to_upload = []
# Collect safe files
for file_name in ALLOWED_FILES:
file_path = current_dir / file_name
if file_path.exists() and is_safe_file(str(file_path)):
files_to_upload.append(file_path)
print(f" ✅ {file_name}")
else:
print(f" ⚠️ {file_name} (not found or blocked)")
if not files_to_upload:
print("❌ No safe files found to upload!")
return False
print(f"\n📤 Uploading {len(files_to_upload)} files...")
try:
for file_path in files_to_upload:
print(f" Uploading {file_path.name}...")
api.upload_file(
path_or_fileobj=str(file_path),
path_in_repo=file_path.name,
repo_id=repo_id,
repo_type="space"
)
print(f" ✅ {file_path.name} uploaded successfully")
print(f"\n🎉 Upload completed successfully!")
print(f"🌐 Your Space: https://huggingface.co/spaces/{repo_id}")
print("\n📝 Next steps:")
print("1. Go to your Space settings")
print("2. Add API keys as Repository secrets:")
print(" - GROQ_API_KEY")
print(" - GOOGLE_API_KEY")
print("3. Your Space will auto-deploy!")
return True
except Exception as e:
print(f"❌ Upload failed: {e}")
return False
def main():
if len(sys.argv) != 2:
print("Usage: python safe_upload.py username/space_name")
print("Example: python safe_upload.py ducnguyen1978/AI_Game")
sys.exit(1)
repo_id = sys.argv[1]
# Validate repo_id format
if "/" not in repo_id:
print("❌ Invalid repo_id. Use format: username/space_name")
sys.exit(1)
print("🔒 Safe Upload for Hugging Face Spaces")
print("=" * 40)
print(f"Target Space: {repo_id}")
print()
success = safe_upload_to_hf(repo_id)
if not success:
sys.exit(1)
if __name__ == "__main__":
main() |