Added run_app(), if name == main; moved get_data functions outside of main function

app.py

@@ -502,18 +502,6 @@ def nuc_monitor(usr_start_date, usr_end_date, photo_date, past_date):
     # -------------------------------------------------
     return 
 
-st.title("Nucmonitor App")
-
-# Get user input (e.g., dates)
-start_date = st.date_input("Start Date")
-end_date = st.date_input("End Date")
-photo_date = st.checkbox("Photodate")
-
-if photo_date == True:
-    past_date = st.date_input("Cutoff Date")
-else:
-    past_date = None
-
 @st.cache_data
 def get_rte_data(start_date, end_date):
     rte_data = get_unavailabilities(start_date, end_date)
@@ -533,49 +521,71 @@ def get_nucmonitor_data(start_date, end_date, photo_date, past_date):
     df = pd.read_json(response_nucmonitor)
     return df
 
-with st.form("nucmonitor_form"):
-    submitted = st.form_submit_button("Get Nucmonitor")
+def run_app():
+
+    st.title("Nucmonitor App")
+
+    # Get user input (e.g., dates)
+    start_date = st.date_input("Start Date")
+    end_date = st.date_input("End Date")
+    photo_date = st.checkbox("Photodate")
 
-if submitted:
-    df = get_nucmonitor_data(start_date, end_date, photo_date, past_date)
-    st.sidebar.write("FILTERS")
-    st.write("Data received from Flask:")
+    if photo_date == True:
+        past_date = st.date_input("Cutoff Date")
+    else:
+        past_date = None
+
+    with st.form("nucmonitor_form"):
+        submitted = st.form_submit_button("Get Nucmonitor")
 
-    st.write(df)  # Display DataFrame
+    if not submitted:
+        pass
+
+    else:
+        st.write("Data received from Flask:")
+        df = get_nucmonitor_data(start_date, end_date, photo_date, past_date)
+        st.sidebar.write("FILTERS")
 
-    # Create a line chart using Streamlit
-    st.title("Power Plant Data Visualization")
-    df1 = df.iloc[:-1, :-1]
-    # Create a line chart using Streamlit
-    st.line_chart(df1)
+        st.write(df)  # Display DataFrame
+
+        # Create a line chart using Streamlit
+        st.title("Power Plant Data Visualization")
+        df1 = df.iloc[:-1, :-1]
+        # Create a line chart using Streamlit
+        st.line_chart(df1)
+
+        # ---------------------------------------------------
+        #######################################################
+        #################### FILTERS ##########################
+        #######################################################
+        # if df is not None:
 
-    if df is not None:
         st.title("Data Filters")
         df_columns_lst = df.columns.tolist()
         print(df_columns_lst)
-        # Select columns for display
 
+        # Select columns for display
         selected_columns = st.sidebar.multiselect("Select Columns to Display", 
             options=df_columns_lst,
             default=df_columns_lst,
             key=None
             )
-        st.sidebar.write('Selected plants:', selected_columns)
-
 
+        st.sidebar.write('Selected plants:', selected_columns)
 
         # Filter rows by checking column 0
         # filter_row_contains_value = st.sidebar.checkbox("Filter Rows by Date", False)
 
-        filtered_df = df.copy()
+        # filtered_df = df.copy()
         print("filtered_df = df.copy()")
-        filtered_df = filtered_df[selected_columns]
+        filtered_by_plant = df.copy()[selected_columns]
         print("filtered_df = filtered_df[selected_columns]")
+        
         # if filter_row_contains_value:
         #     filtered_df = filtered_df[filtered_df.iloc[:, 0].astype(str).str.contains('0', case=False, na=False)]
 
         st.write("Filtered Data:")
-        st.write(filtered_df)
+        st.write(filtered_by_plant)
         print("st.write(filtered_df)")
         
         # Add a download button
@@ -602,4 +612,7 @@ if submitted:
             data=excel_buffer,
             file_name=f"nucmonitor_data_{current_year}-{current_month}-{current_day}-h{current_hour}m{current_minute}s{current_second}.xlsx",
             mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-        )
+        )
+
+if __name__ == '__main__':
+    run_app()