colpali fix

pages/Multimodal_Conversational_Search.py

@@ -279,7 +279,26 @@ def render_answer(question,answer,index,res_img):
     with col2:
         ans_ = answer['answer']
         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() 
+            
+        # if(st.session_state.input_is_colpali):
+        #     placeholder__ = st.empty()
+        #     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])
     with colu2:
@@ -338,8 +357,8 @@ def render_answer(question,answer,index,res_img):
 
             rdn_key = ''.join([random.choice(string.ascii_letters)
                               for _ in range(10)])
-            rdn_key_1 = ''.join([random.choice(string.ascii_letters)
-                              for _ in range(10)])
+            # rdn_key_1 = ''.join([random.choice(string.ascii_letters)
+            #                   for _ in range(10)])
             currentValue = ''.join(st.session_state.input_rag_searchType)+str(st.session_state.input_is_rerank)+str(st.session_state.input_table_with_sql)+st.session_state.input_index
             oldValue = ''.join(st.session_state.inputs_["rag_searchType"])+str(st.session_state.inputs_["is_rerank"])+str(st.session_state.inputs_["table_with_sql"])+str(st.session_state.inputs_["index"])
             #print("changing values-----------------")
@@ -370,26 +389,7 @@ def render_answer(question,answer,index,res_img):
                 pass  
 
             placeholder__ = st.empty()
-            #with placeholder__.container():
-            placeholder__.button("Show similarity map",key=rdn_key_1,on_click=handle_input,args=('colpali_show_similarity_map',None))
-            
-
-        # 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__.button("🔄",key=rdn_key,on_click=on_button_click)
             #placeholder__.button("Show similarity map",key=rdn_key_1,on_click = show_maxsim)
         
         
@@ -546,7 +546,9 @@ with st.sidebar:
         #st.session_state.input_query = ""
     else:
         st.session_state.input_is_colpali = False
-    
+
+    st.button("Show similarity map",key="input_sim_map",on_click=handle_input,args=('colpali_show_similarity_map',None))
+       
     with st.expander("Sample questions for Colpali retriever:"):
         st.write("1. Proportion of female new hires 2021-2023? \n\n 2. First-half 2021 return on unlisted real estate investments? \n\n 3. Trend of the fund's expected absolute volatility between January 2014 and January 2016? \n\n 4. Fund return percentage in 2017? \n\n 5. Annualized gross return of the fund from 1997 to 2008?")