RAG changes

pages/Multimodal_Conversational_Search.py

@@ -397,7 +397,10 @@ 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):
-    res_img = md['image']
+    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)