Credentials added

LocalT_ESG_RAG_1.25.py

@@ -104,7 +104,51 @@ css = """
 """
 
 st.write(css, unsafe_allow_html=True)
-#st.sidebar.image('lt.png', width=250)
+
+def load_credentials(filepath):
+    with open(filepath, 'r') as file:
+        return json.load(file)
+
+# Load credentials from 'credentials.json'
+credentials = load_credentials('chroma_db_LT1/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()
 #------------- 
 llm=ChatGroq(groq_api_key=groq_api_key,
              model_name="Llama-3.1-70b-Versatile", temperature = 0.0, streaming=True)