Merge pull request #47 from theakash04/dev/akash

.gitignore

@@ -3,4 +3,5 @@
 __pycache__
 .venv
 notebook.ipynb
-.snow
+.snow
+.devcontainer

streamlit/app.py

@@ -13,13 +13,15 @@
 from snowflake.main import RAG
 import asyncio
 
+if "session" not in st.session_state:
+    sfConnect = SnowflakeConnector()
+    st.session_state.session = sfConnect.get_session()
 
-st.session_state.setdefault("sfConnect", SnowflakeConnector())
-st.session_state.setdefault("session", st.session_state.sfConnect.get_session())
-st.session_state.setdefault("root", Root(st.session_state.session))
-
-# Access variables
 session = st.session_state.session
+
+if "root" not in st.session_state:
+    st.session_state.root = Root(session)
+
 root = st.session_state.root
 
 # intialize chat history

utils/secret_loader.py

@@ -1,17 +1,22 @@
 import os
 import streamlit as st
+from dotenv import load_dotenv
 
 def get_secret(key: str):
-    """Fetch secret from Streamlit secrets or fallback to environment variables."""
-    secrets_file_path = os.path.join(os.getcwd(), ".streamlit", "secrets.toml")
-    if not os.path.exists(secrets_file_path):
-        return os.environ[key]
-    # Use st.secrets only if the file exists and the key is present
+    """Fetch secret from environment variables or fallback to Streamlit secrets."""
+    secret_value = os.environ[key]
+    if secret_value:
+        return secret_value
+
+    # if not found in env
     try:
         if key in st.secrets:
             return st.secrets[key]
     except FileNotFoundError:
-        pass  # Silently skip st.secrets if secrets.toml doesn't exist
+        pass
+
+    return None
 
+
 
 __all__ = ["get_secret"]