Spaces:
Sleeping
Sleeping
| """Append-only JSONL log of user interactions (queries, follow-ups, feedback). | |
| One line per event in ``<dir>/interactions-YYYYMMDD.jsonl``. When | |
| FIXSENSE_LOG_REPO (a *private* HF dataset repo id) and HF_TOKEN are set, a background | |
| huggingface_hub.CommitScheduler pushes the folder every FIXSENSE_LOG_PUSH_MIN minutes | |
| (default 5) -- the Space's own disk is ephemeral, so this is what makes the log durable. | |
| A crash can lose at most the last interval. | |
| Each Space replica writes under its own subfolder (``path_in_repo``) so concurrent | |
| replicas never overwrite each other's files. | |
| """ | |
| from __future__ import annotations | |
| import json | |
| import logging | |
| import os | |
| import socket | |
| import threading | |
| from datetime import datetime, timezone | |
| from pathlib import Path | |
| logger = logging.getLogger(__name__) | |
| class InteractionLogger: | |
| def __init__(self, directory: str = "logs") -> None: | |
| self.dir = Path(directory) | |
| self.dir.mkdir(parents=True, exist_ok=True) | |
| self._lock = threading.Lock() | |
| self._scheduler = None | |
| repo = os.environ.get("FIXSENSE_LOG_REPO", "").strip() | |
| token = os.environ.get("HF_TOKEN", "").strip() | |
| if repo and token: | |
| try: | |
| from huggingface_hub import CommitScheduler | |
| self._scheduler = CommitScheduler( | |
| repo_id=repo, | |
| repo_type="dataset", | |
| folder_path=self.dir, | |
| path_in_repo=f"interactions/{socket.gethostname()}", | |
| every=float(os.environ.get("FIXSENSE_LOG_PUSH_MIN", "5")), | |
| private=True, | |
| token=token, | |
| ) | |
| self._lock = self._scheduler.lock # recommended: share the scheduler's lock | |
| logger.info("Interaction log will sync to private dataset %s", repo) | |
| except Exception: | |
| logger.exception("Could not start CommitScheduler; logging locally only") | |
| elif repo and not token: | |
| logger.warning("FIXSENSE_LOG_REPO is set but HF_TOKEN is not; logging locally only") | |
| def log(self, event: dict) -> None: | |
| now = datetime.now(timezone.utc) | |
| record = {"ts": now.isoformat(timespec="milliseconds"), **event} | |
| path = self.dir / f"interactions-{now:%Y%m%d}.jsonl" | |
| line = json.dumps(record, ensure_ascii=False, default=str) | |
| try: | |
| with self._lock: | |
| with open(path, "a", encoding="utf-8") as f: | |
| f.write(line + "\n") | |
| except Exception: | |
| logger.exception("Failed to write interaction log") | |
| def flush(self) -> None: | |
| """Push pending lines now (used on shutdown).""" | |
| if self._scheduler is not None: | |
| try: | |
| self._scheduler.trigger().result() | |
| except Exception: | |
| logger.exception("Final log push failed") | |