update: Implement user authentication with credential loading and session management

app.py

@@ -7,6 +7,20 @@ from datasets import Dataset, load_dataset
 from huggingface_hub import login
 import os
 
+# Load credentials from environment variables or a secure source
+def load_credentials():
+    # Example: Load credentials from environment variables
+    credentials = {}
+    for i in range(3):  # Assuming you have 40 credentials
+        username = os.environ.get(f"LOGIN_{i}")
+        password = os.environ.get(f"PASSWORD_{i}")
+        if username and password:
+            credentials[username] = password
+    return credentials
+
+# Authentication function
+def authenticate(username, password, credentials):
+    return credentials.get(username) == password
 
 def load_data(database_file):
     df = pd.read_parquet(database_file)
@@ -55,6 +69,29 @@ def save_reactions_to_dataset(user_type, query, results):
 def main():
     st.title("Semantic Text Retrieval Evaluation Interface")
 
+    # Load credentials
+    credentials = load_credentials()
+
+    # Check if user is authenticated
+    if 'authenticated' not in st.session_state:
+        st.session_state.authenticated = False
+
+    if not st.session_state.authenticated:
+        st.sidebar.title("Login")
+        username = st.sidebar.text_input("Username")
+        password = st.sidebar.text_input("Password", type="password")
+
+        if st.sidebar.button("Login"):
+            if authenticate(username, password, credentials):
+                st.session_state.authenticated = True
+                st.sidebar.success("Logged in successfully!")
+            else:
+                st.sidebar.error("Invalid username or password")
+
+        if not st.session_state.authenticated:
+            st.warning("Please login to access the application.")
+            return
+
     # Initialize session state variables
     if "search_performed" not in st.session_state:
         st.session_state.search_performed = False