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

# Get all available voices
async def get_voices():
    voices = await edge_tts.list_voices()
    return {f"{v['ShortName']} - {v['Locale']} ({v['Gender']})": v['ShortName'] for v in voices}

# Text-to-speech function
async def text_to_speech(text, voice, rate, pitch):
    if not text.strip():
        return None, gr.Warning("Please enter text to convert.")
    if not voice:
        return None, gr.Warning("Please select a voice.")
    
    voice_short_name = voice.split(" - ")[0]
    rate_str = f"{rate:+d}%"
    pitch_str = f"{pitch:+d}Hz"
    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)
    return tmp_path, None

# Gradio interface function
def tts_interface(text, voice, rate, pitch):
    audio, warning = asyncio.run(text_to_speech(text, voice, rate, pitch))
    return audio, warning

# Create Gradio application
import gradio as gr

async def create_demo():
    voices = await get_voices()
    
    description = """
    Convert text to speech using EchoX. Adjust speech rate and pitch: 0 is the default, and positive values increase, while negative values decrease.
    
    🤖 **Exciting News: Introducing our AzGPT AI virtual assistant!** 🤖
    
    Take Your Conversations to the Next Level with AzGPT!
    Experience the power of AzGPT, a cutting-edge AI virtual assistant designed exclusively for Azerbaijani users! Running on a massive 60 billion parameters, AzGPT is fast, efficient, and delivers smart, responsive interactions—all in Azerbaijani!

    ✨ Features: • Instant, high-quality replies in Azerbaijani
    • Powered by advanced AI with 60 billion parameters
    • Perfect for providing quick answers, conversation support, and language assistance
    • Tailored for educators, students, professionals, and anyone seeking a powerful AI in Azerbaijani
    
    Let AzGPT enhance your digital interactions, with replies that are as fast as they are accurate—built just for Azerbaijani speakers!
    
    Ready to revolutionize your chat? [Click here to try our AI Virtual Assistant now!](botintelx.netlify.app)
    """
    
    demo = gr.Interface(
        fn=tts_interface,
        inputs=[
            gr.Textbox(label="Input Text", lines=5),
            gr.Dropdown(choices=[""] + list(voices.keys()), label="Select Voice", value=""),
            gr.Slider(minimum=-50, maximum=50, value=0, label="Speech Rate Adjustment (%)", step=1),
            gr.Slider(minimum=-20, maximum=20, value=0, label="Pitch Adjustment (Hz)", step=1)
        ],
        outputs=[
            gr.Audio(label="Generated Audio", type="filepath"),
            gr.Markdown(label="Warning", visible=False)
        ],
        title="EchoX TTS ( Text-To-Speech )",
        description=description,
        article="Experience the power of EchoX TTS for text-to-speech conversion, and explore our advanced AzGPT AI Virtual Assistant for even more creative possibilities!",
        analytics_enabled=False,
        allow_flagging=False
    )
    return demo

# Run the application
if __name__ == "__main__":
    demo = asyncio.run(create_demo())
    demo.launch()