Update app.py

app.py

@@ -52,6 +52,7 @@ def predict(prompt, language, reference_audio):
         if len(prompt) < 2 or len(prompt) > 600:
             return None, "El texto debe tener entre 2 y 600 caracteres."
 
+        # Obtener los parámetros de la configuración JSON
         temperature = config.model_args.get("temperature", 0.85)
         length_penalty = config.model_args.get("length_penalty", 1.0)
         repetition_penalty = config.model_args.get("repetition_penalty", 2.0)
@@ -79,9 +80,11 @@ def predict(prompt, language, reference_audio):
         inference_time = time.time() - start_time
         
         output_path = "pedro_labattaglia_TTS.wav"
+        # Guardar el audio directamente desde el output del modelo
+        import scipy.io.wavfile as wavfile
         wavfile.write(output_path, config.audio["output_sample_rate"], out["wav"])
 
-        audio_length = len(out["wav"]) / config.audio["output_sample_rate"]
+        audio_length = len(out["wav"]) / config.audio["output_sample_rate"]  # duración del audio en segundos
         real_time_factor = inference_time / audio_length
 
         metrics_text = f"Tiempo de generación: {inference_time:.2f} segundos\n"
@@ -122,6 +125,7 @@ Sintetizador de voz con la voz del locutor argentino Pedro Labattaglia.
 - Presione generar voz
 """
 
+# Interfaz de Gradio
 with gr.Blocks(theme=theme) as demo:
     gr.Markdown(description)
 
@@ -148,24 +152,24 @@ with gr.Blocks(theme=theme) as demo:
             generated_audio = gr.Audio(label="Audio generado", interactive=False)
             metrics_output = gr.Textbox(label="Métricas", value="Tiempo de generación: -- segundos\nFactor de tiempo real: --")
 
+    # Botón de logout
+    logout_button = gr.LogoutButton(label="Logout")
+
+    # Configuración del botón para generar voz
     generate_button.click(
         predict,
         inputs=[input_text, language_selector, reference_audio],
         outputs=[generated_audio, metrics_output]
     )
 
-    # Botón de logout
-    logout_button = gr.LogoutButton(logout_msg="Sesión cerrada correctamente", label="Cerrar sesión")
-    
-    # Script para manejar el cierre o refresco de la pestaña
-    clear_auth_js = """
-    window.addEventListener("beforeunload", function(event) {
-        fetch('/logout', {method: 'POST', credentials: 'same-origin'});
-    });
-    """
-    
-    # Agregar el JavaScript para limpiar los datos de autenticación
-    gr.HTML("<script>" + clear_auth_js + "</script>")
+    # Agregar el JavaScript para ejecutar el logout al refrescar o cerrar la pestaña
+    gr.HTML("""
+        <script>
+            window.addEventListener("beforeunload", function (event) {
+                fetch('/logout', { method: 'POST' });
+            });
+        </script>
+    """)
 
 # Estilos CSS personalizados
 demo.css = """