Update app.py

app.py

@@ -8,7 +8,6 @@ import shutil
 import tempfile
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from langchain_community.document_loaders import PyMuPDFLoader
-from PyPDF2 import PdfReader
 
 # Configurações do modelo
 MODEL_PATH = "numind/NuExtract-1.5"
@@ -142,23 +141,15 @@ def process_and_generate(pdf_file):
     finally:
         os.remove(pdf_path)  # Remove o arquivo temporário após o uso
 
-def extract_text(file):
-    reader = PdfReader(file.name)
-    text = "\n".join([page.extract_text() for page in reader.pages if page.extract_text()])
-    return text, text  # Retorna o texto para ambas as saídas
 
 
-with gr.Blocks() as app:
-    gr.Markdown("# Extrator de Texto de PDF")
-    
-    with gr.Row():
-        pdf_input = gr.File(label="Envie um arquivo PDF")
-    
-    with gr.Row():
-        text_output = gr.Textbox(label="Texto Extraído", lines=15, interactive=False)
-        json_output = gr.JSON(label="Dados Extraídos")
-    
-    extract_button = gr.Button("Extrair Texto")
-    extract_button.click(extract_text, inputs=pdf_input, outputs=[text_output, json_output])
-    
-app.launch()
+
+interface = gr.Interface(
+    fn=process_and_generate,
+    inputs=gr.File(label="Upload PDF"),
+    outputs=gr.JSON(label="Dados Extraídos"),
+    title="Extração de Dados com Modelo Local",
+    description="Envie um PDF para extrair e processar informações automaticamente.",
+)
+
+interface.launch()