Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import pandas as pd
 
 st.set_page_config(layout="wide")
 st.markdown("<h1 style='text-align: center;'>حسن xxx دحوم</h1>", unsafe_allow_html=True)
@@ -36,6 +37,24 @@ with button_col1:
     else:
         st.button(premiumlabel, disabled=True)
 
+def process_premiums_file(uploaded_file):
+    # Read the CSV file into a DataFrame
+    df = pd.read_csv(uploaded_file)
+    
+    # Store the DataFrame in the session state
+    st.session_state['premiums_df'] = df
+
+    # Indicate that the CSV has been processed and we're ready to select columns
+    st.session_state['select_columns'] = True
+
+# Button to select premiums, only enabled if a file has been uploaded
+with button_col1:
+    if uploaded_file_premiums is not None:
+        if st.button(premiumlabel):
+            process_premiums_file(uploaded_file_premiums)
+    else:
+        st.button(premiumlabel, disabled=True)
+
 with button_col2:
     if uploaded_file_claims_recoveries is not None:
         st.button(incurredlabel)
@@ -49,3 +68,34 @@ with button_col3:
         st.button(claimscount)
     else:
         st.button(claimscount, disabled=True)
+
+
+# If the CSV has been processed, display the column selection UI
+if st.session_state.get('select_columns', False):
+    st.write("Please choose the 'lob' column and the values to filter by:")
+
+    # Dropdown to select the 'lob' column
+    lob_column = st.selectbox('Choose the column for "lob":', st.session_state['premiums_df'].columns)
+
+    # Multiselect to choose values from the 'lob' column to filter the DataFrame
+    selected_lob_values = st.multiselect('Choose values from the "lob" column:', 
+                                         st.session_state['premiums_df'][lob_column].unique())
+
+    # Filter the DataFrame based on selected 'lob' values
+    if selected_lob_values:
+        filtered_df = st.session_state['premiums_df'][st.session_state['premiums_df'][lob_column].isin(selected_lob_values)]
+        st.session_state['filtered_premiums_df'] = filtered_df
+
+        # Display other column selections after 'lob' has been selected and DataFrame has been filtered
+        st.write("Now, select the columns for other parameters:")
+        valuation_as_at = st.selectbox('Valuation As At:', filtered_df.columns)
+        quarter_bracket = st.selectbox('Quarter Bracket:', filtered_df.columns)
+        # Continue with other selections as required...
+
+        # Store the column selections in the session state
+        st.session_state['column_selections'] = {
+            'valuation_as_at': valuation_as_at,
+            'quarter_bracket': quarter_bracket,
+            # ... include other selections as well
+        }
+