descriptions update

Home.py

@@ -42,11 +42,18 @@ if __name__ == '__main__':
     # print(st.source_util.get_pages('Home.py'))
 
     st.set_page_config(page_title="Login", page_icon="🏠", layout="wide")
-    st.write('A Research by MAPS Lab, NYU Shanghai')
+    st.write('A Research by [MAPS Lab](https://whongyi.github.io/MAPS-research), [NYU Shanghai](https://shanghai.nyu.edu)')
     st.title("🙌 Welcome to GEMRec Gallery Webapp!")
+    st.info("Getting obsessed with tons of different text-to-image generation models available online?  \n \
+            Want to find the most suitable one for your taste?  \n \
+            **GEMRec** is here to help you!"
+             )
     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 model for you from [civitai](https://civitai.com/).**"
+        **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)!**"
     )
 
     if 'user_id' not in st.session_state:

pages/Gallery.py

@@ -212,15 +212,15 @@ class GalleryApp:
         with st.sidebar:
             prompt_tags = self.promptBook['tag'].unique()
             # sort tags by alphabetical order
-            prompt_tags = np.sort(prompt_tags)[::-1]
+            prompt_tags = np.sort(prompt_tags)[::1]
 
-            tag = st.selectbox('Select a tag', prompt_tags)
+            tag = st.selectbox('Select a tag', prompt_tags, index=5)
 
             items = self.promptBook[self.promptBook['tag'] == tag].reset_index(drop=True)
 
-            prompts = np.sort(items['prompt'].unique())[::-1]
+            prompts = np.sort(items['prompt'].unique())[::1]
 
-            selected_prompt = st.selectbox('Select prompt', prompts)
+            selected_prompt = st.selectbox('Select prompt', prompts, index=3)
 
             mode = st.radio('Select a mode', ['Gallery', 'Graph'], horizontal=True, index=1)
 
@@ -269,6 +269,13 @@ class GalleryApp:
         prompt_tags, tag, prompt_id, items, mode = self.sidebar()
         # items, info, col_num = self.selection_panel(items)
 
+        # subset = st.radio('Select a subset', ['All', 'Selected Only'], index=0, horizontal=True)
+        # try:
+        #     if subset == 'Selected Only':
+        #         items = items[items['modelVersion_id'].isin(st.session_state.selected_dict[prompt_id])].reset_index(drop=True)
+        # except:
+        #     pass
+
         # add safety check for some prompts
         safety_check = True
         unsafe_prompts = {}

pages/Summary.py

@@ -43,6 +43,7 @@ class DashboardApp:
 
         tab1, tab2 = st.tabs(['Top Picks', 'Detailed Info'])
 
+
         with tab1:
             # show the top 3 in metric cards
             st.write('## Top picks')
@@ -92,11 +93,14 @@ class DashboardApp:
                                     image = f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{images[i]}.png"
                                     st.image(image, use_column_width=True)
 
-            # st.write('---')
-            #     expander = st.expander(f'### {icon} {model_name}, [{modelVersion_name}](https://civitai.com/models/{modelVersion_id})')
-            #     with expander:
-            #         images = self.promptBook[self.promptBook['modelVersion_id'] == modelVersion_id]['image_id'].values
-            #         st.write(images)
+
+                # # st.write('---')
+                # expander = st.expander(f'# {icon} {model_name}, [{modelVersion_name}](https://civitai.com/models/{modelVersion_id})')
+                # with expander:
+                #     images = self.promptBook[self.promptBook['modelVersion_id'] == modelVersion_id]['image_id'].values
+                #     st.write(images)
+
+        st.chat_input('Please leave your comments here.', key='comment')
 
         with tab2:
             st.write('## Detailed information of all selected models')
@@ -104,6 +108,8 @@ class DashboardApp:
             st.data_editor(detailed_info, hide_index=True, disabled=True)
 
 
+
+
     def score_calculator(self, results, db_table):
         # sort results by battle time
         results = sorted(results, key=lambda x: x['battletime'])
@@ -151,6 +157,8 @@ class DashboardApp:
             tag = self.sidebar(tags, mode)
             self.leaderboard(tag, db_table)
 
+        # st.chat_input('Please leave your comments here.', key='comment')
+
 
 if __name__ == "__main__":
     st.set_page_config(layout="wide")