Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 import streamlit as st
 from transformers import pipeline
 import re
+from PyPDF2 import PdfFileReader
 
 # Function to truncate text to the nearest word boundary
 def truncate_to_word_boundary(text, max_words=100):
@@ -20,7 +21,7 @@ def question_answering(question, text):
 def main():
     st.title("Question Answering on an Uploaded File")
 
-    uploaded_file = st.file_uploader("Upload a file:", type=["pdf", "txt"]) # , "docx", "csv", "json", "txt"
+    uploaded_file = st.file_uploader("Upload a file:", type=["pdf", "txt", "docx", "csv", "json", "txt"])
     question = st.text_input("Ask your question:")
 
     if st.button("Answer") and uploaded_file is not None:
@@ -28,13 +29,12 @@ def main():
         file_contents = uploaded_file.read()
 
         if file_extension == ".pdf":
-            # Handle PDF files
-            import fitz  # PyMuPDF for parsing PDF
-
+            # Handle PDF files using PyPDF2
+            pdf_reader = PdfFileReader(uploaded_file)
             pdf_text = ""
-            with fitz.open(stream=uploaded_file, filetype="pdf") as pdf_document:
-                for page in pdf_document:
-                    pdf_text += page.get_text()
+            for page_num in range(pdf_reader.getNumPages()):
+                pdf_page = pdf_reader.getPage(page_num)
+                pdf_text += pdf_page.extractText()
 
             # Perform question-answering
             answer = question_answering(question, pdf_text)