graph ui fix

pages/Gallery.py

@@ -69,6 +69,7 @@ class GalleryApp:
 
             nodes.append(Node(id=items.loc[idx, 'image_id'],
                               # label=str(items.loc[idx, 'model_name']),
+                              title = f"model name: {items.loc[idx, 'model_name']}\nmodelVersion name: {items.loc[idx, 'modelVersion_name']}\nclip score: {items.loc[idx, 'clip_score']}\nmcos score: {items.loc[idx, 'mcos_score']}\npopularity: {items.loc[idx, 'model_download_count']}",
                               size=20,
                               shape='image',
                               image=f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{items.loc[idx, 'image_id']}.png",
@@ -84,7 +85,7 @@ class GalleryApp:
                          )
 
         config = Config(width='100%',
-                        height=800,
+                        height='600',
                         directed=True,
                         physics=False,
                         hierarchical=False,
@@ -298,42 +299,47 @@ class GalleryApp:
         with graph_cols[0]:
             return_value = self.gallery_graph(items)
         with graph_cols[1]:
+
             if return_value:
-                image_url = f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{return_value}.png"
+                with st.form(key=f'{prompt_id}'):
+                    image_url = f"https://modelcofferbucket.s3-accelerate.amazonaws.com/{return_value}.png"
 
-                st.image(image_url)
+                    st.image(image_url)
 
-                item = items[items['image_id'] == return_value].reset_index(drop=True).iloc[0]
-                modelVersion_id = item['modelVersion_id']
+                    item = items[items['image_id'] == return_value].reset_index(drop=True).iloc[0]
+                    modelVersion_id = item['modelVersion_id']
 
-                # handle selection
-                if 'selected_dict' in st.session_state:
-                    if item['prompt_id'] not in st.session_state.selected_dict:
-                        st.session_state.selected_dict[item['prompt_id']] = []
+                    # handle selection
+                    if 'selected_dict' in st.session_state:
+                        if item['prompt_id'] not in st.session_state.selected_dict:
+                            st.session_state.selected_dict[item['prompt_id']] = []
 
-                    if modelVersion_id in st.session_state.selected_dict[item['prompt_id']]:
-                        checked = True
-                    else:
-                        checked = False
+                        if modelVersion_id in st.session_state.selected_dict[item['prompt_id']]:
+                            checked = True
+                        else:
+                            checked = False
 
-                if checked:
-                    deselect = st.button('Deselect', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True)
-                    if deselect:
-                        st.session_state.selected_dict[item['prompt_id']].remove(item['modelVersion_id'])
-                        st.experimental_rerun()
+                    if checked:
+                        # deselect = st.button('Deselect', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True)
+                        deselect = st.form_submit_button('Deselect', use_container_width=True)
+                        if deselect:
+                            st.session_state.selected_dict[item['prompt_id']].remove(item['modelVersion_id'])
+                            st.experimental_rerun()
 
-                else:
-                    select = st.button('Select', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True, type='primary')
-                    if select:
-                        st.session_state.selected_dict[item['prompt_id']].append(item['modelVersion_id'])
-                        st.experimental_rerun()
-
-                # st.write(item)
-                infos = ['model_name', 'modelVersion_name', 'model_download_count', 'clip_score', 'mcos_score',
-                         'nsfw_score']
-                for info in infos:
-                    st.write(f"**{info}**:")
-                    st.write(item[info])
+                    else:
+                        # select = st.button('Select', key=f'select_{item["prompt_id"]}_{item["modelVersion_id"]}', use_container_width=True, type='primary')
+                        select = st.form_submit_button('Select', use_container_width=True, type='primary')
+                        if select:
+                            st.session_state.selected_dict[item['prompt_id']].append(item['modelVersion_id'])
+                            st.experimental_rerun()
+
+                    # st.write(item)
+                    infos = ['model_name', 'modelVersion_name', 'model_download_count', 'clip_score', 'mcos_score',
+                             'nsfw_score']
+                    st.table(item[infos])
+                    # for info in infos:
+                    #     st.write(f"**{info}**:")
+                    #     st.write(item[info])
 
             else:
                 st.info('Please click on an image to show')