Update app.py

app.py

@@ -52,7 +52,6 @@ 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)
@@ -80,11 +79,9 @@ 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"]  # duración del audio en segundos
+        audio_length = len(out["wav"]) / config.audio["output_sample_rate"]
         real_time_factor = inference_time / audio_length
 
         metrics_text = f"Tiempo de generación: {inference_time:.2f} segundos\n"
@@ -125,33 +122,6 @@ Sintetizador de voz con la voz del locutor argentino Pedro Labattaglia.
 - Presione generar voz
 """
 
-# JavaScript para limpiar los datos de autenticación
-clear_auth_js = """
-function clearAuthData() {
-    localStorage.removeItem('gradio_auth_token');
-    localStorage.removeItem('gradio_auth_expiration');
-}
-
-// Limpiar datos de autenticación cuando la página se refresca o se cierra
-window.addEventListener('beforeunload', clearAuthData);
-
-// Agregar un botón para limpiar manualmente los datos de autenticación
-document.addEventListener('DOMContentLoaded', (event) => {
-    const clearAuthButton = document.createElement('button');
-    clearAuthButton.textContent = 'Cerrar sesión';
-    clearAuthButton.style.position = 'fixed';
-    clearAuthButton.style.top = '10px';
-    clearAuthButton.style.right = '10px';
-    clearAuthButton.style.zIndex = '1000';
-    clearAuthButton.onclick = function() {
-        clearAuthData();
-        alert('Sesión cerrada. Por favor, recarga la página.');
-    };
-    document.body.appendChild(clearAuthButton);
-});
-"""
-
-# Interfaz de Gradio
 with gr.Blocks(theme=theme) as demo:
     gr.Markdown(description)
 
@@ -178,13 +148,22 @@ 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: --")
 
-    # 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>")
 
@@ -200,4 +179,4 @@ demo.css = """
 """
 
 if __name__ == "__main__":
-    demo.launch(auth=[("Pedro Labattaglia", "PL2024"), ("Invitado", "PLTTS2024")])
+    demo.launch(auth=[("Pedro Labattaglia", "PL2024"), ("Invitado", "PLTTS2024")])