rerank model

pages/Multimodal_Conversational_Search.py

@@ -216,14 +216,14 @@ def render_answer(question,answer,index,res_img):
         #     st.audio(polly_response['AudioStream'].read(), format="audio/ogg")
         # rdn_key_1 = ''.join([random.choice(string.ascii_letters)
         #                       for _ in range(10)])
-        # def show_maxsim():
-        #     st.session_state.show_columns = True
-        #     st.session_state.maxSimImages = colpali.img_highlight(st.session_state.top_img, st.session_state.query_token_vectors, st.session_state.query_tokens)
-        #     handle_input()
-        #     with placeholder.container():
-        #         render_all()    
-        # if(st.session_state.input_is_colpali):
-        #     st.button("Show similarity map",key=rdn_key_1,on_click = show_maxsim)
+        def show_maxsim():
+            st.session_state.show_columns = True
+            st.session_state.maxSimImages = colpali.img_highlight(st.session_state.top_img, st.session_state.query_token_vectors, st.session_state.query_tokens)
+            handle_input()
+            with placeholder.container():
+                render_all()    
+        if(st.session_state.input_is_colpali):
+            st.button("Show similarity map",key=rdn_key_1,on_click = show_maxsim)
                 
     colu1,colu2,colu3 = st.columns([4,82,20])
     with colu2:
@@ -327,9 +327,8 @@ def render_all():
 placeholder = st.empty()
 with placeholder.container():
     with st.spinner("Running search..."):
-        #handle_input()
         render_all()
-        #st.session_state.trigger_search = False  # reset AFTER running
+        
 
 
         
@@ -340,13 +339,6 @@ with col_2:
 with col_3:
     play = st.button("Go",on_click=handle_input, key="play")
 
-# # ❗ Do NOT run anything here
-# if play:
-#     st.session_state.trigger_search = True  # mark to run on next rerun
-
-
-
-
 
 ##### Sidebar #####
 with st.sidebar: