Vihang28 commited on
Commit
faa3c94
1 Parent(s): 641ffd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -5,16 +5,13 @@ from io import BytesIO
5
 
6
 
7
  def search_pdf(pdf_file, search_term):
8
- with BytesIO(pdf_file.read()) as file:
9
- reader = PyPDF2.PdfReader(file)
10
- num_pages = len(reader.pages)
11
- search_results = []
12
-
13
- for page_num in range(num_pages):
14
- page = reader.pages[page_num]
15
- text = page.extract_text()
16
- if search_term.lower() in text.lower():
17
- search_results.append((page_num + 1, text))
18
  return search_results
19
 
20
  def final_result(pdf_file, search_term):
 
5
 
6
 
7
  def search_pdf(pdf_file, search_term):
8
+ search_results = []
9
+ doc = fitz.open(pdf_file)
10
+ for page_num in range(doc.page_count):
11
+ page = doc.load_page(page_num)
12
+ text = page.get_text()
13
+ if search_term.lower() in text.lower():
14
+ search_results.append((page_num + 1, text))
 
 
 
15
  return search_results
16
 
17
  def final_result(pdf_file, search_term):