enhance

audit_page/dialogue_doc.py

@@ -35,6 +35,16 @@ def chat_history_formatter(chat_history):
             formatted_chat += f"Human:\n {message.content}\n\n"
     return formatted_chat
 
+def filter_correspondance(source_list:list[str],ref_dict:dict,reverse=False):
+    source_list = [item.lower().strip() for item in source_list]
+    if reverse:
+        return [key for key, value in ref_dict.items() if value.lower().strip() in source_list]
+    else:
+        # 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]
+
+
 def doc_dialog_main():
     st.title("Dialogue avec le document")
 
@@ -182,7 +192,7 @@ def doc_dialog_main():
     elif choice == "graphe de connaissance":
         if "graph" not in st.session_state or st.session_state.graph == None:
             with st.spinner("Génération du graphe..."):
-                keywords_list = audit["Mots clés"].strip().split(",")
+                keywords_list = [keyword.strip() for keyword in audit["Mots clés"].strip().split(",")]
                 allowed_nodes_types =keywords_list+ ["Person","Organization","Location","Event","Date","Time","Ressource","Concept"]
                 graph = get_graph(text,allowed_nodes=allowed_nodes_types)
                 st.session_state.graph = graph
@@ -219,11 +229,17 @@ def doc_dialog_main():
                 
                 #add mots cles to evry label in audit["Mots clés"]
                 #filter_labels = [ label + " (mot clé)" if label.strip().lower() in audit["Mots clés"].strip().lower().split(",") else label for label in st.session_state.filter_views[st.session_state.current_view] ]
-                filter = filter_col.multiselect("Filtrer selon l'étiquette",st.session_state.node_types.keys(),placeholder="Sélectionner une ou plusieurs étiquettes",default=st.session_state.filter_views[st.session_state.current_view],label_visibility="collapsed")
+                keywords_list = [keyword.strip().lower() for keyword in audit["Mots clés"].strip().split(",")]
+                dict_filters = {label: "Mot clé : "+label if label.strip().lower() in keywords_list else label for label in st.session_state.filter_views[st.session_state.current_view]}
                 
+                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)
+                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"):
                     add_view_dialog(filter)
-                
                 if filter:
                     nodes = filter_nodes_by_types(nodes,filter)