Update app.py

app.py

@@ -1,9 +1,6 @@
 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
@@ -14,7 +11,6 @@ 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
 
 # -----------------------------
@@ -22,11 +18,6 @@ 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)
@@ -36,43 +27,6 @@ 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
 # -----------------------------
@@ -436,22 +390,6 @@ 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")),
@@ -480,7 +418,7 @@ THEME = gr.themes.Soft(
 
 CSS = """
 .gradio-container{
-  max-width: min(96vw, 1500px) !important;
+  max-width: 1040px !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;
 }
@@ -547,7 +485,6 @@ 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(
@@ -599,12 +536,11 @@ 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_with_count,
+        fn=run_enhancement,
         inputs=[source, mic_audio, file_audio, model_key],
-        outputs=[out_noisy, out_enh, img_noisy, img_enh, status, usage_md],
+        outputs=[out_noisy, out_enh, img_noisy, img_enh, status],
         api_name="enhance",
     )
-    demo.load(fn=usage_markdown_text, outputs=usage_md)
 
 if __name__ == "__main__":
     demo.queue(max_size=32).launch()