import streamlit as st
from utils.uploadAndExample import add_upload
from utils.config import model_dict
from utils.vulnerability_classifier import label_dict

with st.sidebar:
    # upload and example doc
    choice = st.sidebar.radio(label = 'Select the Document',
                            help = 'You can upload the document \
                            or else you can try a example document', 
                            options = ('Upload Document', 'Try Example'), 
                            horizontal = True)
    add_upload(choice)

    # Create a list of options for the dropdown
    model_options = ['Llama3.1-8B','Llama3.1-70B','Llama3.1-405B','Zephyr 7B β','Mistral-7B','Mixtral-8x7B']

    # Dropdown selectbox: model
    model_sel = st.selectbox('Select a model:', model_options)
    model_sel_name = model_dict[model_sel]

    st.session_state['model_sel_name'] = model_sel_name

with st.container():
    st.markdown("<h2 style='text-align: center;'> Vulnerability Analysis 3.1 </h2>", unsafe_allow_html=True)
    st.write(' ')