Spaces:
Sleeping
Sleeping
| import os | |
| import shutil | |
| from huggingface_hub import hf_hub_download, HfApi | |
| REPO_ID = "devorah-ai-2016/BeeCareCore" | |
| SYNC_FILES = [ | |
| "data/clients.xlsx", | |
| "data/counseling_logs.xlsx", | |
| "data/case_management.xlsx", | |
| "data/needs_assessment.xlsx", | |
| "data/care_plan_alerts.xlsx" | |
| ] | |
| def get_hf_token(): | |
| # 1. Check environment variable (set on HF spaces) | |
| token = os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN") | |
| if token: | |
| return token.strip() | |
| # 2. Check local token files (for local testing) | |
| token_candidates = [ | |
| os.path.join(os.path.dirname(os.path.abspath(__file__)), "huggingfacetoken.txt"), | |
| os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "huggingfacetoken.txt"), | |
| os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "..", "huggingfacetoken.txt"), | |
| "/Users/devorah/connect/Beecarewebsite/huggingfacetoken.txt", | |
| os.path.expanduser("~/.cache/huggingface/token") | |
| ] | |
| for path in token_candidates: | |
| if os.path.exists(path): | |
| try: | |
| with open(path, "r") as f: | |
| t = f.read().strip() | |
| if t: | |
| return t | |
| except Exception: | |
| pass | |
| return None | |
| def download_all(): | |
| token = get_hf_token() | |
| if not token: | |
| print("HF_TOKEN not found, skipping download.") | |
| return False | |
| success = True | |
| for relative_path in SYNC_FILES: | |
| # Determine local path | |
| local_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), relative_path) | |
| os.makedirs(os.path.dirname(local_path), exist_ok=True) | |
| try: | |
| downloaded_path = hf_hub_download( | |
| repo_id=REPO_ID, | |
| filename=relative_path, | |
| repo_type="space", | |
| token=token | |
| ) | |
| shutil.copy(downloaded_path, local_path) | |
| print(f"Downloaded {relative_path} to {local_path}") | |
| except Exception as e: | |
| print(f"Failed to download {relative_path}: {e}") | |
| success = False | |
| return success | |
| def upload_all(): | |
| token = get_hf_token() | |
| if not token: | |
| print("HF_TOKEN not found, skipping upload.") | |
| return False | |
| success = True | |
| api = HfApi() | |
| for relative_path in SYNC_FILES: | |
| local_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), relative_path) | |
| if not os.path.exists(local_path): | |
| print(f"Local file {local_path} does not exist. Skipping upload.") | |
| continue | |
| try: | |
| api.upload_file( | |
| path_or_fileobj=local_path, | |
| path_in_repo=relative_path, | |
| repo_id=REPO_ID, | |
| repo_type="space", | |
| token=token | |
| ) | |
| print(f"Uploaded {relative_path} to {REPO_ID}") | |
| except Exception as e: | |
| print(f"Failed to upload {relative_path}: {e}") | |
| success = False | |
| return success | |