execute prompts

app.py

@@ -20,7 +20,7 @@ from collaborons import display_company_selection_for_materiality,display_materi
 from documentations import display_documentation
 from RAG_PDF_WEB import rag_pdf_web
 from prompt import get_prompts_list,prompt_execution,execute_prompt
-from chat_with_pps import display_chat
+from chat_with_pps import display_chat,display_container_chat
 from high_chart import test_chart
 from chat_te import display_chat_te
 

chat_te.py

@@ -112,7 +112,7 @@ def display_chat_te():
     # if col1.button("chatbot"):
     #     vote("chatbot")
     # with col2:
-    user_query = st.chat_input("Par ici...")
+    user_query = st.chat_input(placeholder="c'est quoi la transition écologique ?")
     if user_query is not None and user_query != "":
         st.session_state.chat_history_te.append(HumanMessage(content=user_query))
 

chat_with_pps.py

@@ -58,6 +58,12 @@ def display_chart():
     plot = construct_plot()
     st.plotly_chart(plot)
 
+@st.experimental_dialog("Cast your vote")
+def vote(item):
+    st.write(f"Why is {item} your favorite?")
+    reason = st.text_input("Because...")
+    if st.button("Submit"):
+        st.rerun()
 
 def display_chat():
     # app config
@@ -68,7 +74,8 @@ def display_chat():
         st.session_state.chat_history = [
             AIMessage(content="Salut, voici votre cartographie des parties prenantes. Que puis-je faire pour vous?"),
         ]
-    
+    if "user_query" not in st.session_state:
+        st.session_state.user_query = "test"
     
  
     # conversation
@@ -81,6 +88,12 @@ def display_chat():
         elif isinstance(message, HumanMessage):
             with st.chat_message("Moi"):
                 st.write(message.content)
+    
+    #check if the last message is from the user , that means execute button has been clicked
+    last_message = st.session_state.chat_history[-1]
+    if isinstance(last_message, HumanMessage):
+        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.chat_history.append(AIMessage(content=response))
 
     if "pp_grouped" not in st.session_state or st.session_state['pp_grouped'] is None or len(st.session_state['pp_grouped']) == 0:
         return None
@@ -88,14 +101,21 @@ def display_chat():
     user_query = st.chat_input("Par ici...")
     if user_query is not None and user_query != "":
         st.session_state.chat_history.append(HumanMessage(content=user_query))
-
         with st.chat_message("Moi"):
             st.markdown(user_query)
-
         with st.chat_message("AI"):
-
             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'])))
             if "cartographie des parties prenantes" in message.content:
                     display_chart()
 
         st.session_state.chat_history.append(AIMessage(content=response))
+    
+
+def display_container_chat():
+    st.markdown(":point_left: Cliquez pour discuter avec le Chatbot", unsafe_allow_html=True)
+    col1, col2 = st.columns([1, 10])
+    with col1:
+        if st.button("display_prompts"):
+            vote("D")
+    with col2:
+        display_chat()

prompt.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import pandas as pd
 from data_manager_bziiit import get_prompts
+from langchain_core.messages import AIMessage, HumanMessage
 from session import get_rag
 
 prompts = [] 
@@ -31,14 +32,19 @@ def get_prompts_list():
             for name, group in grouped:
                 st.subheader(name)  # Display the context name as a subheader
                 for i, row in group.iterrows():
-                    col1, col2, col3, col4 = st.columns((1, 2, 2, 1))
+                    col1, col2, col3, col4 = st.columns((1, 2, 2, 2))
                     col1.write(num)  # index
                     col2.write(row['name'])  # name
                     col3.write(row['text'])  # text
                     num += 1
                     
                     button_phold = col4.empty()  # create a placeholder
-                    do_action = button_phold.button('Voir plus', key=i)
+                    but1, but2 = button_phold.columns(2)
+                    
+                    do_action = but1.button('Voir plus', key=i)
+                    execute = but2.button('Executer', key=f"e{i}")
+                    if execute:
+                        st.session_state.chat_history.append(HumanMessage(content=prompts[i]['text']))
                     if do_action:
                         prompt_html = prompts[i]['text'].replace('\n', '<br>')
                         prompt_metadata = extract_metadata(prompts[i])
@@ -46,8 +52,10 @@ def get_prompts_list():
                         for text in prompt_metadata:
                             prompt_html = prompt_html.replace(f"{text}", f"<span style='font-weight:bold'>{text}</span>")
                                 
-                        st.html(prompt_html)  # Display the full text
-                        button_phold.empty()  # remove button
+                        st.html(prompt_html)
+                        i
+                            # Display the full text
+                          # remove button
         else:
             st.write("Data does not contain 'name', 'context', and 'text' fields.")
     else: