Update app.py

app.py

@@ -1,18 +1,27 @@
-import torch
 from fastapi import FastAPI
 from fastapi.responses import StreamingResponse
-from chatterbox import ChatterboxMultilingualTTS  # حسب التسمية في الريبو
+from transformers import AutoTokenizer, VitsModel
+import torch
+import soundfile as sf
+import io
 
 app = FastAPI()
 
-model = ChatterboxMultilingualTTS.from_pretrained("ResembleAI/chatterbox")
+# تحميل موديل MMS-TTS عربي
+tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara")
+model = VitsModel.from_pretrained("facebook/mms-tts-ara")
 
 @app.post("/tts")
-async def tts_endpoint(text: str, language: str = "ar"):
-    # يمكنك ضبط باراميترات اللغة/الصوت حسب الدوكمنتشِن
-    audio = model.tts(text, language=language)
-    # audio: numpy array or torch tensor
-    # حوّله إلى wav/ogg stream
-    def iterfile():
-        yield audio.tobytes()
-    return StreamingResponse(iterfile(), media_type="audio/wav")
+async def tts_endpoint(text: str):
+    inputs = tokenizer(text, return_tensors="pt")
+
+    with torch.no_grad():
+        output = model(**inputs).waveform
+
+    audio = output.squeeze().cpu().numpy()
+
+    buffer = io.BytesIO()
+    sf.write(buffer, audio, 16000, format="WAV")
+    buffer.seek(0)
+
+    return StreamingResponse(buffer, media_type="audio/wav")