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."
    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 = gr.inputs.Button(label="Gerar CDU")
output_text = gr.outputs.Textbox(label="Resultado")

def generate_cdu(inputs):
    if inputs:
        return gerar_cdu(inputs)
    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=button).launch()