Update app.py

app.py

@@ -28,7 +28,7 @@ if "messages" not in st.session_state:
     st.session_state.messages = []
 
 if "overlay" not in st.session_state:
-    st.session_state.overlay = 'base'
+    st.session_state.overlay = Image.open(f'agent_function/base.png')
 
 with col1:
     messages_box = st.container(height=500)
@@ -48,5 +48,5 @@ with col1:
         st.session_state.messages.append({"role": "assistant", "content": response})
 
     with col2:
-        image = Image.open(f'agent_function/{st.session_state.overlay}.png')
-        st.image(image)
+        with st.spinner():
+            st.image(st.session_state.overlay)