Update app.py

app.py

@@ -6,31 +6,11 @@ import os
 import random
 import string
 import re
-import logging
 
 app = Flask(__name__)
 
-# Define los nombres asignados a modelos específicos, en caso de no existir no se muestran
-model_names = {
-        "Español México    | Kamora Neuronal": {
-        "model_path": "es_ES-kamora-7539-high.onnx",
-        "replacements": [('\n', '. ')]
-    },
-    "Español México    | Claude": {
-        "model_path": "es_MX-claude-14947-epoch-high.onnx",
-        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
-    }
-}
-
-# Comprueba si los modelos definidos existen en la carpeta de modelos
-model_folder = '/home/app/'
-existing_models = [model_name for model_name in model_names.keys() if os.path.isfile(os.path.join(model_folder, model_names[model_name]["model_path"]))]
-
-def multiple_replace(text, replacements):
-    # Iterar sobre cada par de remplazo
-    for old, new in replacements:
-        text = text.replace(old, new)
-    return text
+# Define the folder where files are saved
+file_folder = '/home/app/'
 
 def filter_text(text, model_name):
     if model_name in model_names:
@@ -44,20 +24,18 @@ def filter_text(text, model_name):
         logging.error(f"No se encontró el modelo '{model_name}'.")
         return None
 
-
-# Define una función para convertir texto a voz
-def convert_text_to_speech(text, model_name):
-    filtered_text = filter_text(text, model_name)[:500]  # Limitar el texto a 500 caracteres
-    if filtered_text is None:
-        return None
-
+def convert_text_to_speech(parrafo, model):
+    # Limit text to 500 characters
+    parrafo = parrafo[:100000]
+    
+    parrafo_filtrado = filter_text(parrafo)
     random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=8)) + '.wav'
-    output_file = os.path.join(model_folder, random_name)
+    output_file = os.path.join(file_folder, random_name)
     app.logger.info("Audio file created at: %s", output_file)
-    piper_exe = os.path.join(model_folder, 'piper')  # Adjusted the path for piper
+    piper_exe = os.path.join(file_folder, 'piper')  # Adjusted the path for piper
     
     if os.path.isfile(piper_exe):
-        comando = f'echo {filtered_text} | "{piper_exe}" -m {model_name} -f {output_file}'
+        comando = f'echo {parrafo_filtrado} | "{piper_exe}" -m {model} -f {output_file}'
         subprocess.run(comando, shell=True)
         return output_file
     else:
@@ -66,9 +44,10 @@ def convert_text_to_speech(text, model_name):
 
 @app.route('/')
 def index():
+    model_folder = file_folder  # Adjusted the model folder to file_folder
     model_options = [file for file in os.listdir(model_folder) if file.endswith('.onnx')]
     # Log the contents of the current folder
-    app.logger.info("Contents of current folder: %s", os.listdir(model_folder))
+    app.logger.info("Contents of current folder: %s", os.listdir(file_folder))
     return render_template('index.html', model_options=model_options)
 
 @app.route('/convert', methods=['POST'])
@@ -96,4 +75,4 @@ def convert_text():
     return response
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=False)
+    app.run(host='0.0.0.0', port=7860, debug=False)