Update app.py

app.py

@@ -34,6 +34,12 @@ from src.chatterbox.mtl_tts import ChatterboxMultilingualTTS
 # ===============================
 MODEL = None
 
+# ===============================
+# MAX QUOTA (from ENV)
+# ===============================
+TTS_MAX_QUOTA = int(os.getenv("TTS_MAX_QUOTA", 10))  # default 10 requests/day
+tts_usage = 0  # simple in-memory counter for demo
+
 # ===============================
 # MODEL LOADER
 # ===============================
@@ -90,6 +96,17 @@ def health():
         "cuda_available": torch.cuda.is_available()
     }
 
+# ===============================
+# QUOTA INFO
+# ===============================
+@app.get("/quota")
+def get_quota():
+    return {
+        "used": tts_usage,
+        "limit": TTS_MAX_QUOTA,
+        "remaining": max(0, TTS_MAX_QUOTA - tts_usage)
+    }
+
 # ===============================
 # TTS INPUT SCHEMA
 # ===============================
@@ -103,6 +120,13 @@ class TTSPayload(BaseModel):
 # ===============================
 @app.post("/tts")
 def generate_tts(payload: TTSPayload):
+    global tts_usage
+    if tts_usage >= TTS_MAX_QUOTA:
+        return {
+            "error": "Quota exceeded",
+            "message": f"Daily limit of {TTS_MAX_QUOTA} TTS requests reached. Try again tomorrow."
+        }
+
     model = get_or_load_model()
 
     # Determine final text
@@ -126,21 +150,22 @@ def generate_tts(payload: TTSPayload):
         sr = model.sr
 
     # Convert numpy -> WAV bytes
-    import io
-    from scipy.io.wavfile import write as write_wav
     buf = io.BytesIO()
     write_wav(buf, sr, wav.astype(np.float32))
     buf.seek(0)
     audio_bytes = buf.read()
 
+    # Increment quota usage
+    tts_usage += 1
+
     # Return as base64
-    import base64
     return {
         "sr": sr,
-        "audio_base64": base64.b64encode(audio_bytes).decode("utf-8")
+        "audio_base64": base64.b64encode(audio_bytes).decode("utf-8"),
+        "quota_used": tts_usage,
+        "quota_limit": TTS_MAX_QUOTA
     }
 
-
 # ===============================
 # RUN: uvicorn app:app --host 0.0.0.0 --port 7860
 # ===============================