- updating requirements.txt

messaging_main_test.py

@@ -7,6 +7,8 @@ from snowflake.snowpark import Session
 from bs4 import BeautifulSoup
 from Messaging_system.Permes import Permes
 import streamlit as st
+
+from Messaging_system.SnowFlakeConnection import SnowFlakeConn
 from Messaging_system.context_validator import Validator
 
 
@@ -78,12 +80,12 @@ def filter_validated_users(users):
 # --------------------------------------------------------------
 if __name__ == "__main__":
     # path to sample data
-    path = "Data/Singeo_Camp.csv"
-
+    # path = "Data/Singeo_Camp.csv"
+    path = "Data/Test_users.csv"
 
     # loading sample data
     users = pd.read_csv(path)
-    users = clean_html_tags(users)
+    # users = clean_html_tags(users)
 
     config_file_path = 'Config_files/message_system_config.json'
     config_file = load_config_(config_file_path)
@@ -100,6 +102,7 @@ if __name__ == "__main__":
         schema=get_credential("snowflake_schema")
     )
 
+
     # --------------------
     # #Do we need to validate user-generated context?
     # user_generated_context = True
@@ -112,7 +115,10 @@ if __name__ == "__main__":
     session = Session.builder.configs(conn).create()
 
     brand = "singeo"
-    identifier_column = "user_id"
+    identifier_column = "email"
+
+    snowflake = SnowFlakeConn(session=session, brand=brand)
+    # users = snowflake.get_users_in_campaign(brand=brand)
 
     segment_info = """Student who haven't practiced for a few days"""
 
@@ -194,7 +200,7 @@ Message: These lessons are curated just for you. Start singing today!
     # o3-mini o1-mini o4-mini o1
 
     users_message = permes.create_personalize_messages(session=session,
-                                                model="gemini-2.5-flash-lite-preview-06-17",
+                                                model="gpt-4.1-mini",
                                                 users=users,
                                                 brand=brand,
                                                 config_file=config_file,