Spaces:
Running
Running
File size: 4,680 Bytes
b3397c2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import streamlit as st
from section_extract import find_cover, find_underwriter, find_financial
from streamlit_pdf_viewer import pdf_viewer
def process_sections():
if "processing" not in st.session_state or st.session_state.get("all_processed", False):
return
# Only process the current page if it's set and not processed
current_page = st.session_state.get("current_page")
current_page_needs_processing = False
if current_page and not st.session_state.get(current_page, None):
if not st.session_state["processing"].get(current_page, False):
current_page_needs_processing = True
key = current_page
if key == "cover_path":
st.session_state[key] = find_cover(st.session_state["uploaded_file"])
elif key == "underwriter_path":
st.session_state[key] = find_underwriter(st.session_state["uploaded_file"])
elif key == "income_statement_path":
st.session_state[key] = find_financial(st.session_state["uploaded_file"], "income_statement")
elif key == "balance_sheet_path":
st.session_state[key] = find_financial(st.session_state["uploaded_file"], "balance_sheet")
elif key == "cash_flow_path":
st.session_state[key] = find_financial(st.session_state["uploaded_file"], "cash_flow")
st.session_state["processing"][key] = True
st.rerun()
# Process remaining sections if not processing current page and not locked
if not current_page_needs_processing and not st.session_state.get("processing_lock", False):
sections_to_process = [
("cover_path", lambda: find_cover(st.session_state["uploaded_file"])),
("underwriter_path", lambda: find_underwriter(st.session_state["uploaded_file"])),
("income_statement_path", lambda: find_financial(st.session_state["uploaded_file"], "income_statement")),
("balance_sheet_path", lambda: find_financial(st.session_state["uploaded_file"], "balance_sheet")),
("cash_flow_path", lambda: find_financial(st.session_state["uploaded_file"], "cash_flow"))
]
# Store the section being processed
if "processing_section" not in st.session_state:
st.session_state["processing_section"] = None
for key, process_func in sections_to_process:
if not st.session_state["processing"].get(key, False):
# Check if we've switched pages during processing
if (st.session_state.get("processing_section") is not None and
st.session_state["processing_section"] != current_page):
st.session_state["processing_section"] = None
return # Stop processing if we've switched pages
st.session_state["processing_section"] = key
st.session_state[key] = process_func()
st.session_state["processing"][key] = True
st.session_state["processing_section"] = None
if not st.session_state.get("all_processed", False):
st.rerun() # Trigger another run to process next section
break
# Check if all sections are processed
st.session_state["all_processed"] = all(st.session_state["processing"].values())
def show_section(section_key):
"""Display the section if available, otherwise inform the user."""
# Check if we're switching to a different page
if st.session_state.get("current_page") != section_key:
# Reset processing state when switching pages
st.session_state["processing_lock"] = True
st.session_state["processing_section"] = None
# Update current page
st.session_state["current_page"] = section_key
temp_path = st.session_state.get(section_key)
if temp_path:
# If we have the section, allow background processing of other sections
st.session_state["processing_lock"] = False
pdf_viewer(temp_path)
# Continue processing remaining sections
if not st.session_state.get("all_processed", False):
process_sections()
else:
# Lock processing and process this section first
st.session_state["processing_lock"] = True
if not st.session_state["processing"].get(section_key, False):
st.info(f"Processing {section_key.replace('_path', '').replace('_', ' ').title()}...")
process_sections()
else:
st.warning(f"Could not process {section_key.replace('_path', '').replace('_', ' ')}.")
|