Update app.py

app.py

@@ -70,12 +70,18 @@ with container:
             st.session_state['messages'].append(model_response)
             
 
+#            with response_container:
+#                for i in range(len(st.session_state['messages'])):
+#                        if (i % 2) == 0:
+#                            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user')
+#                        else:
+#                            message(st.session_state['messages'][i], key=str(i) + '_AI')
+        
             with response_container:
                 for i in range(len(st.session_state['messages'])):
-                        if (i % 2) == 0:
-                            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user')
-                        else:
-                            message(st.session_state['messages'][i], key=str(i) + '_AI')
-        
-            
+                    if (i % 2) == 0:
+                        message(st.session_state['messages'][i], is_user=True, user_icon='<i class="fas fa-user-tie"></i>', key=str(i) + '_user')
+                    else:
+                        message(st.session_state['messages'][i], bot_icon='<i class="fas fa-robot"></i>', key=str(i) + '_AI')
+