Update logic.py

logic.py

@@ -2,6 +2,9 @@ import kuzu
 import logging
 import sys
 import os
+import rdflib
+from rdflib import Graph, Literal, RDF, URIRef
+from rdflib.namespace import FOAF, XSD
 #import llama_index
 from llama_index.graph_stores import KuzuGraphStore
 from llama_index import (
@@ -136,7 +139,33 @@ def get_visualize_embeddings(embedding_series, n_neighbors=15, min_dist=0.1, n_c
     fig = px.scatter(umap_df, x='UMAP Dimension 1', y='UMAP Dimension 2',hover_data=['Label'], title='UMAP Visualization of Embeddings')
     return fig
 
+def generate_rdf(index):
+    g = Graph()
+    
+    # Define namespace prefixes
+    EX = Namespace("http://example.com/")
+    
+    # Iterate over the nodes in the index
+    for node in index.index_struct.node_dict.values():
+        subject = EX[str(node.node_id)]
+        
+        # Add triples for node properties
+        g.add((subject, RDF.type, EX["Node"]))
+        g.add((subject, EX["text"], Literal(node.text)))
+        
+        # Add triples for node relationships
+        for relationship in node.relationships:
+            predicate = EX[relationship.predicate]
+            object_node = EX[str(relationship.object_id)]
+            g.add((subject, predicate, object_node))
+    
+    return g
 
+def visualize_rdf(rdf_graph):
+    # Visualize the RDF graph (you can use a library like PyVis or D3.js)
+    # For simplicity, let's serialize the RDF graph to a string
+    rdf_string = rdf_graph.serialize(format="turtle").decode("utf-8")
+    return rdf_string
 def query_model(index,user_query):
     query_engine = index.as_query_engine(
     include_text=True,