Update app.py

app.py

@@ -190,6 +190,32 @@ def main():
             }
         </style>
         """, unsafe_allow_html=True)
+
+    # Base64-encoded images
+    facebook_icon = get_image_base64("facebook.png")
+    twitter_icon = get_image_base64("twitter.png")
+    linkedin_icon = get_image_base64("linkedin.png")
+    instagram_icon = get_image_base64("Instagram.png")
+    ci_icon = get_image_base64("ci.png")
+    avatar_1 = get_image_base64("avatar_2.png")
+    avatar_2 = get_image_base64("avatar_2.png")
+    avatar_3 = get_image_base64("avatar_3.png")
+    
+
+    # Avatar options (for example purposes, replace these with your actual avatars)
+    avatar_options = {
+        "Avatar 1": "avatar_1",
+        "Avatar 2": "avatar_2",
+        "Avatar 3": "avatar_3"
+    }
+
+    # Sidebar for selecting avatar
+    selected_avatar = st.sidebar.radio("Choose an avatar", list(avatar_options.keys()))
+
+    # Store the selected avatar in session state
+    st.session_state.user_selected_avatar = avatar_options[selected_avatar]
+
+    
   
     # Sidebar
     st.sidebar.title("About Citizens Information Chatbot")
@@ -225,13 +251,7 @@ def main():
         combine_docs_chain_kwargs={"prompt": qa_prompt})
 
     
-    # Base64-encoded images
-    facebook_icon = get_image_base64("facebook.png")
-    twitter_icon = get_image_base64("twitter.png")
-    linkedin_icon = get_image_base64("linkedin.png")
-    instagram_icon = get_image_base64("Instagram.png")
-    ci_icon = get_image_base64("ci.png")
-    user_2_icon = get_image_base64("user_2.png")
+
 
     # HTML for social media links with base64-encoded images
     social_media_html = f"""
@@ -419,7 +439,7 @@ def main():
             st.markdown(f"""
             <div style="display: flex; justify-content: flex-end; align-items: flex-start; margin-bottom: 20px;">
                 <span style="margin-right: 10px;">{question}</span>
-                <img src='data:image/png;base64,{user_2_icon}' style='width: 50px; height: 50px;'>
+                <img src='data:image/png;base64,{st.session_state.user_selected_avatar}' style='width: 50px; height: 50px;'>
             </div>
             """, unsafe_allow_html=True)