render chat input first

pages/Gallery.py

@@ -246,6 +246,10 @@ class GalleryApp:
                     with dynamic_weight_panel[i]:
                         btn = st.button(method, use_container_width=True, disabled=btn_disable, on_click=self.dynamic_weight, args=(prompt_id, items, method))
 
+            prompt = st.chat_input(f"Selected model version ids: {str(st.session_state.selected_dict.get(prompt_id, []))}", disabled=False, key=f'{prompt_id}')
+            if prompt:
+                switch_page("ranking")
+
             with st.form(key=f'{prompt_id}'):
                 # buttons = st.columns([1, 1, 1])
                 buttons_space = st.columns([1, 1, 1, 1])
@@ -273,9 +277,9 @@ class GalleryApp:
                     with st.spinner('Loading images...'):
                         self.gallery_standard(items, col_num, info)
 
-            prompt = st.chat_input(f"checked: {str(st.session_state.selected_dict.get(prompt_id, []))}", disabled=False, key=f'{prompt_id}')
-            if prompt:
-                switch_page("ranking")
+            # prompt = st.chat_input(f"checked: {str(st.session_state.selected_dict.get(prompt_id, []))}", disabled=False, key=f'{prompt_id}')
+            # if prompt:
+            #     switch_page("ranking")
 
     def submit_actions(self, status, prompt_id):
         # remove counter from session state