Update app.py

app.py

@@ -446,7 +446,35 @@ def rerank_with_topics(chunks, topics, alpha=0.2):
 
 
 # =================== Streamlit UI ===================
+def gate_ui():
+    # Get password from secrets (optional env fallback)
+    APP_PASSWORD = st.secrets.get("APP_PASSWORD", os.getenv("APP_PASSWORD", "")).strip()
+
+    # Persist auth in session
+    if "authed" not in st.session_state:
+        st.session_state.authed = False
+
+    if st.session_state.authed:
+        return True
+
+    st.title("🔒 Grant Buddy Login")
+    pwd = st.text_input("Enter password", type="password")
+    col1, col2 = st.columns([1,1])
+    with col1:
+        if st.button("Login"):
+            if APP_PASSWORD and pwd == APP_PASSWORD:
+                st.session_state.authed = True
+                st.experimental_rerun()
+            else:
+                st.error("Incorrect password.")
+    with col2:
+        if st.button("Forgot?"):
+            st.info("Contact the admin to reset the APP_PASSWORD secret.")
+    return False
+
 def main():
+    if not gate_uit:
+        return
     # st.set_page_config(page_title="Grant Buddy RAG", page_icon="🤖")
     st.title("🤖 Grant Buddy: Grant-Writing Assistant")
     USE_OPENAI = st.sidebar.checkbox("Use OpenAI (Costs Tokens)", value=False)
@@ -467,6 +495,7 @@ def main():
     
     # # retriever = init_vector_search().as_retriever(search_kwargs={"k": k_value, "score_threshold": score_threshold})
     retriever = init_vector_search().as_retriever()
+    vectorstore = init_vector_search()  
 
     # pre_k = k_value*4  # Retrieve more chunks first
     # context_docs = retriever.get_relevant_documents(query, k=pre_k)