- changing UI, reading users from source campaign

app.py

@@ -6,6 +6,9 @@ from snowflake.snowpark import Session
 from bs4 import BeautifulSoup
 from Messaging_system.Permes import Permes
 from dotenv import load_dotenv
+
+from Messaging_system.SnowFlakeConnection import SnowFlakeConn
+
 load_dotenv()
 
 
@@ -77,7 +80,7 @@ if not st.session_state.get("authenticated", False):
 def init_state() -> None:
     defaults = dict(
         involve_recsys_result=False, personalization=False,
-        involve_last_interaction=False,
+        involve_last_interaction=False, session=None,
         valid_instructions="",
         invalid_instructions="",
         messaging_type="push",
@@ -136,26 +139,37 @@ st.markdown(
 # ──────────────────────────────────────────────────────────────────────────────
 init_state()
 with st.sidebar:
-    # we will read this from snowflake in future
-    uploaded_file = "Data/Singeo_Camp.csv"
-    if uploaded_file:
-        st.session_state.data = load_data(uploaded_file)
+
+    # ─ Brand
+    st.selectbox(
+        "Select a Brand to Start! *",
+        ["drumeo", "pianote", "guitareo", "singeo"],
+        key="brand",
+    )
+
+    if st.session_state.brand is not None and st.session_state.session is None:
+        # ─ build Snowflake session
+        conn = dict(
+            user=get_credential("snowflake_user"),
+            password=get_credential("snowflake_password"),
+            account=get_credential("snowflake_account"),
+            role=get_credential("snowflake_role"),
+            database=get_credential("snowflake_database"),
+            warehouse=get_credential("snowflake_warehouse"),
+            schema=get_credential("snowflake_schema")
+        )
+        st.session_state.session = Session.builder.configs(conn).create()
+
+        snowflake = SnowFlakeConn(session=st.session_state.session, brand=st.session_state.brand)
+        st.session_state.data = snowflake.get_users_in_campaign(brand=st.session_state.brand)
         st.success("File loaded!")
 
-    st.markdown("---")
+        st.markdown("---")
 
     if st.session_state.data is not None:
-        id_col = "user_id"
+        id_col = "email"
         st.session_state.identifier_column = id_col
 
-
-        # ─ Brand
-        st.selectbox(
-            "Brand *",
-            ["drumeo", "pianote", "guitareo", "singeo"],
-            key="brand",
-        )
-
         st.session_state.config = load_config_("Config_files/message_system_config.json")
 
         # ─ Brand
@@ -219,7 +233,7 @@ with tab0:
     if st.session_state.data is not None:
         st.dataframe(st.session_state.data.head(100))
     else:
-        st.info("Upload a CSV to preview it here.")
+        st.info("Select a brand to preview the data here.")
 
 
 # ------------------------------------------------------------------ TAB 2 ---#
@@ -230,18 +244,6 @@ with tab2:
 
         if st.session_state.get("generate_clicked", False):
 
-            # ─ build Snowflake session
-            conn = dict(
-                user=get_credential("snowflake_user"),
-                password=get_credential("snowflake_password"),
-                account=get_credential("snowflake_account"),
-                role=get_credential("snowflake_role"),
-                database=get_credential("snowflake_database"),
-                warehouse=get_credential("snowflake_warehouse"),
-                schema=get_credential("snowflake_schema")
-            )
-            session = Session.builder.configs(conn).create()
-
             # ─ prepare parameters
             st.session_state.messaging_mode = (
                 "recsys_result" if st.session_state.include_recommendation
@@ -273,7 +275,7 @@ with tab2:
 
             permes = Permes()
             df_msg = permes.create_personalize_messages(
-                session=session,
+                session=st.session_state.session,
                 users=st.session_state.data,
                 brand=st.session_state.brand,
                 segment_info=st.session_state.segment_info,