knowledge graph

app.py

@@ -132,7 +132,7 @@ Low APR and great customer service. I would highly recommend if you’re looking
     ================================================
     """)
     in_verbatim = gr.Textbox(label="Question")
-    out_product = gr.Image(label="Knowledge Graph")
+    out_product = gr.JSON(label="Knowledge Graph")
 
     gr.Examples(
       [

knowledge.py

@@ -49,7 +49,7 @@ client = instructor.from_openai(
 """
 def generate_graph(input) -> KnowledgeGraph:
     return client.chat.completions.create(
-    model='llama-3.1-8b-instant', #"llama3.2",
+    model='llama-3.1-8b-instant', #"llama3.2", #
     max_retries=5,
     messages=[
         {
@@ -59,23 +59,6 @@ def generate_graph(input) -> KnowledgeGraph:
     ],
     response_model=KnowledgeGraph,
 )
-
-
-def visualize_knowledge_graph(kg: KnowledgeGraph):
-    dot = Digraph(comment="Knowledge Graph")
-
-    # Add nodes
-    for node in kg.nodes:
-        dot.node(str(node.id), node.label, color=node.color)
-
-    # Add edges
-    for edge in kg.edges:
-        dot.edge(str(edge.source), str(edge.target), label=edge.label, color=edge.color)
-
-    # Render the graph
-    dot.render("/tmp/knowledge_graph", format="png")
-
 def graph(query):
     graph = generate_graph(query)
-    visualize_knowledge_graph(graph)
-    return "/tmp/knowledge_graph.png"
+    return graph.json()