Update app.py

app.py

@@ -81,7 +81,7 @@ if 'messages' not in st.session_state:
     st.session_state.messages = [{'role': 'assistant', "content": 'Hello! Upload a PDF and ask me anything about its content.'}]
 
 for message in st.session_state.messages:
-    with st.chat_message(message['role'], avatar=icon[message['role']]):
+    with st.chat_message(message['role'], avatar=icons[message['role']]):
         st.write(message['content'])
 
 with st.sidebar:
@@ -108,7 +108,7 @@ if user_prompt and uploaded_file:
         st.write(user_prompt)
 
 if st.session_state.messages[-1]["role"] != "assistant":
-    with st.chat_message("assistant"):
+    with st.chat_message("assistant", avatar="👽"):
         response = handle_query(user_prompt)
         full_response = st.write(response)
     message = {"role": "assistant", "content": full_response}