analyse

audit_page/knowledge_graph.py

@@ -275,13 +275,36 @@ def format_relationships(relationships : list[Edge]):
         for rel in relationships
     )
 
+def fortmat_nodes(nodes : list[Node]):
+    """Format nodes for display in the chat."""
+    return "\n".join(
+        f"- **{node.label}** ({node.title})"
+        for node in nodes
+    )
+
 def add_relationship_to_graph(source_id, target_id, relationship_type):
     st.session_state.edges.append(Edge(source=source_id, label=relationship_type, target=target_id))
+    print(f"Relation ajoutée: {source_id} -- {relationship_type} --> {target_id}")
+
+    if not if_node_exists(st.session_state.nodes, source_id):
+        st.session_state.nodes.append(Node(
+            id=source_id,
+            title="Autre",
+            label=source_id,
+            size=25,
+            shape="circle",
+            color=st.session_state.node_types.get(target.label, "#CCCCCC")
+        ))
+        print(f"Node ajouté: {source_id}")
+
+    print(f"Nodes: {fortmat_nodes(st.session_state.nodes)}")
 
 def delete_relationship_from_graph(source_id, target_id, relationship_type):
     st.session_state.edges = [edge for edge in st.session_state.edges if not (
         edge.source == source_id and edge.to == target_id and edge.label == relationship_type
     )]
+
+
     
 
 
@@ -360,6 +383,8 @@ def change_color_dialog():
             f"La couleur de l'entité **{node_type.strip()}**",
             color
         )
+        print("New color:", new_color)
+        print("Old color:", color)
         st.session_state.node_types[node_type] = new_color
     
     if st.button("Valider"):
@@ -408,6 +433,8 @@ def kg_main():
             
             print("Graph loaded.")
 
+    
+
     # 2. Initialize other session keys if they don’t exist
     if "filter_views" not in st.session_state:
         st.session_state.filter_views = {}
@@ -446,6 +473,15 @@ def kg_main():
     config = st.session_state.config
     print("Graph converti en Agraph")
 
+    #ask chatgpt to analyse the graph
+    prompt = ("Tu es un expert en graphes de connaissances, analyse le graphe et donne une synthèse et differentes conclusions sur les elements du recit, tout en etant pertinent et precis",
+                "**Graphe**:,"
+                f"**Noeuds**: {fortmat_nodes(st.session_state.nodes)}\n"
+                f"Relations: {format_relationships(st.session_state.edges)}",
+                "Output: tu dois donner une synthèse et des conclusions sur les elements du recit , ca sera le premier message de la conversation"),  
+    response = generate_response_via_langchain(prompt)
+    st.session_state.chat_graph_history.append(AIMessage(content=response))
+
     # 4. UI layout: (left) the graph itself, (right) the chat
     col1, col2 = st.columns([3, 1])