Update app.py

app.py

@@ -6,10 +6,7 @@ import asyncio
 import aiohttp
 import os
 
-# Set wide layout
 st.set_page_config(layout="wide")
-
-# Global API key
 API_KEY = os.getenv("FMP_API_KEY")
 
 ########################################
@@ -57,7 +54,8 @@ async def fetch_8k_page(session, page, from_date, to_date, has_financial):
 
 async def fetch_all_8k(pages_to_fetch, from_date, to_date, has_financial):
     async with aiohttp.ClientSession() as session:
-        tasks = [fetch_8k_page(session, p, from_date, to_date, has_financial) for p in range(pages_to_fetch)]
+        tasks = [fetch_8k_page(session, p, from_date, to_date, has_financial)
+                 for p in range(pages_to_fetch)]
         results = []
         progress_bar = st.progress(0)
         completed = 0
@@ -88,7 +86,8 @@ async def fetch_company_page(session, symbol, filing_type, page):
 
 async def fetch_all_company(symbol, filing_type, pages_to_fetch):
     async with aiohttp.ClientSession() as session:
-        tasks = [fetch_company_page(session, symbol, filing_type, p) for p in range(pages_to_fetch)]
+        tasks = [fetch_company_page(session, symbol, filing_type, p)
+                 for p in range(pages_to_fetch)]
         results = []
         progress_bar = st.progress(0)
         completed = 0
@@ -112,7 +111,11 @@ def live_feed_page():
         Use the filters to narrow down the list by filing type, date range, and whether the filing is complete.
         """
     )
-    
+
+    # Initialize session state if not set.
+    if "live_feed_results" not in st.session_state:
+        st.session_state.live_feed_results = None
+
     with st.sidebar:
         with st.expander("Live Feed Filter Parameters", expanded=True):
             filing_type = st.selectbox(
@@ -136,25 +139,30 @@ def live_feed_page():
                 help="Check to show only filings marked as completed."
             )
         run_live = st.button("Run Live Feed Query")
-    
-    # Fixed backend parameter: limit is set to 500
-    limit = 500
 
+    limit = 500
     if run_live:
-        df = get_live_feed_data(limit, filing_type, from_date.isoformat(), to_date.isoformat(), is_done)
+        st.session_state.live_feed_results = get_live_feed_data(
+            limit, filing_type, from_date.isoformat(), to_date.isoformat(), is_done
+        )
+
+    if st.session_state.live_feed_results is not None:
         st.write("### Live SEC Filings Data")
-        st.dataframe(df, use_container_width=True)
+        st.dataframe(st.session_state.live_feed_results, use_container_width=True)
 
 def eight_k_page():
     st.title("8-K Filings Specialized")
     st.write(
         """
         This page displays 8-K filings from public companies.
-        8-K filings are reports companies file when major events occur, like mergers, acquisitions, or leadership changes.
-        Use the filter below to show only filings that include financial information if you prefer.
+        8-K filings are reports companies file when major events occur.
+        Use the filter below to show only filings that include financial information.
         """
     )
-    
+
+    if "eight_k_results" not in st.session_state:
+        st.session_state.eight_k_results = None
+
     with st.sidebar:
         with st.expander("8-K Filter Parameters", expanded=True):
             from_date = st.date_input(
@@ -174,29 +182,38 @@ def eight_k_page():
             )
             has_financial = has_financial_option.lower() == "true"
         run_8k = st.button("Run 8-K Query")
-    
-    pages_to_fetch = 15  # Fixed number of pages to iterate over
+
+    pages_to_fetch = 15
     if run_8k:
-        # Run async requests to fetch 8-K pages concurrently and update a progress bar.
-        results = asyncio.run(fetch_all_8k(pages_to_fetch, from_date.isoformat(), to_date.isoformat(), has_financial))
+        results = asyncio.run(
+            fetch_all_8k(pages_to_fetch, from_date.isoformat(), to_date.isoformat(), has_financial)
+        )
         dfs = [df for df in results if not df.empty]
         if dfs:
             df_all = pd.concat(dfs, ignore_index=True)
-            st.write("### 8-K Filings Data")
-            st.dataframe(df_all, use_container_width=True)
+            st.session_state.eight_k_results = df_all
         else:
+            st.session_state.eight_k_results = None
             st.warning("No data returned for the given parameters.")
 
+    if st.session_state.eight_k_results is not None:
+        st.write("### 8-K Filings Data")
+        st.dataframe(st.session_state.eight_k_results, use_container_width=True)
+
 def company_specific_page():
     st.title("Company Specific SEC Filings")
     st.write(
         """
         This page displays filings for a specific company.
-        SEC filings are official documents companies file with the government.
         Enter a company ticker (e.g., AAPL) and select a filing type (like 10-K or 10-Q) to view the reports.
         """
     )
-    
+
+    if "company_results" not in st.session_state:
+        st.session_state.company_results = None
+    if "company_symbol" not in st.session_state:
+        st.session_state.company_symbol = None
+
     with st.sidebar:
         with st.expander("Company Filings Parameters", expanded=True):
             symbol = st.text_input(
@@ -210,22 +227,28 @@ def company_specific_page():
                 help="Select the type of filing to retrieve."
             )
         run_company = st.button("Run Company Query")
-    
-    pages_to_fetch = 10  # Fixed number of pages to iterate over
 
+    pages_to_fetch = 10
     if run_company:
         if symbol.strip():
-            results = asyncio.run(fetch_all_company(symbol.upper(), filing_type, pages_to_fetch))
+            results = asyncio.run(
+                fetch_all_company(symbol.upper(), filing_type, pages_to_fetch)
+            )
             dfs = [df for df in results if not df.empty]
             if dfs:
                 df_all = pd.concat(dfs, ignore_index=True)
-                st.write(f"### SEC Filings for {symbol.upper()}")
-                st.dataframe(df_all, use_container_width=True)
+                st.session_state.company_results = df_all
+                st.session_state.company_symbol = symbol.upper()
             else:
+                st.session_state.company_results = None
                 st.warning("No filings found for the given parameters.")
         else:
             st.warning("Please enter a valid company ticker.")
 
+    if st.session_state.company_results is not None:
+        st.write(f"### SEC Filings for {st.session_state.company_symbol}")
+        st.dataframe(st.session_state.company_results, use_container_width=True)
+
 ########################################
 # MAIN APPLICATION
 ########################################
@@ -237,7 +260,7 @@ def main():
         ("Live Feed", "8-K Specialized", "Company Specific"),
         help="Navigate to the desired SEC filings page."
     )
-    
+
     if page_selection == "Live Feed":
         live_feed_page()
     elif page_selection == "8-K Specialized":
@@ -248,11 +271,10 @@ def main():
 if __name__ == "__main__":
     main()
 
-
 hide_streamlit_style = """
 <style>
 #MainMenu {visibility: hidden;}
 footer {visibility: hidden;}
 </style>
 """
-st.markdown(hide_streamlit_style, unsafe_allow_html=True)
+st.markdown(hide_streamlit_style, unsafe_allow_html=True)