Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import streamlit as st
 from transformers import pipeline
 import re
-from PyPDF2 import PdfFileReader
+from PyPDF2 import PdfReader
 
 # Function to truncate text to the nearest word boundary
 def truncate_to_word_boundary(text, max_words=100):
@@ -19,9 +19,9 @@ def question_answering(question, text):
     return answer
 
 def main():
-    st.title("Question Answering on an Uploaded File")
+    st.title("Question Answering on Uploaded Files")
 
-    uploaded_file = st.file_uploader("Upload a file:", type=["pdf", "txt"]) # , "docx", "csv", "json"
+    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:
@@ -30,11 +30,10 @@ def main():
 
         if file_extension == ".pdf":
             # Handle PDF files using PyPDF2
-            pdf_reader = PdfFileReader(uploaded_file)
+            pdf_reader = PdfReader(uploaded_file)
             pdf_text = ""
-            for page_num in range(pdf_reader.getNumPages()):
-                pdf_page = pdf_reader.getPage(page_num)
-                pdf_text += pdf_page.extractText()
+            for pdf_page in pdf_reader.pages:
+                pdf_text += pdf_page.extract_text()
 
             # Perform question-answering
             answer = question_answering(question, pdf_text)