Vihang28 commited on
Commit
e3bed55
·
verified ·
1 Parent(s): b7d43df

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
app.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import PyPDF2
3
+
4
+ def search_pdf(pdf_file, search_term):
5
+ with open(pdf_file, 'rb') as file:
6
+ reader = PyPDF2.PdfReader(file)
7
+ num_pages = len(reader.pages)
8
+ search_results = []
9
+
10
+ for page_num in range(num_pages):
11
+ page = reader.pages[page_num]
12
+ text = page.extract_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):
18
+ results = search_pdf(pdf_file, search_term)
19
+ output_text = ""
20
+ if results:
21
+ for page_num, text in results:
22
+ # output_text += f"Found \033[1m{search_term}\033[0m on page {page_num}:\n{text}\n\n"
23
+ output_text += f"Found '{search_term}' on page {page_num}:\n{text}\n\n"
24
+ else:
25
+ output_text = f"No results found for '{search_term}'."
26
+ return output_text
27
+
28
+
29
+ st.markdown("<h3 style='text-align:center; font-size:24px;'>Search in PDF</h3>", unsafe_allow_html=True)
30
+
31
+ input_file = st.text_input(label="Upload .pdf File",value=0)
32
+ search_term = st.text_input(label="Enter Search-term",value=0)
33
+ all_data = st.button("Submit")
34
+ if all_data:
35
+ result = final_result(input_file,search_term)
36
+ st.success(result)