Update app.py

app.py

@@ -247,21 +247,49 @@ def main():
 
     with col1:
         st.image(f"data:image/png;base64,{avatar_1}", width=50)
-        if st.button("Select Avatar 1"):
+        if st.button("Select"):
             st.session_state.user_selected_avatar = avatar_1
-
-    with col2:
         st.image(f"data:image/png;base64,{avatar_2}", width=50)
-        if st.button("Select Avatar 2"):
+        if st.button("Select"):
             st.session_state.user_selected_avatar = avatar_2
-
-    with col3:
         st.image(f"data:image/png;base64,{avatar_3}", width=50)
-        if st.button("Select Avatar 3"):
-            st.session_state.user_selected_avatar = avatar_3
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_3           
+        st.image(f"data:image/png;base64,{avatar_4}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_4
+            
 
+    with col2:
+        st.image(f"data:image/png;base64,{avatar_5}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_5
+        st.image(f"data:image/png;base64,{avatar_6}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_6
+        st.image(f"data:image/png;base64,{avatar_7}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_7
+        st.image(f"data:image/png;base64,{avatar_8}", width=50)
+        if st.button("Select Avatar 2"):
+            st.session_state.user_selected_avatar = avatar_8
+            
 
-    
+    with col3:
+        st.image(f"data:image/png;base64,{avatar_9}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_9
+        st.image(f"data:image/png;base64,{avatar_10}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_10
+        st.image(f"data:image/png;base64,{avatar_11}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_11
+        st.image(f"data:image/png;base64,{avatar_12}", width=50)
+        if st.button("Select"):
+            st.session_state.user_selected_avatar = avatar_12
+
+            
     
     # Create a slider widget in the sidebar
     ai_temp = st.sidebar.slider("Select AI Temperature", min_value=0.0, max_value=1.0, value=0.7, step=0.1)