from prelude import prelude

prelude()

import gradio as gr
from app.setup import SetupTab
from app.extract import FeatureExtractionTab
from app.train import TrainTab
from app.export import ExportTab
from app.infer import InferenceTab


with gr.Blocks() as app:
    gr.Markdown("# ZeroRVC")
    gr.Markdown(
        "Run Retrieval-based Voice Conversion training and inference on HuggingFace ZeroGPU."
    )

    exp_dir = gr.Textbox(
        label="Experiment directory",
        visible=True,
        interactive=False,
    )

    setup = SetupTab()
    feature_extraction = FeatureExtractionTab()
    training = TrainTab()
    export = ExportTab()
    inferencing = InferenceTab()

    with gr.Tabs():
        with gr.Tab(label="Setup"):
            setup.ui()

        with gr.Tab(label="Feature Extraction"):
            feature_extraction.ui()

        with gr.Tab(label="Training"):
            training.ui()

        with gr.Tab(label="Download"):
            export.ui()

        with gr.Tab(label="Inference"):
            inferencing.ui()

    setup.build(exp_dir)
    feature_extraction.build(exp_dir)
    training.build(exp_dir)
    export.build(exp_dir)
    inferencing.build(exp_dir)

    app.launch()