Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import openai
 import json
 from graphviz import Digraph
+from PIL import Image
+import io
 
 def generate_knowledge_graph(api_key, user_input):
     openai.api_key = api_key
@@ -98,11 +100,12 @@ def generate_knowledge_graph(api_key, user_input):
     # Renderizar para o formato PNG
     print("Renderizando o gráfico para o formato PNG...")
     dot.format = "png"
-    dot.render(filename="knowledge_graph", cleanup=True)
+    image_data = dot.pipe(format="png")
+    image = Image.open(io.BytesIO(image_data))
 
     print("Gráfico gerado com sucesso!")
 
-    return "knowledge_graph.png"
+    return image
 
 iface = gr.Interface(
     fn=generate_knowledge_graph,
@@ -110,7 +113,7 @@ iface = gr.Interface(
         gr.inputs.Textbox(label="OpenAI API Key", type="password"),
         gr.inputs.Textbox(label="User Input for Graph"),
     ],
-    outputs=gr.outputs.Image(type="file", label="Generated Knowledge Graph"),
+    outputs=gr.outputs.Image(type="PIL", label="Generated Knowledge Graph"),
     live=False,
 )