Update app.py

app.py

@@ -1,29 +1,30 @@
 import gradio as gr
 import numpy as np
 from fastapi import Request, HTTPException
+from starlette.middleware.base import BaseHTTPMiddleware
 from tts_arabic import tts as arabic_tts
 
-# ==========================================
-# 🔐 API KEY PROTECTION
-# ==========================================
 API_KEY = "nGHjs7oK8jp7OvxZ5dVZdY6JEf3DVwRF"
 
-async def verify_api_key(request: Request):
-    """
-    This function runs BEFORE every prediction API call.
-    It blocks API requests without the correct key.
-    """
-    # Check if this is an API call (not UI)
-    if "api" in str(request.url):
-        key = request.headers.get("x-api-key")
-        if key != API_KEY:
-            raise HTTPException(status_code=401, detail="Invalid or missing API Key.")
-    return
-
-
-# ==========================================
-# 🎙️ TTS FUNCTION
-# ==========================================
+# ==========================================================
+# 🔐 MIDDLEWARE لحماية API
+# ==========================================================
+
+class APIKeyMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        # فقط API calls وليس الواجهة
+        url_path = str(request.url)
+        if "/api/predict" in url_path:
+            key = request.headers.get("x-api-key")
+            if key != API_KEY:
+                return HTTPException(status_code=401, detail="Invalid or missing API Key")
+        return await call_next(request)
+
+
+# ==========================================================
+# 🎙️ TTS
+# ==========================================================
+
 SAMPLE_RATE = 22050
 
 def generate_tts(
@@ -40,7 +41,6 @@ def generate_tts(
     if not text:
         return None, "❌ الرجاء إدخال نص عربي."
 
-    # اختيار موديول التشكيل
     vowelizer_arg = None if vowelizer == "بدون تشكيل" else vowelizer
 
     try:
@@ -51,8 +51,6 @@ def generate_tts(
             denoise=float(denoise),
             volume=float(volume),
             play=False,
-            pitch_mul=1.0,
-            pitch_add=0.0,
             vowelizer=vowelizer_arg,
             model_id=model_id,
             vocoder_id=vocoder_id,
@@ -69,14 +67,15 @@ def generate_tts(
         return None, f"❌ حدث خطأ أثناء التوليد: {e}"
 
 
-# ==========================================
-# 🎛️ GRADIO INTERFACE
-# ==========================================
+# ==========================================================
+# 🎛️ واجهة Gradio
+# ==========================================================
+
 demo = gr.Interface(
     fn=generate_tts,
     inputs=[
         gr.Textbox(label="النص العربي", lines=4),
-        gr.Dropdown(choices=["0", "1", "2", "3"], value="1", label="المتحدث"),
+        gr.Dropdown(["0", "1", "2", "3"], value="1", label="المتحدث"),
         gr.Slider(0.6, 1.4, value=1.0, step=0.05, label="سرعة الكلام"),
         gr.Slider(0.0, 0.02, value=0.005, step=0.001, label="إزالة الضوضاء"),
         gr.Slider(0.4, 1.0, value=0.9, step=0.05, label="مستوى الصوت"),
@@ -85,15 +84,15 @@ demo = gr.Interface(
         gr.Radio(["hifigan", "vocos", "vocos44"], value="hifigan", label="Vocoder"),
     ],
     outputs=[
-        gr.Audio(label="الصوت الناتج", type="numpy"),
-        gr.Textbox(label="الحالة", interactive=False),
+        gr.Audio(type="numpy", label="الصوت الناتج"),
+        gr.Textbox(label="الحالة"),
     ],
     title="Arabic TTS (ONNX / CPU)",
     description="واجهة تحويل النص العربي إلى كلام تعمل على CPU، مع حماية API Key.",
 )
 
-# Attach API key verifier
-demo = gr.routes.App.create_app(demo, dependencies=[verify_api_key])
+# 🎯 إضافة Middleware للـAPI
+demo.app.add_middleware(APIKeyMiddleware)
 
 if __name__ == "__main__":
     demo.launch()