import gradio as gr

import assets.installation_checker as installation_checker
import assets.themes.loadThemes as loadThemes
from assets.i18n.i18n import I18nAuto
from rvc.lib.tools.prerequisites_download import prerequisites_download_pipeline
from tabs.plugins import plugins_core
from tabs.tts.tts import tts_tab
from tabs.workflow.workflow import workflow_tab

plugins_core.check_new_folders()

# Run prerequisites
prerequisites_download_pipeline(
    pretraineds_v1_f0=False,
    pretraineds_v1_nof0=False,
    pretraineds_v2_f0=True,
    pretraineds_v2_nof0=False,
    models=True,
    voices=True,
)

# Initialize i18n

i18n = I18nAuto()

# Check installation

installation_checker.check_installation()

# Start Flask server if enabled

my_applio = loadThemes.load_theme() or "ParityError/Interstellar"

# Define Gradio interface
with gr.Blocks(theme=my_applio, title="TTS Playground", css="footer{display:none !important}") as app:
    gr.Markdown("# Text-to-Speech Playground")
    gr.Markdown(i18n("Enter a page URL, click fetch and then synthesize"))
    with gr.Tab(i18n("Workflow")):
        workflow_tab()
    with gr.Tab(i18n("TTS")):
        tts_tab()