Update app.py

app.py

@@ -15,8 +15,28 @@ db_name = os.environ.get("MONGO_DB", "grant_docs")
 mongo_uri = f"mongodb+srv://{user}:{password}@{cluster}/{db_name}?retryWrites=true&w=majority&tls=true&tlsAllowInvalidCertificates=true"
 client = MongoClient(mongo_uri, tls=True, tlsAllowInvalidCertificates=True, serverSelectionTimeoutMS=20000)
 db = client[db_name]
-
+def gate_ui():
+    APP_PASSWORD=st.secrets.get("APP_PASSWORD", os.getenv("APP_PASSWORD")).strip()
+    if "authed" not in st.session_state:
+         st.session_state.authed = False
+
+    if not APP_PASSWORD:
+        st.session_state.authed = True
+        return True
+    if st.session_state.authed:
+        return True
+    st.title("🔒 Grant Buddy Login")
+    pwd=st.text_input("Enter password", type=password)
+    if st.button("Login"):
+        if pwd==APP_PASSWORD:
+            st.session_state.authed=True
+            st.rerun()
+        else:
+            st.error("Incorrect password.")
+    return False
 # === Streamlit UI ===
+if not gate_ui():
+    return
 st.set_page_config(page_title="Doc Chunker", layout="wide")
 st.title("📄 Document Chunker & Uploader")