Spaces:
Running
Running
import streamlit as st # type: ignore | |
import os.path | |
from collections import OrderedDict | |
from streamlit_option_menu import option_menu # type: ignore | |
# Define TITLE, TEAM_MEMBERS and PROMOTION values, in config.py. | |
import config | |
from tabs.custom_vectorizer import custom_tokenizer, custom_preprocessor | |
import os | |
from translate_app import tr | |
# Initialize a session state variable that tracks the sidebar state (either 'expanded' or 'collapsed'). | |
if 'sidebar_state' not in st.session_state: | |
st.session_state.sidebar_state = 'expanded' | |
else: | |
st.session_state.sidebar_state = 'auto' | |
st.set_page_config ( | |
page_title=config.TITLE, | |
page_icon= "assets/faviconV2.png", | |
layout="wide", | |
initial_sidebar_state=st.session_state.sidebar_state | |
) | |
# Si l'application tourne localement, session_state.Cloud == 0 | |
# Si elle tourne sur le Cloud de Hugging Face, ==1 | |
st.session_state.Cloud = 1 | |
# En fonction de la valeur de varible précédente, le data path est différent | |
if st.session_state.Cloud == 0: | |
st.session_state.DataPath = "../data" | |
st.session_state.ImagePath = "../images" | |
st.session_state.reCalcule = False | |
else: | |
st.session_state.DataPath = "data" | |
st.session_state.ImagePath = "images" | |
st.session_state.reCalcule = False | |
# Define the root folders depending on local/cloud run | |
# thisfile = os.path.abspath(__file__) | |
# if ('/' in thisfile): | |
# os.chdir(os.path.dirname(thisfile)) | |
# Nécessaire pour la version windows 11 | |
if st.session_state.Cloud == 0: | |
os.environ['PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION'] = 'python' | |
# Tabs in the ./tabs folder, imported here. | |
from tabs import intro, sentence_similarity_tab, speech2text_tab, chatbot_tab | |
with open("style.css", "r") as f: | |
style = f.read() | |
st.markdown(f"<style>{style}</style>", unsafe_allow_html=True) | |
# Add tab in this ordered dict by | |
# passing the name in the sidebar as key and the imported tab | |
# as value as follow : | |
TABS = OrderedDict( | |
[ | |
(tr(intro.sidebar_name), intro), | |
(tr(chatbot_tab.sidebar_name), chatbot_tab), | |
(tr(sentence_similarity_tab.sidebar_name), sentence_similarity_tab), | |
# (tr(speech2text_tab.sidebar_name), speech2text_tab), | |
] | |
) | |
# Utilisation du module deep_translator | |
lang_tgt = ['fr', 'en','de','es', 'it', 'nl'] | |
label_lang = ['Français','English','Deutsch','Español','Italiano','Nederlands'] | |
label_lang_en = ['French', 'English', 'German', 'Spanish', 'Italian', 'Dutch'] | |
# @st.cache_data | |
def find_lang_label(lang_sel): | |
global lang_tgt, label_lang | |
return label_lang[lang_tgt.index(lang_sel)] | |
# @st.cache_data | |
def find_lang_label_en(lang_sel): | |
global lang_tgt, label_lang_en | |
return label_lang_en[lang_tgt.index(lang_sel)] | |
def run(): | |
st.sidebar.image( | |
"assets/value_props_logo.png", | |
width=270, | |
) | |
with st.sidebar.expander(":red[**"+tr("Développez moi")+"**]"): | |
# st.markdown(f"<style>{style}</style>", unsafe_allow_html=True) | |
st.markdown(tr(""" | |
:red[Cette application vous permet de tester les futures fonctionnalités de la plateforme Value Props, et plus particulièrement le «Sales Coaching». | |
Amusez vous bien !] | |
""")) | |
with st.sidebar: | |
tab_name = option_menu(None, list(TABS.keys()), | |
# icons=['house', 'bi-binoculars', 'bi bi-graph-up', 'bi-chat-right-text','bi-book', 'bi-body-text'], menu_icon="cast", default_index=0, | |
icons=['house', 'binoculars', 'graph-up', 'search','book', 'chat-right-text','controller'], menu_icon="cast", default_index=0, | |
styles={"container": {"padding": "0!important","background-color": "#10b8dd", "border-radius": "0!important"}, | |
"nav-link": {"font-size": "1rem", "text-align": "left", "margin":"0em", "padding": "0em", | |
"padding-left": "0.2em", "--hover-color": "#eee", "font-weight": "400", | |
"font-family": "Source Sans Pro, sans-serif"} | |
}) | |
# tab_name = st.sidebar.radio("", list(TABS.keys()), 0) | |
st.sidebar.markdown("---") | |
st.sidebar.markdown(f"## {tr(config.PROMOTION)}") | |
st.sidebar.markdown("### "+tr("Auteur:")) | |
for member in config.TEAM_MEMBERS: | |
st.sidebar.markdown(member.sidebar_markdown(), unsafe_allow_html=True) | |
with st.sidebar: | |
st.write("") | |
llm_choice = st.selectbox(tr("Modèle :"),["Mistral large","OpenAI 3.5","OpenAI 4o"], label_visibility="visible") | |
if (llm_choice == "OpenAI 3.5") : st.session_state.model = "gpt-3.5-turbo" | |
elif (llm_choice == "OpenAI 4o") : st.session_state.model = "gpt-4o" | |
else: st.session_state.model = "mistral-large-latest" | |
if (llm_choice in ["OpenAI 3.5","OpenAI 4o"]) and ('OPENAI_API_KEY' not in st.session_state): | |
# Set OpenAI API key | |
st.sidebar.subheader("OpenAI API Key") | |
openai_api_key = st.sidebar.text_input(tr("Saisissez votre Clé API OpenAI:"), type='password') | |
if openai_api_key: | |
os.environ['OPENAI_API_KEY'] = openai_api_key | |
st.session_state['OPENAI_API_KEY'] = openai_api_key | |
st.sidebar.success("OpenAI API Key set successfully.") | |
with st.sidebar: | |
l = st.selectbox("langue:",lang_tgt, format_func = find_lang_label, key="Language", label_visibility="hidden") | |
st.session_state.language_label = find_lang_label_en(l) | |
tab = TABS[tab_name] | |
tab.run() | |
if __name__ == "__main__": | |
run() | |