Update app.py

app.py

@@ -94,14 +94,14 @@ def generate_response(watsonx_llm, prompt_data, params):
         yield chunk
 
 def chat_interface():
-    st.title("Jimmy")
+    st.subheader("Jimmy")
 
     # User input
     user_input = st.chat_input("You:", key="user_input")
 
     if user_input:
         # Add user message to chat history
-        st.session_state.chat_history.append({"role": "user", "content": user_input, avatar="👤"})
+        st.session_state.chat_history.append({"role": "user", "content": user_input})
 
         # Prepare the prompt
         prompt = prepare_prompt(user_input, st.session_state.chat_history)
@@ -137,7 +137,7 @@ def chat_interface():
             response = st.write_stream(stream)
 
         # Add AI response to chat history
-        st.session_state.chat_history.append({"role": "Jimmy", "content": response, avatar="😒"})
+        st.session_state.chat_history.append({"role": "Jimmy", "content": response)
 
 def main():
     initialize_session_state()