Update src/streamlit_app.py

src/streamlit_app.py

@@ -76,8 +76,10 @@ st.title("🤖 cAsh Robo-Advisor")
 st.markdown("---")
 
 # Initialize Chat History
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+if "messages" not in st.session_state.keys(): # Initialize the chat messages history
+    st.session_state.messages = [
+        {"role": "assistant", "content": "Ask me a question about investing in Pokemon Cards!"}
+    ]
 
 # Sidebar Examples
 st.sidebar.header("Example Queries")
@@ -93,16 +95,22 @@ for ex in examples:
     if st.sidebar.button(ex):
         asyncio.run(process_chat(ex))
 
+if prompt := st.chat_input("Your question"): # Prompt for user input and save to chat history
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
 # --- 5. Main Chat Interface ---
 # Display existing messages
-for message in st.session_state.messages:
+for message in st.session_state.messages: # Display the prior chat messages
     with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+        st.write(message["content"])
 
 # User Input
-if prompt := st.chat_input("Ask your Pokemon Quants advisor..."):
-    # Clear the "user" UI immediately then run logic
-    with st.chat_message("user"):
-        st.markdown(prompt)
+if st.session_state.messages[-1]["role"] != "assistant":
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            response = st.session_state.chat_engine.chat(prompt)
+            st.write(response.response)
+            message = {"role": "assistant", "content": response.response}
+            st.session_state.messages.append(message) # Add response to message history
     
     asyncio.run(process_chat(prompt))