enhance

audit_page/dialogue_doc.py

@@ -3,9 +3,11 @@ from textwrap import dedent
 from utils.audit.rag import get_text_from_content_for_doc,get_text_from_content_for_audio
 from utils.audit.response_llm import generate_response_via_langchain
 from langchain_core.messages import AIMessage, HumanMessage
-import pyperclip
+from st_copy_to_clipboard import st_copy_to_clipboard
 from utils.kg.construct_kg import get_graph
 from audit_page.knowledge_graph import *
+import json
+import clipboard
 
 def graph_doc_to_json(graph):
     nodes = []
@@ -132,18 +134,20 @@ def doc_dialog_main():
         if cr:
             col1, col2 = st.columns([2.5, 1.5])
 
-            with col1.container(border=True,height=800):
+            with col1.container(border=True,height=850):
                 st.markdown("##### Compte rendu")
+                keywords_paragraph = f"### Mots clés extraits:\n- {audit['Mots clés'].strip()}"
                 with st.container(height=650,border=False):
-                    keywords_paragraph = f"### Mots clés extraits:\n- {audit['Mots clés'].strip()}"
                     st.markdown(keywords_paragraph)
                     st.write(cr)
                 col_copy , col_success = st.columns([1,11])
                 if col_copy.button("📋",key="copy_cr"):
-                    pyperclip.copy(keywords_paragraph+"\n\n"+cr)
+                # with st.container(height=50,border=False):
+                #     st_copy_to_clipboard(keywords_paragraph+"\n\n"+cr,key="cp_but_cr")
+                    clipboard.copy(keywords_paragraph+"\n\n"+cr)
                     col_success.success("Compte rendu copié dans le presse-papier")
                 
-            with col2.container(border=True,height=800):
+            with col2.container(border=True,height=850):
                 st.markdown("##### Dialoguer avec le CR")
 
                 user_query = st.chat_input("Par ici ...")
@@ -169,11 +173,12 @@ def doc_dialog_main():
                                 wrapped_prompt = f'''Étant donné le contexte suivant {context} et le compte rendu du document {cr}, {last_message.content}'''
                                 response = st.write_stream(generate_response_via_langchain(wrapped_prompt,stream=True))
                                 st.session_state.cr_chat_history.append(AIMessage(content=response))
-                col_copy_c , col_success_c = st.columns([1,7])
-                if col_copy_c.button("📋",key="copy_cr_chat"):
+                # col_copy_c , col_success_c = st.columns([1,7])
+                # if col_copy_c.button("📋",key="copy_cr_chat"):
+                with st.container(height=50,border=False):
                     chat_formatted = chat_history_formatter(st.session_state.cr_chat_history)
-                    pyperclip.copy(chat_formatted)
-                    col_success_c.success("Historique copié !")
+                    st_copy_to_clipboard(chat_formatted,key="cp_but_cr_chat")
+                # col_success_c.success("Historique copié !")
 
     elif choice == "graphe de connaissance":
         if "graph" not in st.session_state or st.session_state.graph == None:
@@ -202,7 +207,7 @@ def doc_dialog_main():
             
             col1, col2 = st.columns([2.5, 1.5])
 
-            with col1.container(border=True,height=800):
+            with col1.container(border=True,height=850):
                 st.write("##### Visualisation du graphe (**"+st.session_state.current_view+"**)")
                 filter_col,add_view_col,change_view_col,color_col = st.columns([9,1,1,1])
                 
@@ -225,13 +230,14 @@ def doc_dialog_main():
 
                 selected = display_graph(edges,nodes,config)
 
-                col_copy , col_success = st.columns([1,11])
-                if col_copy.button("📋",key="copy_graph"):
+                # col_copy , col_success = st.columns([1,11])
+                # if col_copy.button("📋",key="copy_graph"):
+                with st.container(height=50,border=False):
                     graph_json = graph_doc_to_json(graph[0])
-                    pyperclip.copy(graph_json)
-                    col_success.success("Graphe copié dans le presse-papier")
+                    st_copy_to_clipboard(json.dumps(graph_json),key="cp_but_graph")
+                    # col_success.success("Graphe copié dans le presse-papier")
 
-            with col2.container(border=True,height=800):
+            with col2.container(border=True,height=850):
                 st.markdown("##### Dialoguer avec le graphe")
 
                 user_query = st.chat_input("Par ici ...")
@@ -268,11 +274,12 @@ def doc_dialog_main():
                                 for i,prompt in enumerate(prompts):
                                     button = st.button(prompt,key=f"p_{i}",on_click=lambda i=i: st.session_state.chat_graph_history.append(HumanMessage(content=prompts[i])))
                 
-                col_copy_c , col_success_c = st.columns([1,7])
-                if col_copy_c.button("📋",key="copy_graph_chat"):
+                # col_copy_c , col_success_c = st.columns([1,7])
+                # if col_copy_c.button("📋",key="copy_graph_chat"):
+                with st.container(height=50,border=False):
                     chat_formatted = chat_history_formatter(st.session_state.chat_graph_history)
-                    pyperclip.copy(chat_formatted)
-                    col_success_c.success("Historique copié !")
+                    st_copy_to_clipboard(chat_formatted,key="cp_but_graph_chat")   
+                    # col_success_c.success("Historique copié !")
             
 
 

requirements.txt

@@ -23,4 +23,6 @@ python-dotenv
 langchain-experimental 
 neo4j
 streamlit-agraph
+st-copy-to-clipboard
+clipboard