Update app_utils.py

app_utils.py

@@ -1,6 +1,5 @@
 import os
 import json
-import gradio as gr
 import shutil
 import subprocess
 import requests
@@ -12,67 +11,21 @@ import numpy as np
 
 
 models = [
-    ['mms fa','https://huggingface.co/willwade/mms-tts-multilingual-models-onnx/resolve/main/fas'],
-    ['coqui-vits-female1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-female1-vits-coqui'],
-    ['coqui-vits-male1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-male1-vits-coqui'],
-    ['coqui-vits-male-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-male-coqui-vits'],
-    ['coqui-vits-female-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-female-coqui-vits'],
-    ['coqui-vits-female-GPTInformal-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-GPTInformal-coqui-vits'],
-    ['coqui-vits-male-SmartGitiCorp','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-SmartGitiCorp-coqui-vits'],
-    ['vits-piper-fa-gyro-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-gyro-medium.tar.bz2'],
-    ['piper-fa-amir-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-amir-medium.tar.bz2'],
-    ['vits-mimic3-fa-haaniye_low','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-mimic3-fa-haaniye_low.tar.bz2'],
-    ['vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium.tar.bz2'],
+    ['mms fa','https://huggingface.co/willwade/mms-tts-multilingual-models-onnx/resolve/main/fas',"🌠 راد",'https://huggingface.co/facebook/mms-tts-fas'],
+    ['coqui-vits-female1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-female1-vits-coqui',"🌺 نگار",'https://huggingface.co/Kamtera/persian-tts-female1-vits'],
+    ['coqui-vits-male1-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/persian-tts-male1-vits-coqui',"🌟 آرش",'https://huggingface.co/Kamtera/persian-tts-male1-vits'],
+    ['coqui-vits-male-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-male-coqui-vits',"🦁 کیان",'https://huggingface.co/Kamtera/persian-tts-male-vits'],
+    ['coqui-vits-female-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-female-coqui-vits',"🌷 مهتاب",'https://huggingface.co/Kamtera/persian-tts-female-vits'],
+    ['coqui-vits-female-GPTInformal-karim23657','https://huggingface.co/karim23657/persian-tts-vits/tree/main/female-GPTInformal-coqui-vits',"🌼 شیوا",'https://huggingface.co/karim23657/persian-tts-female-GPTInformal-Persian-vits'],
+    ['coqui-vits-male-SmartGitiCorp','https://huggingface.co/karim23657/persian-tts-vits/tree/main/male-SmartGitiCorp-coqui-vits',"🚀 بهمن",'https://huggingface.co/SmartGitiCorp/persian_tts_vits'],
+    ['vits-piper-fa-ganji','https://huggingface.co/karim23657/persian-tts-vits/tree/main/vits-piper-fa-ganji',"🚀 برنا",'https://huggingface.co/SadeghK/persian-text-to-speech'],
+    ['vits-piper-fa-ganji-adabi','https://huggingface.co/karim23657/persian-tts-vits/tree/main/vits-piper-fa-ganji-adabi',"🚀 برنا-1",'https://huggingface.co/SadeghK/persian-text-to-speech'],
+    ['vits-piper-fa-gyro-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-gyro-medium.tar.bz2',"💧 نیما",'https://huggingface.co/gyroing/Persian-Piper-Model-gyro'],
+    ['piper-fa-amir-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_IR-amir-medium.tar.bz2',"⚡️ آریا",'https://huggingface.co/SadeghK/persian-text-to-speech'],
+    ['vits-mimic3-fa-haaniye_low','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-mimic3-fa-haaniye_low.tar.bz2',"🌹 ریما",'https://github.com/MycroftAI/mimic3'],
+    ['vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium','https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium.tar.bz2',"🌠 پیام",'https://huggingface.co/mah92/persian-english-piper-tts-model'],
 ]
-models_n =[
-  [
-    "mms fa", 
-    "🌠 راد"
-  ],
-  [
-    "coqui-vits-female1-karim23657", 
-    "🌺 نگار"
-  ],
-  [
-    "coqui-vits-male1-karim23657", 
-    "🌟 آرش"
-  ],
-  [
-    "coqui-vits-male-karim23657", 
-    "🦁 کیان"
-  ],
-  [
-    "coqui-vits-female-karim23657", 
-    "🌷 مهتاب"
-  ],
-  [
-    "coqui-vits-female-GPTInformal-karim23657", 
-    "🌼 شیوا"
-  ],
-  [
-    "coqui-vits-male-SmartGitiCorp", 
-    "🚀 بهمن"
-  ],
-  [
-    "vits-piper-fa-gyro-medium", 
-    "💧 نیما"
-  ],
-  [
-    "piper-fa-amir-medium", 
-    "⚡️ آریا"
-  ],
-  [
-    "vits-mimic3-fa-haaniye_low", 
-    "🌹 ریما"
-  ],
-  [
-    "vits-piper-fa_en-rezahedayatfar-ibrahimwalk-medium", 
-    "🌠 پیام"
-  ]
-]
-for i,m in enumerate(models):
-    models_n[i][0]=m[1]
-dropdown_choices = list([i[1] for i in models_n])
+
 def download_and_extract_model(url, destination):
     """Download and extract the model files."""
     print(f"Downloading from URL: {url}")
@@ -272,9 +225,9 @@ def generate_audio(text, model_info):
         if not os.path.exists(model_dir):
             print(f"Model directory doesn't exist, downloading {model_info}...")
             os.makedirs(model_dir, exist_ok=True)
-            for i in models_n:
-                if model_info == i[1]:
-                    model_url=i[0]
+            for i in models:
+                if model_info == i[2]:
+                    model_url=i[1]
             download_and_extract_model(model_url, model_dir)
         
         print(f"Contents of {model_dir}:")
@@ -414,48 +367,3 @@ def tts_interface(selected_model, text, status_output):
         error_msg = str(e)
         return None, f"Error: {error_msg}"
 
-# Gradio Interface
-with gr.Blocks() as app:
-    gr.Markdown("# Sherpa-ONNX متن به گفتار")
-    with gr.Row():
-        with gr.Column():
-            model_dropdown = gr.Radio(
-                choices=dropdown_choices,
-                label="مدل",
-                value=dropdown_choices[0] if dropdown_choices else None
-            )
-            text_input = gr.Textbox(
-                label="متن",
-                placeholder="متن را وارد کنید ...",
-                lines=3
-            )
-            
-            with gr.Row():
-                generate_btn = gr.Button("بگو")
-                stop_btn = gr.Button("توقف")
-            
-        with gr.Column():
-            audio_output = gr.Audio(
-                label="گفتار",
-                type="numpy"
-            )
-            status_text = gr.Textbox(
-                label="وضعیت",
-                interactive=False
-            )
-    
-
-    # Set up event handlers
-    gen_event = generate_btn.click(
-        fn=tts_interface,
-        inputs=[model_dropdown, text_input, status_text],
-        outputs=[audio_output, status_text]
-    )
-    
-    stop_btn.click(
-        fn=None,
-        cancels=gen_event,
-        queue=False
-    )
-
-app.launch()