Update src/streamlit_app.py

src/streamlit_app.py

@@ -92,15 +92,29 @@ st.set_page_config(page_title="Campus Helper Bot", page_icon="🤖")
 st.title("🤖 Campus Helper Bot")
 st.caption("Your AI-powered guide, now connected to Notion!")
 
+# --- 1. Fetch Notion data first ---
+notion_data = fetch_notion_database()
+
+# If Notion failed to load or is empty, stop the app
+if not notion_data:
+    st.warning("⚠️ Notion database is empty or failed to load. Please check your API keys and database.")
+    st.stop()
+
+# --- 2. Initialize chat session state ---
 if "messages" not in st.session_state:
     st.session_state.messages = [
-        {"role": "assistant", "content": "Hello! I'm now connected to a live Notion database. How can I help?"}
+        {
+            "role": "assistant",
+            "content": "Hello! How can I help you today?"  # You can change this freely now
+        }
     ]
 
+# --- 3. Display existing messages ---
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
+# --- 4. Handle new user input ---
 if prompt := st.chat_input("Ask about fee deadlines, scholarships, etc."):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
@@ -111,4 +125,4 @@ if prompt := st.chat_input("Ask about fee deadlines, scholarships, etc."):
             context = get_context(prompt)
             response = generate_response(prompt, context)
             st.markdown(response)
-            st.session_state.messages.append({"role": "assistant", "content": response})
+            st.session_state.messages.append({"role": "assistant", "content": response})