Update app.py

app.py

@@ -33,9 +33,8 @@ def load_pdf(pdf_file):
 def answer_question(question, max_length=200, temperature=0.7, top_k=50, top_p=0.95):
     global pdf_content
     if not pdf_content:
-        return "Por favor, carregue um PDF primeiro."
+        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,
@@ -46,7 +45,7 @@ def answer_question(question, max_length=200, temperature=0.7, top_k=50, top_p=0
         num_return_sequences=1
     )
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response
+    return response, prompt
 
 # Interface do Gradio para carregar PDF e fazer perguntas
 pdf_loader = gr.Interface(
@@ -59,7 +58,7 @@ pdf_loader = gr.Interface(
 question_answerer = gr.Interface(
     fn=answer_question,
     inputs=gr.Textbox(lines=2, label="Pergunta"),
-    outputs="text",
+    outputs=[gr.Textbox(label="Resposta"), gr.Textbox(label="Prompt usado")],
     title="Perguntas sobre o PDF"
 )