colpali fix

pages/Multimodal_Conversational_Search.py

@@ -282,6 +282,9 @@ def render_answer(question,answer,index,res_img):
         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)
+            st.session_state.input_query = st.session_state.questions_[-1]["question"]
+            st.session_state.answers_.pop()
+            st.session_state.questions_.pop()
             handle_input()
             with placeholder.container():
                 render_all() 
@@ -338,8 +341,8 @@ def render_answer(question,answer,index,res_img):
                                     st.image(parent_dirname+"/figures/"+st.session_state.input_index+"/"+img+".jpg")
                             #st.write(caption)
                             idx = idx+1
-                    if(st.session_state.show_columns == True):
-                        st.session_state.show_columns = False
+                    # if(st.session_state.show_columns == True):
+                    #     st.session_state.show_columns = False
                 #st.markdown("<div style='color:#e28743';padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'><b>Sources from the document:</b></div>", unsafe_allow_html = True)
                 if(len(answer["table"] )>0):
                     #with st.expander("Table:"):