Update TTS/tts.py

TTS/tts.py

@@ -1,14 +1,15 @@
-from transformers import VitsModel, AutoTokenizer
-import torch
+from TTS.api import TTS
 
+# تحميل موديل XTTS
+tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=True)
 
-model = VitsModel.from_pretrained("facebook/mms-tts-ara")
-tokenizer = AutoTokenizer.from_pretrained("facebook/mms-tts-ara")
+def generate_voice(text, speaker_wav_path, language="ar"):
+    output_path = "output.wav"
+    tts.tts_to_file(
+        text=text,
+        file_path=output_path,
+        speaker_wav=speaker_wav_path,
+        language=language
+    )
+    return output_path
 
-def text_to_speech(text):
-    inputs = tokenizer(text, return_tensors="pt")
-    with torch.no_grad():
-        output = model(**inputs)
-    waveform = output.waveform.squeeze().numpy()
-    sampling_rate = model.config.sampling_rate
-    return waveform, sampling_rate