Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, after_this_request
 from io import BytesIO
 import base64
 import subprocess
@@ -18,24 +18,21 @@ def filter_text(text):
 def convert_text_to_speech(parrafo, model):
     parrafo_filtrado = filter_text(parrafo)
     bundle_dir = os.path.abspath(os.path.dirname(__file__))
-    print("Cargando carpeta Modelos desde:", bundle_dir)
-    random_name = '.'.join(random.choices(string.ascii_letters + string.digits, k=8)) + '.wav'
-    output_file = os.path.join(bundle_dir, random_name)
-    piper_exe = '/app/piper'
-    print("Ejecutando piper desde:", piper_exe)
+    random_name = ''.join(random.choices(string.ascii_letters + string.digits, k=8)) + '.wav'
+    output_file = os.path.join('.', random_name)
+    app.logger.info("Archivo de audio creado en: %s", output_file)
+    piper_exe = os.path.join(bundle_dir, './piper')
     
     if os.path.isfile(piper_exe):
-        comando = f'echo {parrafo_filtrado} | {piper_exe} -m {model} -f {output_file}'
-        subprocess.run(comando, shell=True, check=True)
+        comando = f'echo {parrafo_filtrado} | "{piper_exe}" -m {model} -f {output_file}'
+        subprocess.run(comando, shell=True)
         return output_file
     else:
-        return "El archivo piper no se encontró en el directorio correcto."
+        return "El archivo piper.exe no se encontró en el directorio correcto."
 
 @app.route('/')
 def index():
-    model_folder = '.'
-    bundle_dir = os.path.abspath(os.path.dirname(__file__))
-    print("Cargando carpeta Modelos desde:", bundle_dir)
+    model_folder = os.path.join('./')
     model_options = [file for file in os.listdir(model_folder) if file.endswith('.onnx')]
     return render_template('index.html', model_options=model_options)
 
@@ -45,11 +42,24 @@ def convert_text():
     model = request.form['model']
     output_file = convert_text_to_speech(text, model)
     
+    @after_this_request
+    def remove_file(response):
+        try:
+            os.remove(output_file)
+            app.logger.info("Archivo de audio eliminado: %s", output_file)
+        except Exception as error:
+            app.logger.error("Error eliminando el archivo: %s", error)
+        return response
+    
     with open(output_file, 'rb') as audio_file:
         audio_content = audio_file.read()
         
     audio_base64 = base64.b64encode(audio_content).decode('utf-8')
-    return jsonify({'audio_base64': audio_base64})
+    
+    response = jsonify({'audio_base64': audio_base64})
+    
+    return response
+
 
 if __name__ == '__main__':
-    app.run(debug=True, host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', debug=False)