gerador-cdu / app.py
itacaiunas's picture
Update app.py
672b48d
raw
history blame
1.27 kB
import gradio as gr
import openai
# Configure sua chave de API do OpenAI
openai.api_key = "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"
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. Utilize como base o site: https://udcsummary.info/php/index.php?id=68076&lang=pt"
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=150,
n=1,
stop=None,
temperature=0.5
)
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 as palavras-chave 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()