feat: add spinner

app.py

@@ -60,18 +60,19 @@ def initialize_session_state():
 
 
 def on_click_callback():
-    with get_openai_callback() as cb:
-        human_prompt = st.session_state.human_prompt
-        llm_response = st.session_state.conversation(human_prompt)
-        st.session_state.history.append(Message("human", human_prompt))
-        st.session_state.history.append(
-            Message(
-                "ai", llm_response["answer"], documents=llm_response["source_documents"]
+    with st.spinner('Wait for it...'):
+        with get_openai_callback() as cb:
+            human_prompt = st.session_state.human_prompt
+            llm_response = st.session_state.conversation(human_prompt)
+            st.session_state.history.append(Message("human", human_prompt))
+            st.session_state.history.append(
+                Message(
+                    "ai", llm_response["answer"], documents=llm_response["source_documents"]
+                )
             )
-        )
-        st.session_state.token_count += cb.total_tokens
-        # history_id = insert_chat_history(conn, human_prompt, llm_response["answer"])
-        # insert_chat_history_articles(conn, history_id, llm_response["source_documents"])
+            st.session_state.token_count += cb.total_tokens
+            # history_id = insert_chat_history(conn, human_prompt, llm_response["answer"])
+            # insert_chat_history_articles(conn, history_id, llm_response["source_documents"])
 
 
 load_css()