Update src/app.py

src/app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import numpy as np
 import os
 import re
+import json
 import faiss
 import nltk
 
@@ -40,23 +41,27 @@ def log_activity(user, action, query, search_type):
         pass
 
 # ==============================
-# AUTHENTICATION
+# AUTHENTICATION (MULTI USER)
 # ==============================
 def login():
-    st.title("15 Types of AI-Advanced Multisearch")
+    st.title("🔐 Multi-User Login")
 
-    HF_USER = os.environ.get("USERNAME") or st.secrets.get("USERNAME")
-    HF_PASS = os.environ.get("PASSWORD") or st.secrets.get("PASSWORD")
+    users_json = os.environ.get("USERS") or st.secrets.get("USERS")
+
+    if not users_json:
+        st.error("⚠️ USERS not configured in Hugging Face secrets!")
+        return
+
+    users = json.loads(users_json)
 
     username = st.text_input("Username")
     password = st.text_input("Password", type="password")
 
     if st.button("Login"):
-        if not HF_USER or not HF_PASS:
-            st.error("⚠️ Secrets not configured!")
-        elif username == HF_USER and password == HF_PASS:
+        if username in users and users[username]["password"] == password:
             st.session_state["authenticated"] = True
             st.session_state["user"] = username
+            st.session_state["role"] = users[username]["role"]
             st.session_state["login_time"] = pd.Timestamp.now()
 
             log_activity(username, "Login Success", "-", "-")
@@ -79,6 +84,7 @@ st.set_page_config(page_title="Multi Search Engine", layout="wide")
 st.title("🔍 Advanced Multi-Search Product Engine")
 
 st.sidebar.success(f"👤 {st.session_state['user']}")
+st.sidebar.info(f"Role: {st.session_state['role']}")
 
 if st.sidebar.button("🚪 Logout"):
     log_activity(st.session_state["user"], "Logout", "-", "-")
@@ -159,7 +165,7 @@ def get_synonyms(word):
     return list(synonyms)
 
 # ==============================
-# SEARCH ENGINE
+# SEARCH ENGINE (15 TYPES)
 # ==============================
 def search_engine(query, mode, top_k):
 
@@ -284,10 +290,16 @@ if st.button("Search"):
             st.info("No results found")
 
 # ==============================
-# SIDEBAR LOGS
+# ADMIN LOG VIEW
 # ==============================
-st.sidebar.subheader("📊 Activity Logs")
-if os.path.exists(LOG_FILE):
-    st.sidebar.dataframe(pd.read_csv(LOG_FILE).tail(10))
-else:
-    st.sidebar.write("No logs yet")
+if st.session_state["role"] == "admin":
+    st.sidebar.subheader("📊 Activity Logs")
+
+    if os.path.exists(LOG_FILE):
+        log_df = pd.read_csv(LOG_FILE)
+        st.sidebar.dataframe(log_df.tail(10))
+
+        with open(LOG_FILE, "rb") as f:
+            st.sidebar.download_button("⬇ Download Logs", f, file_name="logs.csv")
+    else:
+        st.sidebar.write("No logs yet")