Update app.py

app.py

@@ -1,6 +1,9 @@
 import io
+import json
 import math
+import os
 import tempfile
+from contextlib import contextmanager
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Dict, Optional, Tuple
@@ -11,6 +14,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 import onnxruntime as ort
 import soundfile as sf
+from filelock import FileLock
 from PIL import Image
 
 # -----------------------------
@@ -18,6 +22,11 @@ from PIL import Image
 # -----------------------------
 MAX_SECONDS = 10.0
 ONNX_DIR = Path("./onnx")
+COUNTER_DIR = Path("./data") if Path("./data").exists() else Path("./")
+COUNTER_PATH = COUNTER_DIR / "dpdfnet_usage_counter.json"
+TMP_PATH = COUNTER_DIR / "dpdfnet_usage_counter.json.tmp"
+LOCK_PATH = str(COUNTER_PATH) + ".lock"
+print(COUNTER_PATH)
 
 
 @dataclass(frozen=True)
@@ -27,6 +36,43 @@ class ModelSpec:
     onnx_path: str
 
 
+@contextmanager
+def _maybe_lock():
+    with FileLock(LOCK_PATH):
+        yield
+
+
+def _read_count() -> int:
+    try:
+        if not COUNTER_PATH.exists():
+            return 0
+        with COUNTER_PATH.open("r", encoding="utf-8") as f:
+            data = json.load(f)
+        return int(data.get("enhance_runs", 0))
+    except Exception:
+        return 0
+
+
+def _atomic_write_json(obj: dict) -> None:
+    COUNTER_DIR.mkdir(parents=True, exist_ok=True)
+    with TMP_PATH.open("w", encoding="utf-8") as f:
+        json.dump(obj, f, separators=(",", ":"))
+        f.flush()
+        os.fsync(f.fileno())
+    os.replace(TMP_PATH, COUNTER_PATH)
+
+
+def increment_and_get_count() -> int:
+    with _maybe_lock():
+        new_value = _read_count() + 1
+        _atomic_write_json({"enhance_runs": new_value})
+        return new_value
+
+
+def usage_markdown_text() -> str:
+    return f"**Total enhancements run:** {_read_count():,}"
+
+
 # -----------------------------
 # Model discovery and metadata
 # -----------------------------
@@ -390,6 +436,22 @@ def run_enhancement(
     return noisy_out, enh_out, noisy_img, enh_img, status
 
 
+def run_enhancement_with_count(
+    source: str,
+    mic_path: Optional[str],
+    file_path: Optional[str],
+    model_key: str,
+):
+    noisy_out, enh_out, noisy_img, enh_img, status = run_enhancement(
+        source=source,
+        mic_path=mic_path,
+        file_path=file_path,
+        model_key=model_key,
+    )
+    total = increment_and_get_count()
+    return noisy_out, enh_out, noisy_img, enh_img, status, f"**Total enhancements run:** {total:,}"
+
+
 def set_source_visibility(source: str):
     return (
         gr.update(visible=(source == "Microphone")),
@@ -418,7 +480,7 @@ THEME = gr.themes.Soft(
 
 CSS = """
 .gradio-container{
-  max-width: 1040px !important;
+  max-width: min(96vw, 1500px) !important;
   margin: 0 auto !important;
   font-family: Arial, ui-sans-serif, system-ui, -apple-system, Segoe UI, Roboto, Helvetica Neue, Noto Sans, Liberation Sans, sans-serif !important;
 }
@@ -485,6 +547,7 @@ with gr.Blocks(theme=THEME, css=CSS, title="DPDFNet Speech Enhancement") as demo
         </div>
         """
     )
+    usage_md = gr.Markdown(usage_markdown_text())
 
     with gr.Row():
         model_key = gr.Dropdown(
@@ -536,11 +599,12 @@ with gr.Blocks(theme=THEME, css=CSS, title="DPDFNet Speech Enhancement") as demo
 
     source.change(fn=set_source_visibility, inputs=source, outputs=[mic_audio, file_audio])
     run_btn.click(
-        fn=run_enhancement,
+        fn=run_enhancement_with_count,
         inputs=[source, mic_audio, file_audio, model_key],
-        outputs=[out_noisy, out_enh, img_noisy, img_enh, status],
+        outputs=[out_noisy, out_enh, img_noisy, img_enh, status, usage_md],
         api_name="enhance",
     )
+    demo.load(fn=usage_markdown_text, outputs=usage_md)
 
 if __name__ == "__main__":
     demo.queue(max_size=32).launch()