Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import fitz  # PyMuPDF
+import os
 
 # Carregar o tokenizer e o modelo
 tokenizer = AutoTokenizer.from_pretrained("Locutusque/gpt2-xl-conversational")
@@ -21,8 +22,12 @@ def read_pdf(file_path):
 # Função para carregar o PDF e armazenar o conteúdo
 def load_pdf(pdf_file):
     global pdf_content
-    pdf_content = read_pdf(pdf_file.name)
-    return "PDF carregado com sucesso!"
+    pdf_path = pdf_file.name  # Acessar o caminho do arquivo carregado
+    pdf_content = read_pdf(pdf_path)
+    if pdf_content:
+        return "PDF carregado com sucesso!"
+    else:
+        return "Falha ao carregar o PDF."
 
 # Função para responder perguntas com base no conteúdo do PDF
 def answer_question(question, max_length=200, temperature=0.7, top_k=50, top_p=0.95):
@@ -30,6 +35,7 @@ def answer_question(question, max_length=200, temperature=0.7, top_k=50, top_p=0
     if not pdf_content:
         return "Por favor, carregue um PDF primeiro."
     prompt = f"Conteúdo do PDF: {pdf_content}\nPergunta: {question}\nResposta em português:"
+    print(f"Prompt: {prompt}")  # Adicionar log para depuração
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
         inputs.input_ids,