Update app.py

app.py

@@ -4,6 +4,7 @@ import pdfplumber
 import transformers
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForQuestionAnswering
+import fitz
 
 tokenizer = AutoTokenizer.from_pretrained("aware-ai/bart-squadv2")
 model = AutoModelForQuestionAnswering.from_pretrained("aware-ai/bart-squadv2")
@@ -43,12 +44,15 @@ def answer_question(question, pdf_text):
 
 import pdfplumber
 
+import fitz  # PyMuPDF
+
 def extract_text_from_pdf(pdf_path):
     text = ""
-    with pdfplumber.open(pdf_path) as pdf:
-        for page in pdf.pages:
-            text += page.extract_text()
-    return text   
+    pdf = fitz.open(pdf_path)
+    for page_num in range(len(pdf)):
+        page = pdf[page_num]
+        text += page.get_text()
+    return text  
 
 
 def chat_with_pdf(question, pdf):