Update chatbot_tab.py

tabs/chatbot_tab.py

@@ -82,11 +82,13 @@ selected_options6 = []
 selected_options7 = []
 selected_index8 = 0
 virulence = 1
+if 'model' in st.session_state:
+    used_model = st.session_state.model
 
 # @st.cache_data
 def init():
     global config,thread_id, context,human_message1,ai_message1,language, app, model_speech,prompt,model
-    global selected_index1, selected_index2, selected_index3, selected_options4,selected_options5,selected_options6,selected_options7, selected_index8, virulence
+    global selected_index1, selected_index2, selected_index3, selected_options4,selected_options5,selected_options6,selected_options7, selected_index8, virulence, used_model
     
     model_speech = whisper.load_model("base") 
     
@@ -100,6 +102,16 @@ def init():
         selected_options7 = []
         selected_index8 = 0
         virulence = 1
+        if 'model' in st.session_state and (st.session_state.model[:3]=="gpt") and ("OPENAI_API_KEY" in st.session_state):
+            model = ChatOpenAI(model=st.session_state.model,
+            temperature=0.8,  # Adjust creativity level
+            max_tokens=150   # Define max output token limit
+            )           
+        else: 
+            model = ChatMistralAI(model=st.session_state.model)
+        if 'model' in st.session_state:
+            used_model=st.session_state.model
+        
       
     options1 = ["Directeur Commercial", "Directeur Général", "Directeur Marketing"]
     translated_options1 = [tr(o) for o in options1]
@@ -231,7 +243,7 @@ Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
 
     st.write("")
     if ("context" in st.session_state) and ("human_message1" in st.session_state):
-        if (st.session_state.context != context) or (st.session_state.human_message1 != human_message1 ):
+        if (st.session_state.context != context) or (st.session_state.human_message1 != human_message1 ) or (used_model != st.session_state.model):
             to_init = True
         else:
             to_init = False
@@ -251,6 +263,15 @@ Attention: Ce n'est pas toi qui m'aide, c'est moi qui t'aide avec ma solution.
         st.session_state.context = context
         st.session_state.human_message1 = human_message1
         st.session_state.messages = []
+        if 'model' in st.session_state and (st.session_state.model[:3]=="gpt") and ("OPENAI_API_KEY" in st.session_state):
+            model = ChatOpenAI(model=st.session_state.model,
+            temperature=0.8,  # Adjust creativity level
+            max_tokens=150   # Define max output token limit
+            )           
+        else: 
+            model = ChatMistralAI(model=st.session_state.model)
+        if 'model' in st.session_state:
+            used_model=st.session_state.model
         
     return config, thread_id