# --- Imports libs ---
import gradio as gr
import pandas as pd


# --- Imports modules ---
from modules.model_embbeding import Embedding

# --- Imports interfaces ---
from interfaces.interface_WordExplorer import interface as wordExplorer_interface
from interfaces.interface_BiasWordExplorer import interface as biasWordExplorer_interface

# --- Tool config ---
AVAILABLE_LOGS      = True                          # [True     | False]
LANGUAGE            = "spanish"                     # [spanish  | english]
EMBEDDING_SUBSET    = "fasttext"                    # [fasttext | mini]

# --- Init classes ---
embedding = Embedding(
    subset_name=EMBEDDING_SUBSET
)
labels = pd.read_json(f"language/{LANGUAGE}.json")["app"]

# --- Main App ---
INTERFACE_LIST = [
    biasWordExplorer_interface(
        embedding=embedding,
        available_logs=AVAILABLE_LOGS,
        lang=LANGUAGE),
    wordExplorer_interface(
        embedding=embedding,
        available_logs=AVAILABLE_LOGS,
        lang=LANGUAGE),
]

TAB_NAMES = [
    labels["biasWordExplorer"],
    labels["wordExplorer"],
]

iface = gr.TabbedInterface(
    interface_list=INTERFACE_LIST, 
    tab_names=TAB_NAMES
)

iface.queue(concurrency_count=8)
iface.launch(debug=False)