Update app.py

app.py

@@ -94,19 +94,28 @@ def generate_knowledge_graph(api_key, user_input):
     # Convert to dictionary if it's a string
     if isinstance(response_data, str):
         response_data = json.loads(response_data)
-        
+
     # Visualizar o conhecimento usando Graphviz
     print("Gerando o conhecimento usando Graphviz...")
-    dot = Digraph(comment="Knowledge Graph")
+    dot = Digraph(comment="Knowledge Graph", format='png')
+    dot.attr(dpi='300')
+    dot.attr(bgcolor='transparent')
+
+    # Estilizar os nós
+    dot.attr('node', shape='box', style='filled', fillcolor='lightblue', fontcolor='black')
+
     for node in response_data.get("nodes", []):
         dot.node(node["id"], f"{node['label']} ({node['type']})", color=node.get("color", "lightblue"))
+
+    # Estilizar as arestas
+    dot.attr('edge', color='black', fontcolor='black')
+
     for edge in response_data.get("edges", []):
         dot.edge(edge["from"], edge["to"], label=edge["relationship"], color=edge.get("color", "black"))
 
     # Renderizar para o formato PNG
     print("Renderizando o gráfico para o formato PNG...")
-    dot.format = "png"
-    image_data = dot.pipe(format="png")
+    image_data = dot.pipe()
     image = Image.open(io.BytesIO(image_data))
 
     print("Gráfico gerado com sucesso!")