Update app.py

app.py

@@ -12,58 +12,50 @@ app = Flask(__name__)
 # Define the folder where files are saved
 file_folder = '/home/app/'
 
-
-import logging
-
-model_names = {
-    "es_MX-claude-14947-epoch-high.onnx": {
-        "replacements": {
-            ('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')
-        }
+# Models with specific character replacements
+models_replacements = {
+    "Español México | Kamora Neuronal": {
+        "model_path": "es_ES-kamora-7539-high.onnx",
+        "replacements": [('\n', '. ')]
     },
-    "kamora": {
-        "replacements": {
-            # Define los reemplazos específicos del modelo 2 aquí
-        }
-    },
-    # Agrega más modelos si es necesario
+    "Español México | Claude": {
+        "model_path": "es_MX-claude-14947-epoch-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
+    }
 }
 
-def filter_text(text, model_name):
-    if model_name in model_names:
-        replacements = model_names[model_name]["replacements"]
-        # Realizar reemplazos específicos del modelo
-        filtered_text = multiple_replace(text, replacements)
-        # Escapar todos los caracteres especiales dentro de las comillas
-        filtered_text = re.sub(r'(["\'])', lambda m: "\\" + m.group(0), filtered_text)
-        return filtered_text
-    else:
-        logging.error(f"No se encontró el modelo '{model_name}'.")
-        return None
-
+def filter_text(text, replacements):
+    filtered_text = re.sub(r'[^\w\s,.\(\):\u00C0-\u00FF]', '', text)
+    for replacement in replacements:
+        filtered_text = filtered_text.replace(replacement[0], replacement[1])
+    return filtered_text
 
 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(file_folder, random_name)
-    app.logger.info("Audio file created at: %s", output_file)
-    piper_exe = os.path.join(file_folder, 'piper')  # Adjusted the path for piper
-    
-    if os.path.isfile(piper_exe):
-        comando = f'echo {parrafo_filtrado} | "{piper_exe}" -m {model} -f {output_file}'
-        subprocess.run(comando, shell=True)
-        return output_file
+    model_info = models_replacements.get(model)
+    if model_info:
+        model_path = model_info.get("model_path")
+        replacements = model_info.get("replacements")
+        parrafo_filtrado = filter_text(parrafo, replacements)
+        random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=8)) + '.wav'
+        output_file = os.path.join(file_folder, random_name)
+        app.logger.info("Audio file created at: %s", output_file)
+        piper_exe = os.path.join(file_folder, 'piper')  # Adjusted the path for piper
+        
+        if os.path.isfile(piper_exe):
+            comando = f'echo {parrafo_filtrado} | "{piper_exe}" -m {model_path} -f {output_file}'
+            subprocess.run(comando, shell=True)
+            return output_file
+        else:
+            return "The piper.exe file was not found in the correct directory."
     else:
-        return "The piper.exe file was not found in the correct directory."
-
+        return "Model not found."
 
 @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')]
+    model_options = list(models_replacements.keys())
     # Log the contents of the current folder
     app.logger.info("Contents of current folder: %s", os.listdir(file_folder))
     return render_template('index.html', model_options=model_options)
@@ -93,4 +85,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)