Update Images

src/streamlit_app.py

@@ -22,7 +22,7 @@ if "messages" not in st.session_state:
 
 # default message
 if not st.session_state.messages:
-    with st.chat_message("assistant", avatar="images/ai.png"):
+    with st.chat_message("assistant", avatar="src/images/ai.png"):
         st.markdown("👋 **Hi! I can help you think, code, or search real-time info. Just ask!**")
 
 # Tool name map
@@ -36,8 +36,8 @@ tool_mapping = {
 # display chat messages from history at every rerun  
 for message in st.session_state.messages:
     avatar_path = (
-        "images/person_15454011.png" if message["role"] == "user"
-        else "images/ai.png"
+        "src/images/person_15454011.png" if message["role"] == "user"
+        else "src/images/ai.png"
     )
     with st.chat_message(message["role"], avatar=avatar_path):
         st.markdown(message["content"])
@@ -62,14 +62,14 @@ selected_tool_name = tool_mapping[st.session_state.selected_tool]
 if prompt:
     try:
         # show user message
-        with st.chat_message("user", avatar="images/person_15454011.png"):
+        with st.chat_message("user", avatar="src/images/person_15454011.png"):
             st.markdown(prompt)
         # Save user message
         st.session_state.messages.append({"role": "user", "content": prompt})
 
         # stream assistant response
         full_response = ""
-        with st.chat_message("assistant", avatar="images/ai.png"):
+        with st.chat_message("assistant", avatar="src/images/ai.png"):
             response_stream = st.session_state.chat.send_message_stream(prompt)
             response_placeholder = st.empty()