Update app.py

app.py

@@ -1,11 +1,6 @@
 import gradio as gr
-import openai
 import PyPDF2
 
-# Configure sua chave de API do OpenAI
-openai.api_key = "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"
-
-# Função para obter o texto do arquivo PDF
 def extrair_texto_pdf():
     with open("dados-cdu.pdf", "rb") as pdf_file:
         pdf_reader = PyPDF2.PdfReader(pdf_file)
@@ -14,19 +9,22 @@ def extrair_texto_pdf():
             text += page.extract_text()
     return text
 
-def gerar_cdu(palavras_chave):
+def consultar_cdu(palavra_chave):
     texto_base = extrair_texto_pdf()
-    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. Dados da base: {texto_base}"
-    response = openai.Completion.create(
-        model="text-davinci-003",
-        prompt=prompt,
-        max_tokens=100,
-        n=1,
-        stop=None,
-        temperature=0.7
-    )
-    cdu = response.choices[0].text.strip()
-    return cdu
+    palavras_chave_encontradas = []
+    linhas = texto_base.split("\n")
+    for linha in linhas:
+        if palavra_chave.lower() in linha.lower():
+            palavras_chave_encontradas.append(linha)
+    return palavras_chave_encontradas
+
+def gerar_cdu(palavras_chave):
+    palavras_chave = [palavra.strip() for palavra in palavras_chave.split(",")]
+    resultados = {}
+    for palavra_chave in palavras_chave:
+        cdu_encontrada = consultar_cdu(palavra_chave)
+        resultados[palavra_chave] = cdu_encontrada
+    return resultados
 
 # Interface Gradio
 input_text = gr.inputs.Textbox(label="Inserir palavras-chave separadas por vírgulas")
@@ -35,11 +33,12 @@ output_text = gr.outputs.Textbox(label="Resultado")
 
 def generate_cdu(palavras_chave):
     if palavras_chave:
-        return gerar_cdu(palavras_chave)
+        resultados = gerar_cdu(palavras_chave)
+        return resultados
     else:
-        return ""
+        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).launch()
+gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()