gerador-cdu / app.py
itacaiunas's picture
Update app.py
e43ad01
raw
history blame
1.2 kB
import gradio as gr
import openai
# Configure sua chave de API do OpenAI
openai.api_key = "ssk-gBqEtrxdoptJnst5BpW2T3BlbkFJej1FLHcITub1EylmooQH"
def gerar_cdu(palavras_chave):
prompt = f"analise as palavras-chave: {palavras_chave} e retorne com o número de Classificação Decimal Universal (CDU) mais relacionado com as palavras-chave."
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=100,
n=1,
stop=None,
temperature=0.7
)
cdu = response.choices[0].text.strip()
return cdu
# Interface Gradio
input_text = gr.inputs.Textbox(label="Inserir palavras-chave separadas por vírgulas")
button_label = "Gerar CDU"
output_text = gr.outputs.Textbox(label="Resultado")
def generate_cdu(palavras_chave):
if palavras_chave:
return gerar_cdu(palavras_chave)
else:
return ""
title = "Gerador de CDU"
description = "Insira palavras-chave separadas por vírgulas e clique em 'Gerar CDU' para obter a classificação relacionada."
gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description, button_text=button_label).launch()