allow shuffling in gallery

pages/Gallery.py

@@ -4,6 +4,7 @@ import requests
 
 import altair as alt
 import extra_streamlit_components as stx
+import random
 import numpy as np
 import pandas as pd
 import streamlit as st
@@ -212,6 +213,8 @@ class GalleryApp:
 
             prompts = np.sort(items['prompt'].unique())[::1].tolist()
 
+            # print('prompts: ', prompts, 'tags: ', prompt_tags)
+
             # selt focus prompt index if exists
             if st.session_state.gallery_focus['prompt'] is None or tag != st.session_state.gallery_focus['tag']:
                 prompt_focus_idx = 0
@@ -221,17 +224,21 @@ class GalleryApp:
             # st.caption('Select a prompt')
             subset_selector = st.columns([3, 1])
             with subset_selector[0]:
+                selector_bar = st.columns([1, 15])
+                with selector_bar[0]:
+                    shuffle = st.button('🎲', key='prompt_shuffle', on_click=self.random_gallery_focus, args=(prompt_tags,), use_container_width=True)
 
-                # add coloring to prompt based on selection
-                prompts_tobe_colored = self.promptBook[self.promptBook['prompt_id'].isin(st.session_state.selected_dict.keys())]['prompt'].unique()
-                colored_prompts = self.text_coloring_add(prompts_tobe_colored, prompts, color_name='✅')
+                with selector_bar[-1]:
+                    # add coloring to prompt based on selection
+                    prompts_tobe_colored = self.promptBook[self.promptBook['prompt_id'].isin(st.session_state.selected_dict.keys())]['prompt'].unique()
+                    colored_prompts = self.text_coloring_add(prompts_tobe_colored, prompts, color_name='✅')
 
-                selected_prompt = selectbox('Select prompt', colored_prompts, key=f'prompt_{tag}', no_selection_label='---', label_visibility='collapsed', index=prompt_focus_idx)
+                    selected_prompt = selectbox('Select prompt', colored_prompts, key=f'prompt_{tag}', no_selection_label='---', label_visibility='collapsed', index=prompt_focus_idx)
 
-                # remove coloring from prompt
-                selected_prompt = self.text_coloring_remove(selected_prompt)
-                print('selected_prompt: ', selected_prompt)
-                st.session_state.prompt_idx_last_time = prompts.index(selected_prompt) if selected_prompt else 0
+                    # remove coloring from prompt
+                    selected_prompt = self.text_coloring_remove(selected_prompt)
+                    print('selected_prompt: ', selected_prompt)
+                    st.session_state.prompt_idx_last_time = prompts.index(selected_prompt) if selected_prompt else 0
 
             if selected_prompt is None:
                 # st.markdown(':orange[Please select a prompt above👆]')
@@ -301,6 +308,14 @@ class GalleryApp:
 
             self.checkout_mode(tag, items)
 
+
+    def random_gallery_focus(self, tags):
+        st.session_state.gallery_focus['tag'] = random.choice(tags)
+        # st.session_state.gallery_focus['prompt'] = random.choice(prompts)
+        prompts = self.promptBook[self.promptBook['tag'] == st.session_state.gallery_focus['tag']]['prompt'].unique()
+        st.session_state.gallery_focus['prompt'] = random.choice(prompts)
+
+
     def graph_mode(self, prompt_id, items):
         graph_cols = st.columns([3, 1])
         # prompt = st.chat_input(f"Selected model version ids: {str(st.session_state.selected_dict.get(prompt_id, []))}",

pages/Summary.py

@@ -233,7 +233,7 @@ class DashboardApp:
             st.info(f'No rankings are finished with {mode} mode yet.')
 
         else:
-            tags = tags[0:1] if len(tags) == 2 else tags
+            tags = tags[1:2] if len(tags) == 2 else tags
             tag = st.radio('Select a tag', tags, index=0, horizontal=True, label_visibility='collapsed')
             self.sidebar(tags, mode)
             self.leaderboard(tag, db_table)