Update app.py

app.py

@@ -1,49 +1,75 @@
-import os
-import re
+from flask import Flask, render_template, request, jsonify, after_this_request
+from io import BytesIO
+import base64
 import subprocess
+import os
 import random
 import string
-import base64
-from flask import Flask, render_template, request, jsonify, after_this_request
+import re
 
 app = Flask(__name__)
 
-# Define the folder where files are saved (use relative path for portability)
-file_folder = 'files/'
+# Define the folder where files are saved
+file_folder = '/home/app/'
 
 # Models with specific character replacements
 models_replacements = {
     "Español México | Voz HirCoir": {
         "model_path": "es_MX-locutor-18488-epoch-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
     },
     "Español México | Kamora Neuronal": {
         "model_path": "kamora.onnx",
+        "replacements": [('\n', '')]  # Modified to remove newlines
     },
     "Español México | Claude": {
         "model_path": "es_MX-claude-14947-epoch-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
     },
     "Español México | Cortana Infinnity": {
         "model_path": "es_MX-cortana-19669-epoch-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
     },
     "Español México | TheGevy": {
         "model_path": "es_MX-gevy-10196-epoch-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
     },
     "English US | Voice": {
         "model_path": "en_US-ljspeech-high.onnx",
+        "replacements": [('(', ','), (')', ','), ('?', ','), ('¿', ','), (':', ','), ('\n', ' ')]
     }
 }
 
+def filter_text(text, replacements):
+    for replacement in replacements:
+        text = text.replace(replacement[0], replacement[1])
+    # Elimina los saltos de línea
+    text = text.replace('\n', '')
+    # Define una función de reemplazo usando una función lambda
+    replace_func = lambda m: "\\" + m.group(0)
+    # Crea una expresión regular con los caracteres que deseas reemplazar
+    regex_pattern = r'(["\'])'
+    # Realiza el reemplazo utilizando la función sub de re
+    filtered_text = re.sub(regex_pattern, replace_func, text)
+    return filtered_text
+
+
 def convert_text_to_speech(parrafo, model):
-    parrafo = parrafo.replace('\n', ' ')  # Eliminar saltos de línea
+    # Limit text to 500 characters
+    parrafo = parrafo[:10000]
+    
     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)
-        piper_exe = os.path.join(os.path.dirname(__file__), 'piper')  # Ruta relativa al archivo piper.exe
+        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} | "{piper_exe}" -m {model_path} -f {output_file}'
+            comando = f'echo {parrafo_filtrado} | "{piper_exe}" -m {model_path} -f {output_file}'
             subprocess.run(comando, shell=True)
             return output_file
         else:
@@ -54,6 +80,8 @@ def convert_text_to_speech(parrafo, model):
 @app.route('/')
 def index():
     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)
 
 @app.route('/convert', methods=['POST'])
@@ -66,6 +94,7 @@ def convert_text():
     def remove_file(response):
         try:
             os.remove(output_file)
+            app.logger.info("Audio file deleted: %s", output_file)
         except Exception as error:
             app.logger.error("Error deleting file: %s", error)
         return response