msr2903 commited on
Commit
162a16c
1 Parent(s): bffe4e4

Update pages.py and section_extract.py

Browse files
__pycache__/pages.cpython-311.pyc CHANGED
Binary files a/__pycache__/pages.cpython-311.pyc and b/__pycache__/pages.cpython-311.pyc differ
 
__pycache__/section_extract.cpython-311.pyc CHANGED
Binary files a/__pycache__/section_extract.cpython-311.pyc and b/__pycache__/section_extract.cpython-311.pyc differ
 
pages.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- from section_extract import find_cover, find_underwriter, find_section
3
 
4
  def home():
5
  st.title("Prospectus Lens")
@@ -12,24 +12,13 @@ def cover():
12
  find_cover(uploaded_file=st.session_state.get("uploaded_file"))
13
 
14
  def underwriter():
15
- find_underwriter(
16
- uploaded_file=st.session_state.get("uploaded_file"),
17
- )
18
 
19
  def income_statement():
20
- find_section(
21
- uploaded_file=st.session_state.get("uploaded_file"),
22
- section_name="income_statement",
23
- )
24
 
25
  def balance_sheet():
26
- find_section(
27
- uploaded_file=st.session_state.get("uploaded_file"),
28
- section_name="balance_sheet",
29
- )
30
 
31
  def cash_flow():
32
- find_section(
33
- uploaded_file=st.session_state.get("uploaded_file"),
34
- section_name="cash_flow",
35
- )
 
1
  import streamlit as st
2
+ from section_extract import find_cover, find_underwriter, find_financial
3
 
4
  def home():
5
  st.title("Prospectus Lens")
 
12
  find_cover(uploaded_file=st.session_state.get("uploaded_file"))
13
 
14
  def underwriter():
15
+ find_underwriter(uploaded_file=st.session_state.get("uploaded_file"))
 
 
16
 
17
  def income_statement():
18
+ find_financial(uploaded_file=st.session_state.get("uploaded_file"), section_name="income_statement")
 
 
 
19
 
20
  def balance_sheet():
21
+ find_financial(uploaded_file=st.session_state.get("uploaded_file"), section_name="balance_sheet")
 
 
 
22
 
23
  def cash_flow():
24
+ find_financial(uploaded_file=st.session_state.get("uploaded_file"), section_name="cash_flow")
 
 
 
section_extract.py CHANGED
@@ -92,7 +92,7 @@ def find_underwriter(uploaded_file):
92
  st.warning("Please upload a PDF on the Home page first.")
93
 
94
 
95
- def find_section(uploaded_file, section_name):
96
  """
97
  Extracts and displays sections of a PDF based on keyword matches.
98
 
 
92
  st.warning("Please upload a PDF on the Home page first.")
93
 
94
 
95
+ def find_financial(uploaded_file, section_name):
96
  """
97
  Extracts and displays sections of a PDF based on keyword matches.
98