colpali fix

pages/Multimodal_Conversational_Search.py

@@ -193,7 +193,9 @@ if clear:
     # st.session_state.input_maxTokens = 1024
 
 
-def handle_input():
+def handle_input(state,dummy):
+    if(state == 'colpali_show_similarity_map'):
+        st.session_state.show_columns = True    
     # st.session_state.answer_ready = True
     # st.session_state.show_columns = False  # reset column display
     print("Question: "+st.session_state.input_query)
@@ -279,23 +281,23 @@ def render_answer(question,answer,index,res_img):
         st.write(ans_)
         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.input_query = st.session_state.questions_[-1]["question"]
-            # st.session_state.answers_.pop()
-            # st.session_state.questions_.pop()
-            handle_input()
-            print("*"*20)
-            print(st.session_state.input_query) 
-            print(st.session_state.answers_)
-            print(st.session_state.questions_)
-            print("*"*20) 
-            with placeholder.container():
-                render_all() 
+        # def show_maxsim():
+        #     st.session_state.show_columns = True
+        #     # st.session_state.input_query = st.session_state.questions_[-1]["question"]
+        #     # st.session_state.answers_.pop()
+        #     # st.session_state.questions_.pop()
+        #     handle_input()
+        #     print("*"*20)
+        #     print(st.session_state.input_query) 
+        #     print(st.session_state.answers_)
+        #     print(st.session_state.questions_)
+        #     print("*"*20) 
+        #     with placeholder.container():
+        #         render_all() 
             
         if(st.session_state.input_is_colpali):
             placeholder__ = st.empty()
-            placeholder__.button("Show similarity map",key=rdn_key_1,on_click=show_maxsim)
+            placeholder__.button("Show similarity map",key=rdn_key_1,on_click=handle_input,args=('colpali_show_similarity_map',None))
                 
 
     colu1,colu2,colu3 = st.columns([4,82,20])
@@ -365,7 +367,7 @@ def render_answer(question,answer,index,res_img):
                     st.session_state.input_query = st.session_state.questions_[-1]["question"]
                     st.session_state.answers_.pop()
                     st.session_state.questions_.pop()
-                    handle_input()
+                    handle_input("regenerate_",None)
                     with placeholder.container():
                         render_all()
                     
@@ -425,7 +427,7 @@ with col_2:
     input = st.text_input( "Ask here",label_visibility = "collapsed",key="input_query")
 with col_3:
     #hidden = st.button("RUN",disabled=True,key = "hidden")
-    play = st.button("Go",on_click=handle_input,key = "play")
+    play = st.button("Go",on_click=handle_input,key = "play",args=(None,None),help = 'Enter the questions and click on "GO"')
 with st.sidebar:
     st.page_link("app.py", label=":orange[Home]", icon="🏠")
     st.subheader(":blue[Sample Data]")