Update app.py

app.py

@@ -18,6 +18,21 @@ static_folder = os.path.join(script_path, "static")
 # Ensure the static folder exists
 os.makedirs(static_folder, exist_ok=True)
 
+models = [
+    ['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-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'],
+]
+
+
 class Component:
     def __init__(self, component_type, label, options=None, value=None):
         self.component_type = component_type
@@ -143,8 +158,9 @@ def copy_lang_dict(data_folder, destination,lang):
 def compile_espeak():
     compile_command = [espeak_path, "--compile=fa"]
     cwd_path = os.path.join(static_folder,"dictsource")
+    os.makedirs(cwd_path, exist_ok=True)
     subprocess.run(compile_command, cwd=cwd_path)
-    print("Compilation done!")  # Giving feedback in the console for now
+    print("Compilation done!")  
 
 
 
@@ -167,6 +183,9 @@ with gzip.open(phonetc_data_file_path, 'at',encoding='utf-8') as f:  # 'at' for
     f.write(f'// started at : {get_iran_time_no_packages()}\n')
 
 
+import soundfile as sf
+from app_utils import tts_interface
+
 def tts(input_data):
     text = input_data.get('text')
     voice = input_data.get('voice')
@@ -186,10 +205,18 @@ def tts(input_data):
     phonemized_text = phonemes_result.stdout.strip()
 
     # Generate audio file
-    audio_cmd = f'"{espeak_path}" -v {voice} -w "{audio_file}" -f "{input_file_path}"'
-    subprocess.run(audio_cmd, shell=True)
-    
-    status = f'مدل : espeak-ng\nآوانگاشت: {phonemized_text}'
+    (sample_rate, audio_data), final_status = tts_interface(voice, text, '')
+    sf.write(
+        audio_file,
+        audio_data,
+        samplerate=sample_rate,
+        subtype="PCM_16",
+    )
+
+    for model in models:
+        if voice==model[2]:
+            model_url = model[3]
+    status = f'مدل : {model_url}\nآوانگاشت: {phonemized_text}'
 
     return {"audio": f"/static/{os.path.basename(audio_file)}",  # Return the static path
             "status": status
@@ -251,8 +278,9 @@ def listWords(input_data):
 def example_function(input_data):
     return input_data
 
+input_options = list([i[2] for i in models])
 input_components = [
-    Component("radio", "voice", options=["fa", "en", "ar"], value="fa")
+    Component("radio", "voice", options=input_options, value=input_options[7])
 ]
 
 output_components = [
@@ -328,6 +356,7 @@ def download():
 @app.route('/download/fa_dict')
 def download1():
     dict_file='fa_dict'
+    compile_espeak()
     return send_file(os.path.join(find_espeak_ng_data_folder(),dict_file ), as_attachment=True)
 if __name__ == "__main__":
     app.run(host='0.0.0.0',  port= 7860)