paste

.gitignore

@@ -1,4 +1,4 @@
 __pycache__/
-.streamlit/secrets.toml
-.streamlit/.env
+.streamlit/
+.streamlit/
 DATA_bziiit/vectorstore_op

chat_with_pps.py

@@ -7,6 +7,7 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_mistralai.chat_models import ChatMistralAI
 from download_chart import construct_plot
 from prompt import get_prompts_list
+from st_copy_to_clipboard import st_copy_to_clipboard
 
 
 load_dotenv()
@@ -87,6 +88,7 @@ def choose_model(index):
 def display_chat():
     # app config
     st.title("Chatbot")
+    
 
     models_name = {
         "Mistral (FR)":1,
@@ -147,6 +149,7 @@ def display_chat():
     if isinstance(last_message, HumanMessage):
         with st.chat_message("AI"):
             response = st.write_stream(get_response(last_message.content, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque']),st.session_state.model))
+            st_copy_to_clipboard(response)
             st.session_state.chat_history.append(AIMessage(content=response))
 
 
@@ -165,7 +168,8 @@ def display_chat():
         with st.chat_message("AI"):
             st.markdown(f"**{st.session_state.model}**")
             response = st.write_stream(get_response(user_query, st.session_state.chat_history,format_context(st.session_state['pp_grouped'],st.session_state['Nom de la marque']),st.session_state.model))
-            if "cartographie des parties prenantes" in message.content:
+            st_copy_to_clipboard(response)
+            if "cartographie" in response:
                     display_chart()
 
         st.session_state.chat_history.append(AIMessage(content=response))