Update app.py

app.py

@@ -1,28 +1,21 @@
 import gradio as gr
 import openai
-import requests
 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 a partir do link
-def extrair_texto_pdf(url):
-    response = requests.get(url)
-    with open("dados-cdu.pdf", "wb") as f:
-        f.write(response.content)
-
-    pdf_file = open("dados-cdu.pdf", "rb")
-    pdf_reader = PyPDF2.PdfReader(pdf_file)
-    text = ""
-    for page in pdf_reader.pages:
-        text += page.extract_text()
-    pdf_file.close()
-
+# 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)
+        text = ""
+        for page in pdf_reader.pages:
+            text += page.extract_text()
     return text
 
 def gerar_cdu(palavras_chave):
-    texto_base = extrair_texto_pdf("https://huggingface.co/spaces/itacaiunas/gerador-cdu/blob/main/dados-cdu.pdf")
+    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",
@@ -49,5 +42,4 @@ def generate_cdu(palavras_chave):
 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, model_name="gpt-3.5-turbo", button_text=button_label).launch()
-
+gr.Interface(fn=generate_cdu, inputs=input_text, outputs=output_text, title=title, description=description).launch()