rerank model

pages/Multimodal_Conversational_Search.py

@@ -195,10 +195,10 @@ def write_user_message(placeholder,md):
             #st.warning(md['question'])
 
             st.markdown("<div style='color:#e28743';font-size:18px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;font-style: italic;'>"+md['question']+"</div>", unsafe_allow_html = True)
-        
+    return placeholder
 
 
-def render_answer(question,answer,index,res_img):
+def render_answer(placeholder,question,answer,index,res_img):
     
     
     col1, col2, col_3 = st.columns([4,74,22])
@@ -218,15 +218,15 @@ 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():
+        def show_maxsim(placeholder,dummy):
             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()
-            placeholder = st.empty()
-            with placeholder.container():
-                render_all()    
+            # placeholder = st.empty()
+            # with placeholder.container():
+            render_all(placeholder)    
         if(st.session_state.input_is_colpali):
-            st.button("Show similarity map",key=rdn_key_1,on_click = show_maxsim)
+            st.button("Show similarity map",key=rdn_key_1,on_click = show_maxsim,args=(placeholder,"default_img"))
                 
     colu1,colu2,colu3 = st.columns([4,82,20])
     with colu2:
@@ -310,22 +310,20 @@ def render_answer(question,answer,index,res_img):
             
         
 #Each answer will have context of the question asked in order to associate the provided feedback with the respective question
-def write_chat_message(md, q,index):
+def write_chat_message(placeholder, md, q,index):
     if(st.session_state.show_columns):
         res_img = st.session_state.maxSimImages
     else:
         res_img = md['image']
-    chat = st.container()
-    with chat:
-        render_answer(q,md,index,res_img)
+    render_answer(placeholder,q,md,index,res_img)
     
 def render_all(placeholder):  
     index = 0
     for (q, a) in zip(st.session_state.questions_, st.session_state.answers_):
         index = index +1
         
-        write_user_message(placeholder,q)
-        write_chat_message(a, q,index)
+        placeholder = write_user_message(placeholder,q)
+        write_chat_message(placeholder, a, q,index)
 
 placeholder = st.empty()
 render_all(placeholder)