import gradio as gr
import sys
import os

now_dir = os.getcwd()
sys.path.append(now_dir)

# Tabs
from tabs.inference.inference import inference_tab
from tabs.train.train import train_tab
from tabs.extra.extra import extra_tab
from tabs.download.download import download_tab
from tabs.tts.tts import tts_tab
from tabs.plugins.plugins import plugins_tab

# Assets
import assets.themes.loadThemes as loadThemes
from assets.i18n.i18n import I18nAuto
import assets.installation_checker as installation_checker

i18n = I18nAuto()
installation_checker.check_installation()

my_Kanoyo = loadThemes.load_json()
if my_Kanoyo:
    pass
else:
    my_Kanoyo = "ParityError/Interstellar"

with gr.Blocks(theme=my_Kanoyo, title="Kanoyo") as Kanoyo:
    gr.Markdown("# Kanoyo")
    gr.Markdown(
        i18n(
            "Ultimate voice cloning tool, meticulously optimized for unrivaled power, modularity, and user-friendly experience."
        )
    )
    gr.Markdown(
        i18n(
            "[Оригинальный форк](https://github.com/IAHispano/Applio) — [Телега](https://t.me/kanoyotelegram)"
        )
    )
    with gr.Tab(i18n("Inference")):
        inference_tab()

    with gr.Tab(i18n("Train")):
        train_tab()

    with gr.Tab(i18n("TTS")):
        tts_tab()

    with gr.Tab(i18n("Extra")):
        extra_tab()

    with gr.Tab(i18n("Plugins")):
        plugins_tab()

    with gr.Tab(i18n("Download")):
        download_tab()

if __name__ == "__main__":
    Kanoyo.launch(
        favicon_path="assets/ICON.ico",
        share="--share" in sys.argv,
        inbrowser="--open" in sys.argv,
        server_port=6969,
    )