import gradio as gr
import edge_tts
import asyncio
import tempfile
import os

# Obtener todas las voces disponibles
async def get_voices():
    voices = await edge_tts.list_voices()
    return {f"{v['ShortName']} - {v['Locale']} ({v['Gender']})": v['ShortName'] for v in voices}

# Función de texto a voz
async def text_to_speech(text, voice, rate, pitch):
    if not text.strip():
        return None, gr.Warning("Por favor, ingresa texto para convertir.")  # Warning si no se ingresa texto
    if not voice:
        return None, gr.Warning("Por favor, selecciona una voz.")  # Warning si no se selecciona una voz
    
    voice_short_name = voice.split(" - ")[0]
    rate_str = f"{rate:+d}%"  # Ajustar velocidad en formato adecuado
    pitch_str = f"{pitch:+d}Hz"  # Ajustar tono en formato adecuado
    communicate = edge_tts.Communicate(text, voice_short_name, rate=rate_str, pitch=pitch_str)
    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
        tmp_path = tmp_file.name
        await communicate.save(tmp_path)  # Guardar archivo temporal con el audio generado
    return tmp_path, None

# Función para la interfaz de Gradio
def tts_interface(text, voice, rate, pitch):
    audio, warning = asyncio.run(text_to_speech(text, voice, rate, pitch))  # Ejecutar función asíncrona
    return audio, warning

# Crear aplicación Gradio
import gradio as gr

async def create_demo():
    voices = await get_voices()  # Obtener lista de voces disponibles
    
    description = """
    Convierte texto en audio utilizando Microsoft Edge TTS. Ajusta la velocidad y el tono del habla: 0 es el valor predeterminado, valores positivos aumentan, valores negativos disminuyen.
    """
    
    demo = gr.Interface(
        fn=tts_interface,
        inputs=[
            gr.Textbox(label="Texto de entrada", lines=5),  # Entrada para texto
            gr.Dropdown(choices=[""] + list(voices.keys()), label="Seleccionar voz", value=""),  # Selección de voz
            gr.Slider(minimum=-50, maximum=50, value=0, label="Ajuste de velocidad (%)", step=1),  # Ajuste de velocidad
            gr.Slider(minimum=-20, maximum=20, value=0, label="Ajuste de tono (Hz)", step=1)  # Ajuste de tono
        ],
        outputs=[
            gr.Audio(label="Audio generado", type="filepath"),  # Salida del audio generado
            gr.Markdown(label="Advertencia", visible=False)  # Salida de advertencias
        ],
        title="Edge TTS Texto a Voz",
        description=description,
       # article="Experimenta el poder de Edge TTS para la conversión de texto a voz y explora nuestro avanzado Convertidor de Texto a Video para posibilidades creativas aún mayores.",
        analytics_enabled=False,
        allow_flagging=False
    )
    return demo

# Ejecutar la aplicación
if __name__ == "__main__":
    demo = asyncio.run(create_demo())
    demo.launch()