Update app.py

app.py

@@ -78,12 +78,9 @@ with col2:
 
 # Display chat messages from history on app rerun
 for message in st.session_state.messages:
-    if hasattr(message, "role"):
-        avatar = "🤖" if message.role == "assistant" else "🕺"
-        with st.chat_message(message.role, avatar=avatar):
-            st.markdown(message.content)
-    else:
-        st.warning("Message object does not have the 'role' attribute.")
+    avatar = "🤖" if message["role"] == "assistant" else "🕺"
+    with st.chat_message(message["role"], avatar=avatar):
+        st.markdown(message["content"])
 
 def generate_chat_responses(chat_completion) -> Generator[str, None, None]:
     """Yield chat response content from the Groq API response."""