Upload 2 files

app.py

@@ -60,7 +60,25 @@ LOAD_IN_4BIT = os.getenv("LOAD_IN_4BIT", "true").strip().lower() in {"1", "true"
 PRELOAD_MODEL = os.getenv("PRELOAD_MODEL", "true").strip().lower() in {"1", "true", "yes", "y"}
 SPACE_DIR = Path(__file__).resolve().parent
 METRICS_FILE = Path(os.getenv("METRICS_FILE", "tonebridge_usage_metrics.jsonl"))
+METRICS_REPO_SYNC = os.getenv("METRICS_REPO_SYNC", "false").strip().lower() in {"1", "true", "yes", "y"}
+METRICS_REPO_ID = (
+    os.getenv("METRICS_REPO_ID")
+    or os.getenv("SPACE_ID")
+    or os.getenv("HF_SPACE_ID")
+    or ""
+).strip()
+DEFAULT_METRICS_REPO_PATH = (
+    METRICS_FILE.name if METRICS_FILE.is_absolute() else str(METRICS_FILE).replace("\\", "/")
+)
+METRICS_REPO_PATH = os.getenv("METRICS_REPO_PATH", DEFAULT_METRICS_REPO_PATH).strip().lstrip("/")
+HF_METRICS_TOKEN = (
+    os.getenv("HF_METRICS_TOKEN")
+    or os.getenv("HF_TOKEN")
+    or os.getenv("HUGGING_FACE_HUB_TOKEN")
+    or ""
+).strip() or None
 METRICS_LOCK = Lock()
+metrics_sync_error: Optional[str] = None
 
 tokenizer = None
 model = None
@@ -463,6 +481,37 @@ def metrics_file_path() -> Path:
     return METRICS_FILE if METRICS_FILE.is_absolute() else SPACE_DIR / METRICS_FILE
 
 
+def sync_usage_metrics_to_repo(commit_message: str) -> None:
+    global metrics_sync_error
+    if not METRICS_REPO_SYNC:
+        return
+
+    path = metrics_file_path()
+    if not path.exists():
+        return
+    if not METRICS_REPO_ID:
+        metrics_sync_error = "Metrics repo sync is enabled, but METRICS_REPO_ID or SPACE_ID is missing."
+        return
+    if not HF_METRICS_TOKEN:
+        metrics_sync_error = "Metrics repo sync is enabled, but HF_METRICS_TOKEN or HF_TOKEN is missing."
+        return
+
+    try:
+        from huggingface_hub import upload_file
+
+        upload_file(
+            path_or_fileobj=str(path),
+            path_in_repo=METRICS_REPO_PATH or path.name,
+            repo_id=METRICS_REPO_ID,
+            repo_type="space",
+            token=HF_METRICS_TOKEN,
+            commit_message=commit_message,
+        )
+        metrics_sync_error = None
+    except Exception as exc:
+        metrics_sync_error = f"Metrics repo sync failed: {exc}"
+
+
 def read_usage_records_unlocked() -> list[dict]:
     path = metrics_file_path()
     if not path.exists():
@@ -497,6 +546,7 @@ def append_usage_record(record: dict) -> None:
     with METRICS_LOCK:
         with path.open("a", encoding="utf-8") as handle:
             handle.write(json.dumps(record, ensure_ascii=False, sort_keys=True) + "\n")
+    sync_usage_metrics_to_repo("Update ToneBridge usage metrics")
 
 
 def update_usage_evaluation(request_id: str, evaluation: str) -> Optional[dict]:
@@ -511,7 +561,9 @@ def update_usage_evaluation(request_id: str, evaluation: str) -> Optional[dict]:
                 break
         if updated_record is not None:
             write_usage_records_unlocked(records)
-        return updated_record
+    if updated_record is not None:
+        sync_usage_metrics_to_repo("Update ToneBridge feedback metrics")
+    return updated_record
 
 
 def metric_public_view(record: dict) -> dict:
@@ -773,6 +825,7 @@ def generate_correction(
         "generation_time_seconds": generation_time_seconds,
         "status": "ok" if corrected_sentence else "unparsed_or_error",
         "metrics_error": metrics_error,
+        "metrics_sync_error": metrics_sync_error,
     }
 
 
@@ -807,6 +860,7 @@ def rate_response(request_id: str, evaluation: str) -> dict:
         "ok": True,
         "request_id": request_id,
         "evaluation": normalized_evaluation,
+        "metrics_sync_error": metrics_sync_error,
         "record": metric_public_view(record),
     }
 
@@ -830,6 +884,10 @@ def usage_metrics(limit: int = 500) -> dict:
         "count": len(records),
         "returned": len(recent),
         "metrics_file": str(metrics_file_path()),
+        "metrics_repo_sync": METRICS_REPO_SYNC,
+        "metrics_repo_id": METRICS_REPO_ID,
+        "metrics_repo_path": METRICS_REPO_PATH,
+        "metrics_sync_error": metrics_sync_error,
         "records": [metric_public_view(record) for record in recent],
     }
 
@@ -994,6 +1052,10 @@ def runtime_info() -> str:
             f"MAX_INPUT_CHARS: {MAX_INPUT_CHARS}",
             f"MAX_NEW_TOKENS: {MAX_NEW_TOKENS}",
             f"METRICS_FILE: {metrics_file_path()}",
+            f"METRICS_REPO_SYNC: {METRICS_REPO_SYNC}",
+            f"METRICS_REPO_ID: {METRICS_REPO_ID or '(not configured)'}",
+            f"METRICS_REPO_PATH: {METRICS_REPO_PATH}",
+            f"METRICS_SYNC_ERROR: {metrics_sync_error or '(none)'}",
             f"VOXCPM_INFERENCE_TIMESTEPS: {VOXCPM_INFERENCE_TIMESTEPS}",
             f"VOXCPM_OUTPUT_SAMPLE_RATE: {VOXCPM_OUTPUT_SAMPLE_RATE}",
             f"VOXCPM_RETRY_BADCASE: {VOXCPM_RETRY_BADCASE}",

requirements.txt

@@ -2,6 +2,7 @@ gradio>=6.13.0
 spaces>=0.32.0
 torch>=2.5.0
 transformers>=4.56.0,<5.0.0
+huggingface_hub>=0.24.0
 accelerate>=0.30.0
 bitsandbytes>=0.43.0
 pypinyin==0.49.0