Update app.py

app.py

@@ -1,35 +1,38 @@
-import gradio as gr
-from transformers import pipeline
-
-# بارگذاری مدل TTS فارسی با استفاده از مدل موجود در Huggingface.
-# در این مثال از مدل "soroushmehrani/persian-tts" استفاده شده است.
-tts_pipeline = pipeline("text-to-speech", model="soroushmehrani/persian-tts", framework="pt")
-
-def convert_text_to_speech(text: str):
-    """
-    این تابع متن ورودی را به صوت تبدیل می‌کند.
-    """
-    # اجرای مدل TTS برای تبدیل متن به صدا.
-    result = tts_pipeline(text)
-    
-    # بررسی نوع خروجی: اگر نتیجه به صورت دیکشنری باشد و شامل کلید 'audio' باشد.
-    if isinstance(result, dict) and "audio" in result:
-        return result["audio"]
-    # اگر خروجی یک لیست از آرایه‌های صوتی باشد، اولین عنصر را برمی‌گردانیم.
-    elif isinstance(result, list) and len(result) > 0:
-        return result[0]
-    else:
-        # در غیر این صورت، خروجی را به همان صورت برمی‌گردانیم.
-        return result
-
-# ایجاد واسط کاربری تعاملی با استفاده از Gradio
-interface = gr.Interface(
-    fn=convert_text_to_speech,
-    inputs=gr.Textbox(lines=4, placeholder="متن فارسی خود را وارد کنید...", label="متن ورودی"),
-    outputs=gr.Audio(type="auto", label="صوت خروجی"),
-    title="دموی تبدیل متن فارسی به صدا",
-    description="در این اپلیکیشن، متن فارسی وارد شده به صوت تبدیل می‌شود. لطفاً متن مورد نظر خود را وارد نمایید."
-)
-
-if __name__ == "__main__":
-    interface.launch()
+import gradio as gr
+from transformers import pipeline
+
+# بارگذاری مدل TTS فارسی با استفاده از مدل HooshvareLab/ParsiTTS
+# توجه: اگر این مدل خصوصی یا نیازمند توکن باشد، می‌توانید پارامتر token را نیز اضافه کنید.
+tts_pipeline = pipeline("text-to-speech", model="HooshvareLab/ParsiTTS", framework="pt")
+
+def convert_text_to_speech(text: str):
+    """
+    این تابع متن ورودی را به صوت تبدیل می‌کند.
+    """
+    # اجرای مدل TTS جهت تولید صوت
+    result = tts_pipeline(text)
+    
+    # بررسی خروجی مدل؛ ساختار خروجی ممکن است بر اساس مدل متفاوت باشد.
+    if isinstance(result, dict):
+        # ممکن است کلید 'audio' یا 'wav' وجود داشته باشد؛ در اینجا هر دو بررسی می‌شود.
+        if "audio" in result:
+            return result["audio"]
+        elif "wav" in result:
+            return result["wav"]
+    elif isinstance(result, list) and len(result) > 0:
+        return result[0]
+    
+    # در صورت عدم تطابق ساختار، خروجی نهایی را برمی‌گرداند.
+    return result
+
+# ایجاد واسط کاربری تعاملی با استفاده از Gradio
+interface = gr.Interface(
+    fn=convert_text_to_speech,
+    inputs=gr.Textbox(lines=4, placeholder="متن فارسی خود را وارد کنید...", label="متن ورودی"),
+    outputs=gr.Audio(type="auto", label="صوت خروجی"),
+    title="دموی تبدیل متن فارسی به صدا",
+    description="در این اپلیکیشن، متن فارسی وارد شده به صوت تبدیل می‌شود. لطفاً متن خود را وارد نمایید."
+)
+
+if __name__ == "__main__":
+    interface.launch()