Update app.py

app.py

@@ -101,7 +101,7 @@ def chat_interface():
 
     if user_input:
         # Add user message to chat history
-        st.session_state.chat_history.append({"role": "user", "content": user_input})
+        st.session_state.chat_history.append({"role": "user", "content": user_input, , avatar="👤"})
 
         # Prepare the prompt
         prompt = prepare_prompt(user_input, st.session_state.chat_history)
@@ -132,12 +132,12 @@ def chat_interface():
         }
 
        # Generate and stream response
-        with st.chat_message("assistant"):
+        with st.chat_message("Jimmy", avatar="😒"):
             stream = generate_response(watsonx_llm, prompt_data, params)
             response = st.write_stream(stream)
 
         # Add AI response to chat history
-        st.session_state.chat_history.append({"role": "assistant", "content": response})
+        st.session_state.chat_history.append({"role": "Jimmy", "content": response, avatar="😒"})
 
 def main():
     initialize_session_state()