Fix multi messages by companies

app.py

@@ -32,6 +32,9 @@ def init_app():
         st.session_state["data_dict"] = {}
         st.session_state["company_code"] = None
         st.session_state["prompt_system"] = config['prompt_system']
+        st.session_state["chat_history_afp"] = []
+        st.session_state["chat_history_cettons"] = []
+        st.session_state["chat_history_steel"] = []
 
         # Initialize data_dict for each company
         for company in config['companies']:

pages/chatbot_afp.py

@@ -8,7 +8,7 @@ from subscription.index import is_usage_limit_reached, daily_limit_enabled
 
 def display_messages():
 
-    for i, message in enumerate(st.session_state.chat_history):
+    for i, message in enumerate(st.session_state.chat_history_afp):
         if isinstance(message, AIMessage):
             with st.chat_message("AI"):
                 # Display the model from the kwargs
@@ -16,7 +16,7 @@ def display_messages():
                 st.write(f"**Model :** {model}")
                 st.markdown(message.content)
                 st_copy_to_clipboard(message.content,key=f"message_{i}")
-                # show_retrieved_documents(st.session_state.chat_history[i-1].content)
+                # show_retrieved_documents(st.session_state.chat_history_afp[i-1].content)
         
         elif isinstance(message, HumanMessage):
             with st.chat_message("Moi"):
@@ -48,12 +48,12 @@ def launchQuery(query: str = None):
         st.session_state["assistant"].ask(
             query,
             # prompt_system=st.session_state.prompt_system,
-            messages=st.session_state["chat_history"] if "chat_history" in st.session_state else [],
+            messages=st.session_state["chat_history_afp"] if "chat_history_afp" in st.session_state else [],
             variables=st.session_state["data_dict"]
         ))
 
     # Temporary placeholder AI message in chat history
-    st.session_state["chat_history"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
+    st.session_state["chat_history_afp"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
     st.rerun()
 
 
@@ -94,9 +94,8 @@ def page():
     st.subheader("Auto France Parts")
 
     company_code = "afp_"
-    st.session_state["chat_history"] = []
 
-    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+    if "chat_history_afp" not in st.session_state or len(st.session_state["chat_history_afp"]) <= 2:
         # print("Initialisation de l'historique")
         if st.session_state["data_dict"][company_code] is not None:
             # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
@@ -105,7 +104,7 @@ def page():
         system_template = st.session_state.prompt_system
         full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
 
-        st.session_state["chat_history"] = [
+        st.session_state["chat_history_afp"] = [
             SystemMessage(content=full),
         ]
 
@@ -120,9 +119,9 @@ def page():
     # Models selector
     selector.ModelSelector()
 
-    if(len(st.session_state["chat_history"])):
+    if(len(st.session_state["chat_history_afp"])):
         if st.button("Effacer l'historique"):
-            st.session_state["chat_history"] = []
+            st.session_state["chat_history_afp"] = []
 
     # Displaying messages
     display_messages()
@@ -139,7 +138,7 @@ def page():
 
     if user_query is not None and user_query != "":
 
-        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        st.session_state["chat_history_afp"].append(HumanMessage(content=user_query))
         
         # Stream and display response
         launchQuery(user_query)

pages/chatbot_cettons.py

@@ -7,8 +7,8 @@ from st_copy_to_clipboard import st_copy_to_clipboard
 from subscription.index import is_usage_limit_reached, daily_limit_enabled
 
 def display_messages():
-
-    for i, message in enumerate(st.session_state.chat_history):
+    print(st.session_state.chat_history_cettons)
+    for i, message in enumerate(st.session_state.chat_history_cettons):
         if isinstance(message, AIMessage):
             with st.chat_message("AI"):
                 # Display the model from the kwargs
@@ -16,7 +16,7 @@ def display_messages():
                 st.write(f"**Model :** {model}")
                 st.markdown(message.content)
                 st_copy_to_clipboard(message.content,key=f"message_{i}")
-                # show_retrieved_documents(st.session_state.chat_history[i-1].content)
+                # show_retrieved_documents(st.session_state.chat_history_cettons[i-1].content)
         
         elif isinstance(message, HumanMessage):
             with st.chat_message("Moi"):
@@ -48,12 +48,12 @@ def launchQuery(query: str = None):
         st.session_state["assistant"].ask(
             query,
             # prompt_system=st.session_state.prompt_system,
-            messages=st.session_state["chat_history"] if "chat_history" in st.session_state else [],
+            messages=st.session_state["chat_history_cettons"] if "chat_history_cettons" in st.session_state else [],
             variables=st.session_state["data_dict"]
         ))
 
     # Temporary placeholder AI message in chat history
-    st.session_state["chat_history"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
+    st.session_state["chat_history_cettons"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
     st.rerun()
 
 
@@ -94,9 +94,8 @@ def page():
     st.subheader("La Pyramide des Cettons")
 
     company_code = "cettons_"
-    st.session_state["chat_history"] = []
 
-    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+    if "chat_history_cettons" not in st.session_state or len(st.session_state["chat_history_cettons"]) <= 2:
         # print("Initialisation de l'historique")
         if st.session_state["data_dict"][company_code] is not None:
             # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
@@ -105,7 +104,7 @@ def page():
         system_template = st.session_state.prompt_system
         full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
 
-        st.session_state["chat_history"] = [
+        st.session_state["chat_history_cettons"] = [
             SystemMessage(content=full),
         ]
 
@@ -120,9 +119,9 @@ def page():
     # Models selector
     selector.ModelSelector()
 
-    if(len(st.session_state["chat_history"])):
+    if(len(st.session_state["chat_history_cettons"])):
         if st.button("Effacer l'historique"):
-            st.session_state["chat_history"] = []
+            st.session_state["chat_history_cettons"] = []
 
     # Displaying messages
     display_messages()
@@ -139,7 +138,7 @@ def page():
 
     if user_query is not None and user_query != "":
 
-        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        st.session_state["chat_history_cettons"].append(HumanMessage(content=user_query))
         
         # Stream and display response
         launchQuery(user_query)

pages/chatbot_steel.py

@@ -8,7 +8,7 @@ from subscription.index import is_usage_limit_reached, daily_limit_enabled
 
 def display_messages():
 
-    for i, message in enumerate(st.session_state.chat_history):
+    for i, message in enumerate(st.session_state.chat_history_steel):
         if isinstance(message, AIMessage):
             with st.chat_message("AI"):
                 # Display the model from the kwargs
@@ -16,7 +16,7 @@ def display_messages():
                 st.write(f"**Model :** {model}")
                 st.markdown(message.content)
                 st_copy_to_clipboard(message.content,key=f"message_{i}")
-                # show_retrieved_documents(st.session_state.chat_history[i-1].content)
+                # show_retrieved_documents(st.session_state.chat_history_steel[i-1].content)
         
         elif isinstance(message, HumanMessage):
             with st.chat_message("Moi"):
@@ -48,12 +48,12 @@ def launchQuery(query: str = None):
         st.session_state["assistant"].ask(
             query,
             # prompt_system=st.session_state.prompt_system,
-            messages=st.session_state["chat_history"] if "chat_history" in st.session_state else [],
+            messages=st.session_state["chat_history_steel"] if "chat_history_steel" in st.session_state else [],
             variables=st.session_state["data_dict"]
         ))
 
     # Temporary placeholder AI message in chat history
-    st.session_state["chat_history"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
+    st.session_state["chat_history_steel"].append(AIMessage(content=full_response, kwargs={"model": st.session_state["assistant"].getReadableModel()}))
     st.rerun()
 
 
@@ -94,9 +94,8 @@ def page():
     st.subheader("Steel France")
 
     company_code = "steel_"
-    st.session_state["chat_history"] = []
 
-    if "chat_history" not in st.session_state or len(st.session_state["chat_history"]) <= 2:
+    if "chat_history_steel" not in st.session_state or len(st.session_state["chat_history_steel"]) <= 2:
         # print("Initialisation de l'historique")
         if st.session_state["data_dict"][company_code] is not None:
             # Convertir la liste en dictionnaire avec 'key' comme clé et 'value' comme valeur
@@ -105,7 +104,7 @@ def page():
         system_template = st.session_state.prompt_system
         full = remplir_texte(system_template, vars, st.session_state["remove_undefined_value"])
 
-        st.session_state["chat_history"] = [
+        st.session_state["chat_history_steel"] = [
             SystemMessage(content=full),
         ]
 
@@ -120,9 +119,9 @@ def page():
     # Models selector
     selector.ModelSelector()
 
-    if(len(st.session_state["chat_history"])):
+    if(len(st.session_state["chat_history_steel"])):
         if st.button("Effacer l'historique"):
-            st.session_state["chat_history"] = []
+            st.session_state["chat_history_steel"] = []
 
     # Displaying messages
     display_messages()
@@ -139,7 +138,7 @@ def page():
 
     if user_query is not None and user_query != "":
 
-        st.session_state["chat_history"].append(HumanMessage(content=user_query))
+        st.session_state["chat_history_steel"].append(HumanMessage(content=user_query))
         
         # Stream and display response
         launchQuery(user_query)