Spaces:
Sleeping
Sleeping
Commit
·
4f2de6f
1
Parent(s):
a4b8ea3
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,6 +8,10 @@ import io
|
|
| 8 |
def generate_knowledge_graph(api_key, user_input):
|
| 9 |
openai.api_key = api_key
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
# Chamar a API da OpenAI
|
| 12 |
print("Chamando a API da OpenAI...")
|
| 13 |
completion = openai.ChatCompletion.create(
|
|
@@ -122,15 +126,31 @@ def generate_knowledge_graph(api_key, user_input):
|
|
| 122 |
|
| 123 |
return image
|
| 124 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 125 |
iface = gr.Interface(
|
| 126 |
fn=generate_knowledge_graph,
|
| 127 |
inputs=[
|
| 128 |
gr.inputs.Textbox(label="OpenAI API Key", type="password"),
|
| 129 |
-
gr.inputs.Textbox(label="User Input for Graph"),
|
| 130 |
],
|
| 131 |
outputs=gr.outputs.Image(type="pil", label="Generated Knowledge Graph"),
|
| 132 |
live=False,
|
|
|
|
| 133 |
)
|
| 134 |
|
|
|
|
|
|
|
|
|
|
| 135 |
print("Iniciando a interface Gradio...")
|
| 136 |
-
iface.launch()
|
|
|
|
| 8 |
def generate_knowledge_graph(api_key, user_input):
|
| 9 |
openai.api_key = api_key
|
| 10 |
|
| 11 |
+
# Ensure both API key and user input are provided
|
| 12 |
+
if not api_key or not user_input:
|
| 13 |
+
raise ValueError("Please provide both the OpenAI API Key and User Input")
|
| 14 |
+
|
| 15 |
# Chamar a API da OpenAI
|
| 16 |
print("Chamando a API da OpenAI...")
|
| 17 |
completion = openai.ChatCompletion.create(
|
|
|
|
| 126 |
|
| 127 |
return image
|
| 128 |
|
| 129 |
+
# Define a title and description for the Gradio interface using Markdown
|
| 130 |
+
title_and_description = """
|
| 131 |
+
# Instagraph - Knowledge Graph Generator
|
| 132 |
+
|
| 133 |
+
**Created by [ArtificialGuyBR](https://twitter.com/ArtificialGuyBR)**
|
| 134 |
+
|
| 135 |
+
This interactive knowledge graph generator is inspired by [this GitHub project](https://github.com/yoheinakajima/instagraph/).
|
| 136 |
+
|
| 137 |
+
Enter your OpenAI API Key and a question, and let the AI create a detailed knowledge graph for you.
|
| 138 |
+
"""
|
| 139 |
+
|
| 140 |
+
# Create the Gradio interface with queueing enabled and concurrency_count set to 10
|
| 141 |
iface = gr.Interface(
|
| 142 |
fn=generate_knowledge_graph,
|
| 143 |
inputs=[
|
| 144 |
gr.inputs.Textbox(label="OpenAI API Key", type="password"),
|
| 145 |
+
gr.inputs.Textbox(label="User Input for Graph", type="text"),
|
| 146 |
],
|
| 147 |
outputs=gr.outputs.Image(type="pil", label="Generated Knowledge Graph"),
|
| 148 |
live=False,
|
| 149 |
+
title=title_and_description,
|
| 150 |
)
|
| 151 |
|
| 152 |
+
# Enable queueing system for multiple users
|
| 153 |
+
iface.queue(concurrency_count=10)
|
| 154 |
+
|
| 155 |
print("Iniciando a interface Gradio...")
|
| 156 |
+
iface.launch()
|