Update tribunal_2.py

tribunal_2.py

@@ -298,7 +298,7 @@ def main():
         
         with col1:
             st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
-            st.subheader(genparam.BOT_1_NAME)
+            st.subheader(f"{genparam.BOT_1_AVATAR} {genparam.BOT_1_NAME}")
             # Display chat history for bot 1
             for message in st.session_state.chat_history_1:
                 with st.chat_message(message["role"], avatar=genparam.USER_AVATAR if message["role"] == "user" else genparam.BOT_1_AVATAR):
@@ -329,7 +329,7 @@ def main():
             
         with col2:
             st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
-            st.subheader(genparam.BOT_2_NAME)
+            st.subheader(f"{genparam.BOT_2_AVATAR} {genparam.BOT_2_NAME}")
             # Display chat history for bot 2
             for message in st.session_state.chat_history_2:
                 with st.chat_message(message["role"], avatar=genparam.USER_AVATAR if message["role"] == "user" else genparam.BOT_2_AVATAR):
@@ -361,7 +361,7 @@ def main():
             
         with col3:
             st.markdown("<div class='chat-container'>", unsafe_allow_html=True)
-            st.subheader(genparam.BOT_3_NAME)
+            st.subheader(f"{genparam.BOT_3_AVATAR} {genparam.BOT_3_NAME}")
             # Display chat history for bot 3
             for message in st.session_state.chat_history_3:
                 with st.chat_message(message["role"], avatar=genparam.USER_AVATAR if message["role"] == "user" else genparam.BOT_3_AVATAR):