update summary ui

pages/Summary.py

@@ -111,47 +111,33 @@ class DashboardApp:
             icon = '🥇'if i == 0 else '🥈' if i == 1 else '🥉' if i == 2 else '🎈'
             podium_display = st.columns([1, 14], gap='medium')
             with podium_display[0]:
-                st.title(f'{icon}')
+                # st.title(f'{icon}')
+                st.write(f'# {icon}')
                 # if summary_mode == 'display':
                 #     st.title(f'{icon}')
                 # elif summary_mode == 'edit':
-                #     settop = st.button('🔝', key=f'settop_{modelVersion_id}', help='Set this model to the top', disabled=i == 0, on_click=self.switch_order, args=(tag, i, 0), use_container_width=True)
-                #     moveup = st.button('⬆', key=f'moveup_{modelVersion_id}', help='Move this model up', disabled=i == 0, on_click=self.switch_order, args=(tag, i, i - 1), use_container_width=True)
-                #     movedown = st.button('⬇', key=f'movedown_{modelVersion_id}', help='Move this model down', disabled=i == n - 1, on_click=self.switch_order, args=(tag, i, i + 1), use_container_width=True)
+                settop = st.button('🔝', key=f'settop_{modelVersion_id}', help='Set this model to the top', disabled=i == 0, on_click=self.switch_order, args=(tag, i, 0))
+                moveup = st.button('⬆', key=f'moveup_{modelVersion_id}', help='Move this model up', disabled=i == 0, on_click=self.switch_order, args=(tag, i, i - 1))
+                movedown = st.button('⬇', key=f'movedown_{modelVersion_id}', help='Move this model down', disabled=i == n - 1, on_click=self.switch_order, args=(tag, i, i + 1))
             with podium_display[1]:
-                title_display = st.columns([3.5, 2, 2, 2, 0.5, 0.5, 0.5])
+                title_display = st.columns([4, 1, 1])
                 with title_display[0]:
                     st.write(f'##### {model_name}, {modelVersion_name}')
                     # st.write(f'Ranking Score: {winning_times}')
                 with title_display[1]:
-                    # image_display = st.selectbox('image display', ['Featured', 'All Images'], key=f'image_display_{modelVersion_id}', label_visibility='collapsed')
-                    image_display = st.toggle('Show all images', key=f'image_display_{modelVersion_id}')
-
-                with title_display[2]:
                     st.link_button('Download', url, use_container_width=True)
-                with title_display[3]:
+                with title_display[2]:
                     st.link_button('Civitai', f'https://civitai.com/models/{model_id}?modelVersionId={modelVersion_id}', use_container_width=True, type='primary')
                 # st.write(f'[Civitai Page](https://civitai.com/models/{model_id}?modelVersionId={modelVersion_id}), [Model Download Link]({url}), Ranking Score: {winning_times}')
                 # with st.expander(f'**{icon} {model_name}, [{modelVersion_name}](https://civitai.com/models/{model_id}?modelVersionId={modelVersion_id})**, Ranking Score: {winning_times}'):
-                with title_display[4]:
-                    settop = st.button('🔝', key=f'settop_{modelVersion_id}', help='Set this model to the top',
-                                       disabled=i == 0, on_click=self.switch_order, args=(tag, i, 0),
-                                       use_container_width=True)
-                with title_display[5]:
-                    moveup = st.button('⬆', key=f'moveup_{modelVersion_id}', help='Move this model up',
-                                       disabled=i == 0, on_click=self.switch_order, args=(tag, i, i - 1),
-                                       use_container_width=True)
-                with title_display[6]:
-                    movedown = st.button('⬇', key=f'movedown_{modelVersion_id}', help='Move this model down',
-                                         disabled=i == n - 1, on_click=self.switch_order, args=(tag, i, i + 1),
-                                         use_container_width=True)
 
+                image_display = st.toggle('Show all images', key=f'image_display_{modelVersion_id}')
                 if not image_display:
                     example_images = self.promptBook[self.promptBook['prompt_id'].isin(example_prompts) & (self.promptBook['modelVersion_id']==modelVersion_id)]['image_id'].values
                     example_images = [f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{image}.png" for image in example_images]
                     clickable_images(
                         example_images,
-                        img_style={"margin": "5px", "height": "100px"}
+                        img_style={"margin": "5px", "height": "120px"}
                     )
 
                 else:
@@ -171,7 +157,7 @@ class DashboardApp:
                     images = [f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{image}.png" for image in images]
                     clickable_images(
                         images,
-                        img_style={"margin": "5px", "height": "100px"}
+                        img_style={"margin": "5px", "height": "120px"}
                     )
                     st.write('🐌 It may take a while to load all images. Please be patient, and **NEVER USE THE REFRESH BUTTON ON YOUR BROWSER**.')