add state indicator

pages/Gallery.py

@@ -14,6 +14,7 @@ from datasets import load_dataset, Dataset, load_from_disk
 from huggingface_hub import login
 from streamlit_agraph import agraph, Node, Edge, Config
 from streamlit_extras.switch_page_button import switch_page
+from streamlit_extras.tags import tagger_component
 from streamlit_extras.no_default_selectbox import selectbox
 from sklearn.svm import LinearSVC
 
@@ -279,8 +280,10 @@ class GalleryApp:
         # return prompt_tags, tag, prompt_id, items
 
     def app(self):
-        print(st.session_state.gallery_focus)
+        # print(st.session_state.gallery_focus)
         st.write('### Model Visualization and Retrieval')
+        with st.sidebar:
+            tagger_component('**Gallery State:**', [st.session_state.gallery_state.title()], color_name=['orange'])
         # st.write('This is a gallery of images generated by the models')
 
         # build the tabular view