.

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import random
 import time
 from index import build_index, build_service_context
-from loader import load_documents 
+from loader import load_documents
 
 st.title("SAIRA")
 
@@ -14,7 +14,11 @@ def load_context():
     return build_service_context()
 
 context = load_context()
+
+print("Loading documents and building index...")
 index = load_docs_and_build_index(context)
+print("Done!")
+
 query_engine = index.as_query_engine(streaming=True)
 
 # Initialize chat history
@@ -53,4 +57,4 @@ if prompt := st.chat_input("What is up?"):
             message_placeholder.markdown(full_response + "▌")
         message_placeholder.markdown(full_response)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": full_response})
+    st.session_state.messages.append({"role": "assistant", "content": full_response})