rerank model

pages/Multimodal_Conversational_Search.py

@@ -330,22 +330,24 @@ def render_all():
 
 placeholder = st.empty()
 with placeholder.container():
-    if st.session_state.trigger_search and not st.session_state.run_done:
+    if st.session_state.trigger_search:
         with st.spinner("Running search..."):
             handle_input()
         render_all()
-        st.session_state.run_done = True
+        st.session_state.trigger_search = False
 
+        
 st.markdown("")
-col_2, col_3 = st.columns([75,20])  
-with col_2:
-    input = st.text_input( "Ask here",label_visibility = "collapsed",key="input_query")
-with col_3:
-    #play = st.button("Go",on_click=handle_input,key = "play")
-    play = st.button("Go", key="play")
-    if play:
-        st.session_state.trigger_search = True
-        st.session_state.run_done = False
+with st.form("query_form"):
+    col_2, col_3 = st.columns([75, 20])
+    with col_2:
+        st.text_input("Ask here", key="input_query", label_visibility="collapsed")
+    with col_3:
+        submitted = st.form_submit_button("Go")
+
+if submitted:
+    st.session_state.trigger_search = True
+