Update app.py

app.py

@@ -1,13 +1,16 @@
 import streamlit as st
 # import PyPDF2
 import fitz
-from io import BytesIO
+from io
 
 
 
 def search_pdf(pdf_file, search_term):
     search_results = []
-    doc = fitz.open(pdf_file)
+    if isinstance(pdf_file, io.BytesIO):  # Handling Streamlit case
+        doc = fitz.open(stream=pdf_file, filetype="pdf")
+    else:  # Handling local file case
+        doc = fitz.open(pdf_file)
     for page_num in range(doc.page_count):
         page = doc.load_page(page_num)
         text = page.get_text()