Log in functions included

app.py

@@ -106,6 +106,53 @@ css = """
 """
 
 st.write(css, unsafe_allow_html=True)
+
+#--------------
+def load_credentials(filepath):
+    with open(filepath, 'r') as file:
+        return json.load(file)
+
+# Load credentials from 'credentials.json'
+credentials = load_credentials('credentials.json')
+
+# Initialize session state if not already done
+if 'logged_in' not in st.session_state:
+    st.session_state.logged_in = False
+    st.session_state.username = ''
+
+# Function to handle login
+def login(username, password):
+    if username in credentials and credentials[username] == password:
+        st.session_state.logged_in = True
+        st.session_state.username = username
+        st.rerun()  # Rerun to reflect login state
+    else:
+        st.session_state.logged_in = False
+        st.session_state.username = ''
+        st.error("Invalid username or password.")
+
+# Function to handle logout
+def logout():
+    st.session_state.logged_in = False
+    st.session_state.username = ''
+    st.rerun()  # Rerun to reflect logout state
+
+# If not logged in, show login form
+if not st.session_state.logged_in:
+    st.sidebar.write("Login")
+    username = st.sidebar.text_input('Username')
+    password = st.sidebar.text_input('Password', type='password')
+    if st.sidebar.button('Login'):
+        login(username, password)
+    # Stop the script here if the user is not logged in
+    st.stop()
+
+# If logged in, show logout button and main content
+if st.session_state.logged_in:
+    st.sidebar.write(f"Welcome, {st.session_state.username}!")
+    if st.sidebar.button('Logout'):
+        logout()
+
 st.sidebar.image('StratXcel.png', width=150)
 #------------- 
 llm=ChatGroq(groq_api_key=groq_api_key,