Spaces:
Running
Running
| import os | |
| import shutil | |
| from huggingface_hub import HfApi | |
| def deploy(): | |
| hf_token = os.environ.get("HF_TOKEN") | |
| hf_space = os.environ.get("HF_SPACE") | |
| if not hf_token or not hf_space: | |
| print("HF_TOKEN or HF_SPACE environment variables are not set. Skipping deployment.") | |
| return | |
| print(f"Starting deployment to Hugging Face Space: {hf_space}") | |
| # 1. Copy Dockerfile.hf to Dockerfile | |
| if os.path.exists("Dockerfile.hf"): | |
| shutil.copy("Dockerfile.hf", "Dockerfile") | |
| print("Copied Dockerfile.hf to Dockerfile") | |
| else: | |
| print("Warning: Dockerfile.hf not found.") | |
| # 2. Modify README.md in place (prepend frontmatter) | |
| readme_path = "README.md" | |
| if os.path.exists(readme_path): | |
| with open(readme_path, "r", encoding="utf-8") as f: | |
| content = f.read() | |
| # Check if frontmatter already exists to avoid prepending multiple times | |
| frontmatter = ( | |
| "---\n" | |
| "title: CustomerCore API\n" | |
| "emoji: 🚀\n" | |
| "colorFrom: blue\n" | |
| "colorTo: indigo\n" | |
| "sdk: docker\n" | |
| "app_port: 7860\n" | |
| "pinned: false\n" | |
| "---\n" | |
| ) | |
| if not content.startswith("---"): | |
| new_content = frontmatter + content | |
| with open(readme_path, "w", encoding="utf-8") as f: | |
| f.write(new_content) | |
| print("Prepended Hugging Face metadata to README.md") | |
| else: | |
| print("README.md already has frontmatter. Skipping modification.") | |
| else: | |
| print("Warning: README.md not found.") | |
| # 3. Upload to Hugging Face Spaces | |
| api = HfApi() | |
| ignore_patterns = [ | |
| ".git/*", | |
| ".git", | |
| ".venv/*", | |
| ".venv", | |
| "__pycache__/*", | |
| "__pycache__", | |
| "*.pyc", | |
| ".pytest_cache/*", | |
| ".pytest_cache", | |
| ".ruff_cache/*", | |
| ".ruff_cache", | |
| "mlruns/*", | |
| "mlruns", | |
| "mlruns.db", | |
| "*.duckdb", | |
| "*.db", | |
| "logs/*", | |
| "logs", | |
| "*.log", | |
| "scratch/*", | |
| "scratch", | |
| "customercore_v*.docx", | |
| "customercore_v*.js", | |
| ".env", | |
| ".env.*", | |
| "BUILD_GUIDE.md", | |
| "CustomerCore_Progress_Summary.md" | |
| ] | |
| print("Uploading folder contents to HF Space...") | |
| try: | |
| api.upload_folder( | |
| folder_path=".", | |
| repo_id=hf_space, | |
| repo_type="space", | |
| token=hf_token, | |
| ignore_patterns=ignore_patterns | |
| ) | |
| print("Successfully deployed to Hugging Face Spaces!") | |
| except Exception as e: | |
| print(f"Error uploading to Hugging Face: {e}") | |
| raise e | |
| if __name__ == "__main__": | |
| deploy() | |