chai

app.py

@@ -46,7 +46,8 @@ def main():
     if user_prompt:
         st.chat_message("user").markdown(user_prompt)
         st.session_state.messages.append({"role": "user", "content": user_prompt})
-        response = get_response(llm, user_prompt)
+        with st.spinner('Thinking...'):
+            response = get_response(llm, user_prompt)
         st.chat_message("bot").markdown(response)
         st.session_state.messages.append({"role": "bot", "content": response})