Update app.py

app.py

@@ -50,9 +50,14 @@ if uploaded_file:
     st.write("## Ai Skin Check Assitant")
     st.write("Have questions? Type your message below.")
 
-    # Display all previous messages
     for msg in st.session_state.messages:
-        st.chat_message(msg["role"]).write(msg["content"])
+        avatar_image = None  # Default avatar
+        if msg["role"] == "assistant":
+            avatar_image =   # Set the assistant's avatar image
+    
+        # Render the message with the appropriate avatar
+        st.chat_message(msg["role"], avatar="LOGO.png").write(msg["content"])
+        
 
     # User input
     user_input = st.chat_input("Type your message here...")