Update app.py

app.py

@@ -78,7 +78,7 @@ with col2:
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
-    avatar = "🤖" if message["role"] == "assistant" else "🕺"
+    avatar = "🧠" if message["role"] == "assistant" else "❓"
     with st.chat_message(message["role"], avatar=avatar):
         st.markdown(message["content"])
 
@@ -91,7 +91,7 @@ def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
 if prompt := st.chat_input("Enter your prompt here..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    with st.chat_message("user", avatar="🕺"):  
+    with st.chat_message("user", avatar="❓"):  
         st.markdown(prompt)
 
     # Fetch response from Groq API
@@ -107,7 +107,7 @@ if prompt := st.chat_input("Enter your prompt here..."):
         )
 
         # Use the generator function with st.write_stream
-        with st.chat_message("assistant", avatar="🤖"):
+        with st.chat_message("assistant", avatar="🧠"):
             chat_responses_generator = generate_chat_responses(chat_completion)
             full_response = st.write_stream(chat_responses_generator)
     except Exception as e: