|
import gradio as gr |
|
import openai |
|
import json |
|
from graphviz import Digraph |
|
from PIL import Image |
|
import io |
|
import requests |
|
from bs4 import BeautifulSoup |
|
|
|
|
|
def scrape_text_from_url(url): |
|
response = requests.get(url) |
|
if response.status_code != 200: |
|
return "Error: Could not retrieve content from URL." |
|
soup = BeautifulSoup(response.text, "html.parser") |
|
paragraphs = soup.find_all("p") |
|
text = " ".join([p.get_text() for p in paragraphs]) |
|
return text |
|
|
|
def generate_knowledge_graph(api_key, user_input): |
|
openai.api_key = api_key |
|
|
|
|
|
if user_input.startswith("http://") or user_input.startswith("https://"): |
|
user_input = scrape_text_from_url(user_input) |
|
|
|
|
|
print("Chamando a API da OpenAI...") |
|
completion = openai.ChatCompletion.create( |
|
model="gpt-3.5-turbo-16k", |
|
messages=[ |
|
{ |
|
"role": "user", |
|
"content": f"Help me understand following by describing as a detailed knowledge graph: {user_input}", |
|
} |
|
], |
|
functions=[ |
|
{ |
|
"name": "knowledge_graph", |
|
"description": "Generate a knowledge graph with entities and relationships. Use the colors to help differentiate between different node or edge types/categories. Always provide light pastel colors that work well with black font.", |
|
"parameters": { |
|
"type": "object", |
|
"properties": { |
|
"metadata": { |
|
"type": "object", |
|
"properties": { |
|
"createdDate": {"type": "string"}, |
|
"lastUpdated": {"type": "string"}, |
|
"description": {"type": "string"}, |
|
}, |
|
}, |
|
"nodes": { |
|
"type": "array", |
|
"items": { |
|
"type": "object", |
|
"properties": { |
|
"id": {"type": "string"}, |
|
"label": {"type": "string"}, |
|
"type": {"type": "string"}, |
|
"color": {"type": "string"}, |
|
"properties": { |
|
"type": "object", |
|
"description": "Additional attributes for the node", |
|
}, |
|
}, |
|
"required": [ |
|
"id", |
|
"label", |
|
"type", |
|
"color", |
|
], |
|
}, |
|
}, |
|
"edges": { |
|
"type": "array", |
|
"items": { |
|
"type": "object", |
|
"properties": { |
|
"from": {"type": "string"}, |
|
"to": {"type": "string"}, |
|
"relationship": {"type": "string"}, |
|
"direction": {"type": "string"}, |
|
"color": {"type": "string"}, |
|
"properties": { |
|
"type": "object", |
|
"description": "Additional attributes for the edge", |
|
}, |
|
}, |
|
"required": [ |
|
"from", |
|
"to", |
|
"relationship", |
|
"color", |
|
], |
|
}, |
|
}, |
|
}, |
|
"required": ["nodes", "edges"], |
|
}, |
|
} |
|
], |
|
function_call={"name": "knowledge_graph"}, |
|
) |
|
|
|
response_data = completion.choices[0]["message"]["function_call"]["arguments"] |
|
print(response_data) |
|
print("Type of response_data:", type(response_data)) |
|
print("Value of response_data:", response_data) |
|
|
|
|
|
if isinstance(response_data, str): |
|
response_data = json.loads(response_data) |
|
|
|
|
|
print("Gerando o conhecimento usando Graphviz...") |
|
dot = Digraph(comment="Knowledge Graph", format='png') |
|
dot.attr(dpi='300') |
|
dot.attr(bgcolor='transparent') |
|
|
|
|
|
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")) |
|
|
|
|
|
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")) |
|
|
|
|
|
print("Renderizando o gráfico para o formato PNG...") |
|
image_data = dot.pipe() |
|
image = Image.open(io.BytesIO(image_data)) |
|
|
|
print("Gráfico gerado com sucesso!") |
|
|
|
return image |
|
|
|
|
|
title_and_description = """ |
|
# Instagraph - Knowledge Graph Generator |
|
|
|
**Created by [ArtificialGuyBR](https://twitter.com/ArtificialGuyBR)** |
|
|
|
This interactive knowledge graph generator is inspired by [this GitHub project](https://github.com/yoheinakajima/instagraph/). |
|
|
|
Enter your OpenAI API Key and a question, and let the AI create a detailed knowledge graph for you. |
|
""" |
|
|
|
|
|
iface = gr.Interface( |
|
fn=generate_knowledge_graph, |
|
inputs=[ |
|
gr.inputs.Textbox(label="OpenAI API Key", type="password"), |
|
gr.inputs.Textbox(label="User Input for Graph or URL", type="text"), |
|
], |
|
outputs=gr.outputs.Image(type="pil", label="Generated Knowledge Graph"), |
|
live=False, |
|
title=title_and_description, |
|
) |
|
|
|
|
|
iface.queue(concurrency_count=10) |
|
|
|
print("Iniciando a interface Gradio...") |
|
iface.launch() |
|
|