TrixyZ/trainer/components/nlp.py

from dash import html, dcc
import dash_bootstrap_components as dbc

def NLPLayout():
    return html.Div([
        html.H2("NLP Management"),
        dbc.Button("Start NLP Training", id="start-training-nlp", color="primary", className="mb-3"),
        dbc.Toast(
            id="nlp-training-status",
            header="Training Status",
            is_open=False,
            dismissable=True,
            icon="info",
        ),
        html.H3("Upload NLP Data"),
        dcc.Upload(
            id="upload-nlp-data",
            children=html.Div([
                'Drag and Drop or ',
                html.A('Select Files')
            ]),
            style={
                'width': '100%',
                'height': '60px',
                'lineHeight': '60px',
                'borderWidth': '1px',
                'borderStyle': 'dashed',
                'borderRadius': '5px',
                'textAlign': 'center',
                'margin': '10px'
            },
            multiple=True
        ),
        html.Div(id="nlp-data-upload-status"),
        dbc.Button("Refresh NLP Data", id="refresh-nlp-data", color="secondary", className="mt-3"),
        html.Div(id="nlp-data-display"),
        dbc.Button("Refresh NLP Model Status", id="refresh-nlp-model-status", color="secondary", className="mt-3"),
        html.Div(id="nlp-model-status"),
    ])