Update app.py

app.py

@@ -10,11 +10,17 @@ icons = {"assistant": "./Snowflake_Logomark_blue.svg", "user": "⛷️"}
 st.set_page_config(page_title="Snowflake Arctic Enu")
 
 @st.cache_resource
-def create_session():
-    return Session.builder.configs(st.secrets.snowflake).create()
-
-session = create_session()
-st.success("Connected to Snowflake!")
+def connect_snowflake():
+    """
+    Function establish the snowflake connections
+    @st.cache_resource --> run the function only once
+    :return:
+    """
+    try:
+        print("Connecting to snowflake")
+        return snowflake.connector.connect(**st.secrets["snowflake"])
+    except Exception as snf_error:
+        print(snf_error)
 
 # Replicate Credentials
 with st.sidebar: