"""
Copyright 2025 Balacoon

TTSLeaderboard app.
"""

import gradio as gr
from leaderboard import create_leaderboard_tab
from samples import create_samples_tab


def main():
    with gr.Blocks(css="footer {visibility: hidden}") as demo:
        gr.Markdown(
            """
            # TTSLeaderboard
            ### Objective evaluation of speech generation systems
            Evaluations are done with [speech_gen_eval](https://github.com/balacoon/speech_gen_eval),
            on [speech_gen_eval_testsets](https://huggingface.co/datasets/balacoon/speech_gen_eval_testsets).
            Results and audio are fetched from [speech_gen_baselines](https://huggingface.co/datasets/balacoon/speech_gen_baselines).
            """
        )
        
        with gr.Tabs():
            with gr.TabItem("📊 Leaderboard"):
                create_leaderboard_tab()

            with gr.TabItem("🔊 Samples"):
                create_samples_tab()
                gr.Markdown("""
                Click on the audio files to play sample generations from each model.
                """)

    demo.launch()


if __name__ == "__main__":
    main()