Update app.py

app.py

@@ -1,98 +1,182 @@
 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
+# =============================
 API_KEY = "nGHjs7oK8jp7OvxZ5dVZdY6JEf3DVwRF"
 
-# ==========================================================
-# 🔐 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
-# ==========================================================
-
+# نموذج vocoder الافتراضي 22.05 kHz حسب tts_arabic
 SAMPLE_RATE = 22050
 
+
 def generate_tts(
-    text,
-    speaker,
-    pace,
-    denoise,
-    volume,
-    vowelizer,
-    model_id,
-    vocoder_id,
+    request: gr.Request,   # أول بارامتر: كائن الطلب لمعرفة الـ headers
+    text: str,
+    speaker: str,
+    pace: float,
+    denoise: float,
+    volume: float,
+    vowelizer: str,
+    model_id: str,
+    vocoder_id: str,
 ):
+    """
+    الدالة الأساسية لتحويل النص إلى صوت.
+    تُستخدم من واجهة Gradio ومن أي استدعاء API خارجي.
+    """
+
+    # =============================
+    # 🔐 التحقق من الـ API KEY
+    # =============================
+    headers = request.headers or {}
+    referer = headers.get("referer", "") or ""
+    host = headers.get("host", "") or ""
+    api_key = headers.get("x-api-key")
+
+    # نعتبر الطلب "خارجي" لو الـ referer لا يحتوي على نفس الـ host
+    # (الواجهة من المتصفح يكون الـ referer فيها https://<space>.hf.space)
+    is_external = host not in referer
+
+    if is_external:
+        # أي استدعاء خارجي لازم يكون معه x-api-key صحيح
+        if api_key != API_KEY:
+            # gr.Error تُرجع خطأ واضح في JSON و في الواجهة
+            raise gr.Error("Invalid or missing API Key")
+
+    # =============================
+    # ✅ التحقق من صحة النص والمدخلات
+    # =============================
     text = (text or "").strip()
     if not text:
         return None, "❌ الرجاء إدخال نص عربي."
 
-    vowelizer_arg = None if vowelizer == "بدون تشكيل" else vowelizer
+    try:
+        pace = float(pace)
+        denoise = float(denoise)
+        volume = float(volume)
+    except ValueError:
+        return None, "❌ قيم pace / denoise / volume غير صحيحة."
+
+    if vowelizer == "بدون تشكيل":
+        vowelizer_arg = None
+    else:
+        vowelizer_arg = vowelizer  # مثل: "shakkelha" أو "catt_eo"
 
     try:
+        # =============================
+        # 🧠 استدعاء مكتبة tts_arabic
+        # =============================
         wave = arabic_tts(
             text,
-            speaker=int(speaker),
-            pace=float(pace),
-            denoise=float(denoise),
-            volume=float(volume),
-            play=False,
-            vowelizer=vowelizer_arg,
-            model_id=model_id,
-            vocoder_id=vocoder_id,
-            cuda=None,
-            save_to=None,
+            speaker=int(speaker),      # 0–3
+            pace=pace,                 # سرعة الكلام
+            denoise=denoise,           # إزالة الضوضاء في vocoder
+            volume=volume,             # مستوى الصوت (0–1)
+            play=False,                # لا نشغّل الصوت داخل السيرفر
+            pitch_mul=1.0,
+            pitch_add=0.0,
+            vowelizer=vowelizer_arg,   # موديول التشكيل أو None
+            model_id=model_id,         # fastpitch / mixer128 / mixer80
+            vocoder_id=vocoder_id,     # hifigan / vocos / vocos44
+            cuda=None,                 # CPU فقط
+            save_to=None,              # لا نحفظ ملف هنا
         )
 
+        # wave ممكن يرجع list أو numpy
         if isinstance(wave, list):
             wave = np.array(wave, dtype=np.float32)
+        elif isinstance(wave, np.ndarray):
+            wave = wave.astype(np.float32)
+        else:
+            wave = np.array(wave, dtype=np.float32)
+
+        # تأكد أن الصوت داخل المدى [-1, 1]
+        max_abs = np.max(np.abs(wave)) if wave.size > 0 else 1.0
+        if max_abs > 1.0:
+            wave = wave / max_abs
 
         return (SAMPLE_RATE, wave), "�� تم توليد الصوت بنجاح."
 
     except Exception as e:
+        # طباعة الخطأ في اللوجز على Hugging Face ليسهل التشخيص
+        print("TTS ERROR:", repr(e))
         return None, f"❌ حدث خطأ أثناء التوليد: {e}"
 
 
-# ==========================================================
+# =============================
 # 🎛️ واجهة Gradio
-# ==========================================================
+# =============================
 
 demo = gr.Interface(
     fn=generate_tts,
     inputs=[
-        gr.Textbox(label="النص العربي", lines=4),
-        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="مستوى الصوت"),
-        gr.Dropdown(["بدون تشكيل", "shakkelha", "catt_eo"], value="بدون تشكيل", label="التشكيل"),
-        gr.Radio(["fastpitch", "mixer128", "mixer80"], value="fastpitch", label="موديل النص→ميل"),
-        gr.Radio(["hifigan", "vocos", "vocos44"], value="hifigan", label="Vocoder"),
+        gr.Textbox(
+            label="النص العربي",
+            lines=4,
+            placeholder="اكتب هنا الجملة أو الفقرة التي تريد تحويلها إلى صوت...",
+        ),
+        gr.Dropdown(
+            choices=["0", "1", "2", "3"],
+            value="1",
+            label="المتحدث (Speaker ID)",
+            info="جرّب أرقام مختلفة لاختيار صوت مختلف.",
+        ),
+        gr.Slider(
+            minimum=0.6,
+            maximum=1.4,
+            value=1.0,
+            step=0.05,
+            label="سرعة الكلام (pace)",
+        ),
+        gr.Slider(
+            minimum=0.0,
+            maximum=0.02,
+            value=0.005,
+            step=0.001,
+            label="إزالة الضوضاء (denoise)",
+        ),
+        gr.Slider(
+            minimum=0.4,
+            maximum=1.0,
+            value=0.9,
+            step=0.05,
+            label="مستوى الصوت (volume)",
+        ),
+        gr.Dropdown(
+            choices=["بدون تشكيل", "shakkelha", "catt_eo"],
+            value="بدون تشكيل",
+            label="تشكيل تلقائي للنص (Vowelizer)",
+            info="اختَر 'shakkelha' لتحسين النطق إذا كان النص غير مشكّل (أبطأ قليلاً).",
+        ),
+        gr.Radio(
+            choices=["fastpitch", "mixer128", "mixer80"],
+            value="fastpitch",
+            label="موديل Text→Mel (model_id)",
+        ),
+        gr.Radio(
+            choices=["hifigan", "vocos", "vocos44"],
+            value="hifigan",
+            label="Vocoder (vocoder_id)",
+        ),
     ],
     outputs=[
-        gr.Audio(type="numpy", label="الصوت الناتج"),
-        gr.Textbox(label="الحالة"),
+        gr.Audio(
+            label="الصوت الناتج",
+            type="numpy",  # Gradio يرجّع (sample_rate, data)
+        ),
+        gr.Textbox(
+            label="الحالة",
+            interactive=False,
+        ),
     ],
     title="Arabic TTS (ONNX / CPU)",
-    description="واجهة تحويل النص العربي إلى كلام تعمل على CPU، مع حماية API Key.",
+    description=(
+        "نموذج tts_arabic لتحويل النص العربي إلى كلام على CPU.\n"
+        "الـAPI محمية بـ API Key عبر الهيدر x-api-key."
+    ),
 )
 
-# 🎯 إضافة Middleware للـAPI
-demo.app.add_middleware(APIKeyMiddleware)
-
 if __name__ == "__main__":
     demo.launch()