beta testing for new gallery page

Home.py

@@ -42,12 +42,20 @@ def info():
     with st.sidebar:
         st.write('## About')
 
+        # st.write(
+        #     "This is an web application to collect personal preference to images synthesised by generative models fine-tuned on stable diffusion. \
+        #     **You might consider it as a tool for quickly digging out the most suitable text-to-image generation model for you from [civitai](https://civitai.com/).**"
+        # )
+        # st.write(
+        #     "After you picking images from gallery page, and ranking them in the ranking page, you will be able to see a dashboard showing your preferred models in the summary page, **with download links of the models ready to use in [Automatic1111 webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)!**"
+        # )
+
         st.write(
-            "This is an web application to collect personal preference to images synthesised by generative models fine-tuned on stable diffusion. \
-            **You might consider it as a tool for quickly digging out the most suitable text-to-image generation model for you from [civitai](https://civitai.com/).**"
+            "This is a web application for individual users to quickly dig out the most suitable text-to-image generation model from civitai. Our research aims to understand personal preference to images synthesized by generative models fine-tuned on stable diffusion and you can contribute by playing with this tool and giving us your feedback! "
         )
+
         st.write(
-            "After you picking images from gallery page, and ranking them in the ranking page, you will be able to see a dashboard showing your preferred models in the summary page, **with download links of the models ready to use in [Automatic1111 webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)!**"
+            "After picking images you liked from Gallery and a battle-mode Ranking Contest, a summary dashboard will be presented indicating your preferred models with download links ready to be deployed in Webui !"
         )
 
 

pages/Gallery.py

@@ -12,6 +12,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.no_default_selectbox import selectbox
 from sklearn.svm import LinearSVC
 
 SCORE_NAME_MAPPING = {'clip': 'clip_score', 'rank': 'msq_score', 'pop': 'model_download_count'}
@@ -226,6 +227,7 @@ class GalleryApp:
 
             # items = items[items['prompt'] == selected_prompt].reset_index(drop=True)
 
+            st.title('Model Visualization and Retrieval')
 
             # show source
             if isinstance(note, str):
@@ -282,36 +284,46 @@ class GalleryApp:
 
                     subset_selector = st.columns([3, 1])
                     with subset_selector[0]:
-                        selected_prompt = st.selectbox('Select prompt', prompts, index=3)
+                        # selected_prompt = st.selectbox('Select prompt', prompts, index=3)
+                        selected_prompt = selectbox('Select prompt', prompts, key=f'prompt_{tag}', no_selection_label='---')
                     with subset_selector[1]:
-                        subset = st.selectbox('Select a subset', ['All', 'Selected Only'], index=0, key=f'subset_{selected_prompt}')
-                    items = items[items['prompt'] == selected_prompt].reset_index(drop=True)
-                    prompt_id = items['prompt_id'].unique()[0]
-                    note = items['note'].unique()[0]
-
-                    # add safety check for some prompts
-                    safety_check = True
-                    unsafe_prompts = {}
-                    # initialize unsafe prompts
-                    for prompt_tag in prompt_tags:
-                        unsafe_prompts[prompt_tag] = []
-                    # manually add unsafe prompts
-                    unsafe_prompts['world knowledge'] = [83]
-                    unsafe_prompts['abstract'] = [1, 3]
-
-                    if int(prompt_id.item()) in unsafe_prompts[tag]:
-                        st.warning('This prompt may contain unsafe content. They might be offensive, depressing, or sexual.')
-                        safety_check = st.checkbox('I understand that this prompt may contain unsafe content. Show these images anyway.', key=f'safety_{prompt_id}')
-
-                    if safety_check:
-
-                        # if subset == 'Selected Only' and 'selected_dict' in st.session_state:
-                        #     items = items[items['modelVersion_id'].isin(st.session_state.selected_dict[prompt_id])].reset_index(drop=True)
-                        #     self.gallery_mode(prompt_id, items)
-                        # else:
-                        self.graph_mode(prompt_id, items)
-
-        self.sidebar(items, prompt_id, note)
+                        subset = st.selectbox('Select a subset', ['All', 'Selected Only'], index=0, key=f'subset_{tag}')
+
+                    if selected_prompt is None:
+                        st.markdown(':orange[Please select a prompt above👆]')
+                    else:
+                        items = items[items['prompt'] == selected_prompt].reset_index(drop=True)
+                        prompt_id = items['prompt_id'].unique()[0]
+                        note = items['note'].unique()[0]
+
+                        # add safety check for some prompts
+                        safety_check = True
+                        unsafe_prompts = {}
+                        # initialize unsafe prompts
+                        for prompt_tag in prompt_tags:
+                            unsafe_prompts[prompt_tag] = []
+                        # manually add unsafe prompts
+                        unsafe_prompts['world knowledge'] = [83]
+                        unsafe_prompts['abstract'] = [1, 3]
+
+                        if int(prompt_id.item()) in unsafe_prompts[tag]:
+                            st.warning('This prompt may contain unsafe content. They might be offensive, depressing, or sexual.')
+                            safety_check = st.checkbox('I understand that this prompt may contain unsafe content. Show these images anyway.', key=f'safety_{prompt_id}')
+
+                        if safety_check:
+
+                            if subset == 'Selected Only' and 'selected_dict' in st.session_state:
+                                # try:
+                                items = items[items['modelVersion_id'].isin(st.session_state.selected_dict[prompt_id])].reset_index(drop=True)
+                                self.gallery_mode(prompt_id, items)
+                                # except:
+                                #     st.warning('No selected images found')
+                            else:
+                                self.graph_mode(prompt_id, items)
+        try:
+            self.sidebar(items, prompt_id, note)
+        except:
+            pass
 
     def graph_mode(self, prompt_id, items):
         graph_cols = st.columns([3, 1])
@@ -397,9 +409,9 @@ class GalleryApp:
         #         with dynamic_weight_panel[i]:
         #             btn = st.button(method, use_container_width=True, disabled=btn_disable, on_click=self.dynamic_weight, args=(prompt_id, items, method))
 
-        prompt = st.chat_input(f"Selected model version ids: {str(st.session_state.selected_dict.get(prompt_id, []))}", disabled=False, key=f'{prompt_id}')
-        if prompt:
-            switch_page("ranking")
+        # prompt = st.chat_input(f"Selected model version ids: {str(st.session_state.selected_dict.get(prompt_id, []))}", disabled=False, key=f'{prompt_id}')
+        # if prompt:
+        #     switch_page("ranking")
 
         with st.form(key=f'{prompt_id}'):
             # buttons = st.columns([1, 1, 1])