Update app.py

app.py

@@ -10,6 +10,8 @@ import base64
 # Load environment variables
 load_dotenv()
 
+icons = {"assistant": "./assistant.jpeg", "user": "⛷️"}
+
 # Configure the Llama index settings
 Settings.llm = HuggingFaceInferenceAPI(
     model_name="google/gemma-1.1-7b-it",
@@ -86,7 +88,7 @@ if 'messages' not in st.session_state:
 # Display or clear chat messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
-        st.write(message["content"])
+        st.write(message["content"],avatar=icons[message["role"]])
         
 with st.sidebar:
     st.title("Menu:")