Spaces:
Sleeping
Sleeping
GhostGPT Deploy
Fix 404 (robust model pulls + fallback), pure English, Space-based memory backup
8701399 | """ | |
| GhostGPT Memory - Space based (Salman's order 2026-11-09: NO Google Cloud) | |
| How it works: | |
| - Primary storage = the Space's own files: chats.json + MEMORY.md (saved on every chat) | |
| - Backup storage = a PRIVATE Hugging Face Dataset in the same HF account | |
| (env: HF_TOKEN + HF_CHAT_DATASET, e.g. fitzpatrick9/misty-store-1349) | |
| save() uploads both files to the dataset, restore() downloads them on startup. | |
| This is what makes chat history survive Space restarts, like ChatGPT. | |
| - Google Cloud code was removed on Salman's order. | |
| """ | |
| import os | |
| import json | |
| BASE_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| MEMORY_FILE = os.path.join(BASE_DIR, "MEMORY.md") | |
| CHATS_FILE = os.path.join(BASE_DIR, "chats.json") | |
| HF_TOKEN = os.environ.get("HF_TOKEN", "").strip() | |
| HF_DATASET = os.environ.get("HF_CHAT_DATASET", "").strip() | |
| def _ensure_local_files(): | |
| if not os.path.exists(CHATS_FILE): | |
| with open(CHATS_FILE, "w", encoding="utf-8") as f: | |
| json.dump({}, f) | |
| if not os.path.exists(MEMORY_FILE): | |
| with open(MEMORY_FILE, "w", encoding="utf-8") as f: | |
| f.write("# GhostGPT Memory\n\n") | |
| def _hf_api(): | |
| if not HF_TOKEN or not HF_DATASET: | |
| return None | |
| try: | |
| from huggingface_hub import HfApi | |
| return HfApi(token=HF_TOKEN) | |
| except Exception as e: | |
| print(f"[memory] HF api error: {e}") | |
| return None | |
| def save(): | |
| """Upload chats.json + MEMORY.md to the private HF Dataset backup.""" | |
| api = _hf_api() | |
| if api is None: | |
| print("[memory] HF backup not configured (HF_TOKEN / HF_CHAT_DATASET missing) - local save only") | |
| return False | |
| try: | |
| for local_path, name in [(CHATS_FILE, "chats.json"), (MEMORY_FILE, "MEMORY.md")]: | |
| if os.path.exists(local_path): | |
| api.upload_file(path_or_fileobj=local_path, path_in_repo=name, | |
| repo_id=HF_DATASET, repo_type="dataset", | |
| commit_message="GhostGPT memory backup") | |
| print(f"[memory] backed up {name} -> {HF_DATASET}") | |
| return True | |
| except Exception as e: | |
| print(f"[memory] HF backup error: {e}") | |
| return False | |
| def restore(): | |
| """On startup, download chats.json + MEMORY.md from the HF Dataset backup.""" | |
| _ensure_local_files() | |
| api = _hf_api() | |
| if api is None: | |
| print("[memory] HF backup not configured - starting with local files") | |
| return False | |
| try: | |
| from huggingface_hub import hf_hub_download | |
| for name, local_path in [("chats.json", CHATS_FILE), ("MEMORY.md", MEMORY_FILE)]: | |
| try: | |
| downloaded = hf_hub_download(repo_id=HF_DATASET, filename=name, | |
| repo_type="dataset", token=HF_TOKEN, | |
| local_dir=BASE_DIR, force_download=True) | |
| print(f"[memory] restored {name} from {HF_DATASET}") | |
| except Exception as e: | |
| print(f"[memory] no backup for {name} yet ({str(e)[:100]}) - first run") | |
| _ensure_local_files() | |
| return True | |
| except Exception as e: | |
| print(f"[memory] restore error: {e}") | |
| return False | |
| if __name__ == "__main__": | |
| import sys | |
| cmd = sys.argv[1] if len(sys.argv) > 1 else "save" | |
| if cmd == "restore": | |
| restore() | |
| else: | |
| save() | |