enhance

audit_page/dialogue_doc.py

@@ -8,6 +8,7 @@ from utils.kg.construct_kg import get_graph
 from audit_page.knowledge_graph import *
 import json
 import clipboard
+from time import sleep
 
 def graph_doc_to_json(graph):
     nodes = []
@@ -30,9 +31,9 @@ def chat_history_formatter(chat_history):
     formatted_chat = ""
     for message in chat_history:
         if isinstance(message, AIMessage):
-            formatted_chat += f"AI:\n {message.content}\n\n"
+            formatted_chat += f"AI:{message.content}\n\n"
         elif isinstance(message, HumanMessage):
-            formatted_chat += f"Human:\n {message.content}\n\n"
+            formatted_chat += f"Human:{message.content}\n\n"
     return formatted_chat
 
 def filter_correspondance(source_list:list[str],ref_dict:dict,reverse=False):
@@ -43,6 +44,16 @@ def filter_correspondance(source_list:list[str],ref_dict:dict,reverse=False):
         # st.write(source_list)
         # st.write(ref_dict.keys())
         return [value for key, value in ref_dict.items() if key.lower().strip() in source_list]
+    
+@st.fragment()
+def radio_choice():
+    options = ["compte_rendu","graphe de connaissance"]
+    choice = st.radio("Choisissez une option",options,index=st.session_state.radio_choice,horizontal=True,label_visibility="collapsed")
+    sleep(1)
+    if choice and options.index(choice) != st.session_state.radio_choice:
+        sleep(1)
+        st.session_state.radio_choice = options.index(choice)
+    return choice
 
 
 def doc_dialog_main():
@@ -74,13 +85,19 @@ def doc_dialog_main():
     if "chat_graph_history" not in st.session_state:
         st.session_state.chat_graph_history = []
     
+    #init a radio button for the choice
     if "radio_choice" not in st.session_state:
         st.session_state.radio_choice = None
+    # if "choice" not in st.session_state:
+    #     st.session_state.choice = st.radio("Choisissez une option",["compte_rendu","graphe de connaissance"],index=st.session_state.radio_choice,horizontal=True,label_visibility="collapsed")
 
+    # choice = radio_choice()
+    
     options = ["compte_rendu","graphe de connaissance"]
     choice = st.radio("Choisissez une option",options,index=st.session_state.radio_choice,horizontal=True,label_visibility="collapsed")
-    if choice:
+    if choice and options.index(choice) != st.session_state.radio_choice:
         st.session_state.radio_choice = options.index(choice)
+        
     
     audit = st.session_state.audit_simplified
     content = st.session_state.audit["content"]
@@ -186,7 +203,7 @@ def doc_dialog_main():
                 # 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)
-                    st_copy_to_clipboard(chat_formatted,key="cp_but_cr_chat")
+                    st_copy_to_clipboard(chat_formatted,key="cp_but_cr_chat",show_text=False)
                 # col_success_c.success("Historique copié !")
 
     elif choice == "graphe de connaissance":
@@ -202,7 +219,10 @@ def doc_dialog_main():
                 st.session_state.chat_graph_history = []
             
             node_types = get_node_types(graph[0])
-            nodes_type_dict = list_to_dict_colors(node_types)
+            list_node_types = list(node_types)
+            sorted_node_types = sorted(list_node_types,key=lambda x: x.lower())
+            print(sorted_node_types)
+            nodes_type_dict = list_to_dict_colors(sorted_node_types)
             st.session_state.node_types = nodes_type_dict
             st.session_state.filter_views["Vue par défaut"] = list(node_types)
             st.session_state.current_view = "Vue par défaut"
@@ -235,7 +255,9 @@ def doc_dialog_main():
                 default_target_filter = filter_correspondance(st.session_state.filter_views[st.session_state.current_view],dict_filters)
                 # st.write(default_target_filter)
                 # st.write(dict_filters)
-                target_filter = filter_correspondance(st.session_state.node_types.keys(),dict_filters)
+                sorted_default_target_filter = sorted(default_target_filter,key=lambda x: x.lower())
+                target_filter = filter_correspondance(list(st.session_state.node_types.keys()),dict_filters)
+                target_filter = sorted(target_filter,key=lambda x: x.lower())
                 filter = filter_col.multiselect("Filtrer selon l'étiquette",target_filter,placeholder="Sélectionner une ou plusieurs étiquettes",default=default_target_filter,label_visibility="collapsed")
                 filter = filter_correspondance(filter,dict_filters,reverse=True)
                 if add_view_col.button("➕",help="Ajouter une vue"):
@@ -292,16 +314,11 @@ def doc_dialog_main():
                 # 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)
-                    st_copy_to_clipboard(chat_formatted,key="cp_but_graph_chat")   
+                    st_copy_to_clipboard(chat_history_formatter(st.session_state.chat_graph_history),key="cp_but_graph_chat",show_text=False)   
                     # col_success_c.success("Historique copié !")
             
 
 
-        
-
-    
-    
 doc_dialog_main()
 
 

audit_page/knowledge_graph.py

@@ -159,6 +159,13 @@ def change_view_dialog():
             st.rerun()
         if len(st.session_state.filter_views.keys()) > index:
             with col1.expander(item):
+                if index > 0:
+                    change_name = st.text_input("Nom de la vue", label_visibility="collapsed", placeholder="Changez le nom de la vue",key=f"change_name{index}")
+                    if st.button("Renommer",key=f"rename{index}"):
+                        if change_name != "":
+                            st.session_state.filter_views[change_name] = st.session_state.filter_views.pop(item)
+                            st.session_state.current_view = change_name
+                            st.rerun()
                 st.markdown("\n".join(f"- {label.strip()}" for label in st.session_state.filter_views[item]))
         else:
             emp.empty()