Update TTS_X/tts.py

TTS_X/tts.py

@@ -1,12 +1,36 @@
+import torch
 from TTS.api import TTS
+import os
 
-# تحميل موديل xtts v2 من Hugging Face مباشرة
-tts = TTS("tts_models--multilingual--multi-dataset--xtts_v2", gpu=True)
+def generate_voice(text, speaker_wav_path="audio/sample.wav", language="ar", output_path="output.wav"):
+    # تحديد الجهاز
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    print(f"🚀 الجهاز المستخدم: {device}")
+
+    # التأكد من وجود ملف الصوت للتقليد
+    if not os.path.exists(speaker_wav_path):
+        print(f"❌ ملف الصوت للتقليد غير موجود: {speaker_wav_path}")
+        return
+
+    try:
+        print("📦 جاري تحميل الموديل: xtts_v2 ...")
+        tts = TTS("tts_models/multilingual/multi-dataset/xtts_v2", gpu=(device=="cuda"))
+        print("✅ تم تحميل الموديل بنجاح")
+    except Exception as e:
+        print("❌ خطأ أثناء تحميل الموديل:")
+        print(e)
+        return
+
+    try:
+        print("🎤 جاري توليد الصوت...")
+        tts.tts_to_file(
+            text=text,
+            speaker_wav=speaker_wav_path,
+            language=language,
+            file_path=output_path
+        )
+        print(f"✅ تم توليد الملف الصوتي بنجاح: {output_path}")
+    except Exception as e:
+        print("❌ خطأ أثناء توليد الصوت:")
+        print(e)
 
-# توليد صوت باستخدام ملف صوت موجود مسبقاً
-tts.tts_to_file(
-    text="مرحباً بك في عالم الأصوات الذكية!",
-    file_path="output.wav",
-    speaker_wav="TTS_X/wed_voice.wav",
-    language="ar"
-)