import gradio as gr
from NeuralNetwork_UI import NeuralNetwork_UI
from RandomForest_UI import RandomForest_UI
from SVM_UI import SVM_UI
from KNN_UI import KNN_UI


def default_ui() -> gr.Blocks:
    with gr.Blocks() as interface:
        gr.Markdown(
            """
        For this project we will be using [this](https://www.kaggle.com/datasets/valakhorasani/gym-members-exercise-dataset/discussion/543798) dataset.
        """
        )

    return interface


def launch_UI():
    nn_ui = NeuralNetwork_UI()
    rf_ui = RandomForest_UI()
    svm_ui = SVM_UI()
    knn_ui = KNN_UI()

    ui = gr.TabbedInterface(
        [
            default_ui(),
            nn_ui.get_interface(),
            rf_ui.get_interface(),
            svm_ui.get_interface(),
            knn_ui.get_interface(),
        ],
        ["Menu", "Neural Network", "Random Forest", "SVM", "KNN"],
    )
    ui.launch()


if __name__ == "__main__":
    launch_UI()