add default expand

Home.py

@@ -74,7 +74,7 @@ if __name__ == '__main__':
     st.set_page_config(page_title="Login", page_icon="🏠", layout="wide")
     info()
     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.title("🙌 Welcome to GEMRec Gallery!")
     # 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!"

pages/Gallery.py

@@ -372,14 +372,22 @@ class GalleryApp:
     def checkout_mode(self, tag, items):
         # st.write(items)
         if len(items) > 0:
-            for prompt_id in items['prompt_id'].unique():
+            prompt_ids = items['prompt_id'].unique()
+            for i in range(len(prompt_ids)):
+                prompt_id = prompt_ids[i]
                 prompt = items[items['prompt_id'] == prompt_id]['prompt'].unique()[0]
-                default_expand = True if st.session_state.gallery_focus['prompt'] == prompt else False
+                # default_expand = True if st.session_state.gallery_focus['prompt'] == prompt else False
+                if tag == st.session_state.gallery_focus['tag'] and prompt == st.session_state.gallery_focus['prompt']:
+                    default_expand = True
+                elif tag != st.session_state.gallery_focus['tag'] and i==0:
+                    default_expand = True
+                else:
+                    default_expand = False
+                    
                 with st.expander(f'**{prompt}**', expanded=default_expand):
                     # st.caption('select info to show')
                     checkout_panel = st.columns([5, 3])
                     with checkout_panel[0]:
-                        pass
                         info = st.multiselect('Show Info',
                                               ['model_name', 'model_id', 'modelVersion_name', 'modelVersion_id',
                                                'total_score', 'model_download_count', 'clip_score', 'mcos_score',