remove gallery form

pages/Gallery.py

@@ -151,7 +151,7 @@ def selection_panel(items):
 
     return items, info, col_num
 
-def sidebar(promptbook, images_db):
+def sidebar(promptBook, images_ds):
     with st.sidebar:
         prompt_tags = promptBook['tag'].unique()
         # sort tags by alphabetical order
@@ -203,11 +203,11 @@ def sidebar(promptbook, images_db):
 
     return prompt_tags, tag, prompt_id, items
 
-def app(promptbook, images_db):
+def app(promptBook, images_ds):
     st.title('Model Visualization and Retrieval')
     st.write('This is a gallery of images generated by the models')
 
-    prompt_tags, tag, prompt_id, items = sidebar(promptbook, images_db)
+    prompt_tags, tag, prompt_id, items = sidebar(promptBook, images_ds)
 
     # add safety check for some prompts
     safety_check = True
@@ -243,32 +243,32 @@ def app(promptbook, images_db):
                 with dynamic_weight_panel[i]:
                     btn = st.button(method, use_container_width=True, disabled=btn_disable, on_click=dynamic_weight, args=(prompt_id, items, method))
 
-        with st.form(key=f'{prompt_id}'):
-            # buttons = st.columns([1, 1, 1])
-            buttons_space = st.columns([1, 1, 1, 1])
-            gallery_space = st.empty()
-
-            with buttons_space[0]:
-                continue_btn = st.form_submit_button('Confirm Selection', use_container_width=True, type='primary')
-                if continue_btn:
-                    submit_actions('Continue', prompt_id)
-
-            with buttons_space[1]:
-                select_btn = st.form_submit_button('Select All', use_container_width=True)
-                if select_btn:
-                    submit_actions('Select', prompt_id)
-
-            with buttons_space[2]:
-                deselect_btn = st.form_submit_button('Deselect All', use_container_width=True)
-                if deselect_btn:
-                    submit_actions('Deselect', prompt_id)
-
-            with buttons_space[3]:
-                refresh_btn = st.form_submit_button('Refresh', on_click=gallery_space.empty, use_container_width=True)
-
-            with gallery_space.container():
-                with st.spinner('Loading images...'):
-                    gallery_standard(items, col_num, info)
+        # with st.form(key=f'{prompt_id}'):
+        #     # buttons = st.columns([1, 1, 1])
+        #     buttons_space = st.columns([1, 1, 1, 1])
+        #     gallery_space = st.empty()
+        #
+        #     with buttons_space[0]:
+        #         continue_btn = st.form_submit_button('Confirm Selection', use_container_width=True, type='primary')
+        #         if continue_btn:
+        #             submit_actions('Continue', prompt_id)
+        #
+        #     with buttons_space[1]:
+        #         select_btn = st.form_submit_button('Select All', use_container_width=True)
+        #         if select_btn:
+        #             submit_actions('Select', prompt_id)
+        #
+        #     with buttons_space[2]:
+        #         deselect_btn = st.form_submit_button('Deselect All', use_container_width=True)
+        #         if deselect_btn:
+        #             submit_actions('Deselect', prompt_id)
+        #
+        #     with buttons_space[3]:
+        #         refresh_btn = st.form_submit_button('Refresh', on_click=gallery_space.empty, use_container_width=True)
+        #
+        #     with gallery_space.container():
+        #         with st.spinner('Loading images...'):
+        #             gallery_standard(items, col_num, info)
 
 def submit_actions(status, prompt_id):
     if status == 'Select':
@@ -436,124 +436,4 @@ if __name__ == "__main__":
         if 'selected_dict' not in st.session_state:
             st.session_state['selected_dict'] = {}
 
-        # app(promptBook, images_ds)
-
-        # specify sidebar
-        with st.sidebar:
-            prompt_tags = promptBook['tag'].unique()
-            # sort tags by alphabetical order
-            prompt_tags = np.sort(prompt_tags)[::-1]
-
-            tag = st.selectbox('Select a tag', prompt_tags)
-
-            items = promptBook[promptBook['tag'] == tag].reset_index(drop=True)
-
-            prompts = np.sort(items['prompt'].unique())[::-1]
-
-            selected_prompt = st.selectbox('Select prompt', prompts)
-
-            items = items[items['prompt'] == selected_prompt].reset_index(drop=True)
-            prompt_id = items['prompt_id'].unique()[0]
-            note = items['note'].unique()[0]
-
-            # show source
-            if isinstance(note, str):
-                if note.isdigit():
-                    st.caption(f"`Source: civitai`")
-                else:
-                    st.caption(f"`Source: {note}`")
-            else:
-                st.caption("`Source: Parti-prompts`")
-
-            # show image metadata
-            image_metadatas = ['prompt_id', 'prompt', 'negativePrompt', 'sampler', 'cfgScale', 'size', 'seed']
-            for key in image_metadatas:
-                label = ' '.join(key.split('_')).capitalize()
-                st.write(f"**{label}**")
-                if items[key][0] == ' ':
-                    st.write('`None`')
-                else:
-                    st.caption(f"{items[key][0]}")
-
-            # for note as civitai image id, add civitai reference
-            if isinstance(note, str) and note.isdigit():
-                try:
-                    st.write(f'**[Civitai Reference](https://civitai.com/images/{note})**')
-                    res = requests.get(f'https://civitai.com/images/{note}')
-                    # st.write(res.text)
-                    soup = BeautifulSoup(res.text, 'html.parser')
-                    image_section = soup.find('div', {'class': 'mantine-12rlksp'})
-                    image_url = image_section.find('img')['src']
-                    st.image(image_url, use_column_width=True)
-                except:
-                    pass
-
-
-        # move things out of app()
-        st.title('Model Visualization and Retrieval')
-        st.write('This is a gallery of images generated by the models')
-
-        # 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['art'] = [23]
-        unsafe_prompts['abstract'] = [1, 3]
-        # unsafe_prompts['food'] = [34]
-
-        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'{prompt_id}')
-
-        if safety_check:
-            items, info, col_num = selection_panel(items)
-
-            if 'selected_dict' in st.session_state:
-                st.write('checked: ', str(st.session_state.selected_dict.get(prompt_id, [])))
-                dynamic_weight_options = ['Grid Search', 'SVM', 'Greedy']
-                dynamic_weight_panel = st.columns(len(dynamic_weight_options))
-
-                if len(st.session_state.selected_dict.get(prompt_id, [])) > 0:
-                    btn_disable = False
-                else:
-                    btn_disable = True
-
-                for i in range(len(dynamic_weight_options)):
-                    method = dynamic_weight_options[i]
-                    with dynamic_weight_panel[i]:
-                        btn = st.button(method, use_container_width=True, disabled=btn_disable, on_click=dynamic_weight,
-                                        args=(prompt_id, items, method))
-
-            with st.form(key=f'{prompt_id}'):
-                # buttons = st.columns([1, 1, 1])
-                buttons_space = st.columns([1, 1, 1, 1])
-                gallery_space = st.empty()
-
-                with buttons_space[0]:
-                    continue_btn = st.form_submit_button('Confirm Selection', use_container_width=True, type='primary')
-                    if continue_btn:
-                        submit_actions('Continue', prompt_id)
-
-                with buttons_space[1]:
-                    select_btn = st.form_submit_button('Select All', use_container_width=True)
-                    if select_btn:
-                        submit_actions('Select', prompt_id)
-
-                with buttons_space[2]:
-                    deselect_btn = st.form_submit_button('Deselect All', use_container_width=True)
-                    if deselect_btn:
-                        submit_actions('Deselect', prompt_id)
-
-                with buttons_space[3]:
-                    refresh_btn = st.form_submit_button('Refresh', on_click=gallery_space.empty,
-                                                        use_container_width=True)
-
-                with gallery_space.container():
-                    with st.spinner('Loading images...'):
-                        gallery_standard(items, col_num, info)
+        app(promptBook, images_ds)