Update app.py

app.py

@@ -69,8 +69,18 @@ def text_to_audio(text):
         generated_audio = generated_audio.astype(np.float32)
         
         print(f"Forma del audio generado: {generated_audio.shape}")
+        current_length = len(generated_audio)
         
-        return (sample_rate, generated_audio)
+        if current_length > 8000:
+            # Recortar si es más largo de 2 segundos
+            print(f"Recortando audio de {current_length} a {8000} muestras")
+            final_audio = generated_audio[:8000]
+        else:
+            # Rellenar con ceros si es más corto de 2 segundos
+            print(f"Rellenando audio de {current_length} a {8000} muestras")
+            final_audio = np.zeros(8000, dtype=np.float32)
+            final_audio[:current_length] = generated_audio
+        return (final_audio, generated_audio)
     
     except Exception as e:
         print(f"Error en la generación de audio: {e}")
@@ -82,13 +92,13 @@ def text_to_audio(text):
 # Crear interfaz en Gradio
 interface = gr.Interface(
     fn=text_to_audio,
-    inputs=gr.Textbox(lines=2, placeholder="Escribe algo (ej. 'Hello world')"),
+    inputs=gr.Textbox(lines=2, placeholder="Escribe nine"),
     outputs=gr.Audio(label="Audio generado"),
     title="Demo de TTS con Tacotron2 + Generador",
     description="Convierte texto en audio usando Tacotron2 + modelo Generator entrenado.",
-    examples=[["Hello"], ["Hi there"]]
+    examples=[["nine"], ["nine"]]
 )
 
 # Lanzar aplicación
-if _name_ == "_main_":
+if __name__ == "__main__":
     interface.launch(debug=True)