rerank model

pages/Multimodal_Conversational_Search.py

@@ -345,55 +345,55 @@ def render_answer(question,answer,index,res_img):
         with st.expander("Relevant Sources:"):
             with st.container():
                 if(len(res_img)>0):
-                    with st.expander("Images:"):
+                    #with st.expander("Images:"):
                         
-                        idx = 0
-                        print(res_img)
-                        for i in range(0,len(res_img)):
-                            
-                            if(st.session_state.input_is_colpali):
-                                if(st.session_state.show_columns == True):
-                                    cols_per_row = 3
-                                    st.session_state.image_placeholder=st.empty()
-                                    with st.session_state.image_placeholder.container():
-                                        row = st.columns(cols_per_row)
-                                    for j, item in enumerate(res_img[i:i+cols_per_row]):
-                                        with row[j]:
-                                            st.image(item['file'])
-                                    
-                                else:
-                                    st.session_state.image_placeholder = st.empty()
-                                    with st.session_state.image_placeholder.container():
-                                        col3_,col4_,col5_ = st.columns([33,33,33])
-                                        with col3_:
-                                            st.image(res_img[i]['file'])
-                                    
-                                    
-
-                                
+                    idx = 0
+                    print(res_img)
+                    for i in range(0,len(res_img)):
+                        
+                        if(st.session_state.input_is_colpali):
+                            if(st.session_state.show_columns == True):
+                                cols_per_row = 3
+                                st.session_state.image_placeholder=st.empty()
+                                with st.session_state.image_placeholder.container():
+                                    row = st.columns(cols_per_row)
+                                for j, item in enumerate(res_img[i:i+cols_per_row]):
+                                    with row[j]:
+                                        st.image(item['file'])
                                 
                             else:
-                                if(res_img[i]['file'].lower()!='none' and idx < 2):
-                                    col3,col4,col5 = st.columns([33,33,33])
-                                    cols = [col3,col4]
-                                    img = res_img[i]['file'].split(".")[0]
-                                    caption = res_img[i]['caption']
+                                st.session_state.image_placeholder = st.empty()
+                                with st.session_state.image_placeholder.container():
+                                    col3_,col4_,col5_ = st.columns([33,33,33])
+                                    with col3_:
+                                        st.image(res_img[i]['file'])
                                 
-                                    with cols[idx]:
-                                        
-                                        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
+                                
+
+                            
+                            
+                        else:
+                            if(res_img[i]['file'].lower()!='none' and idx < 2):
+                                col3,col4,col5 = st.columns([33,33,33])
+                                cols = [col3,col4]
+                                img = res_img[i]['file'].split(".")[0]
+                                caption = res_img[i]['caption']
+                            
+                                with cols[idx]:
+                                    
+                                    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
                 #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:"):
-                        df = pd.read_csv(answer["table"][0]['name'],skipinitialspace = True, on_bad_lines='skip',delimiter='`')
-                        df.fillna(method='pad', inplace=True)
-                        st.table(df)
-                with st.expander("Raw sources:"):
-                    st.write(answer["source"])
+                    #with st.expander("Table:"):
+                    df = pd.read_csv(answer["table"][0]['name'],skipinitialspace = True, on_bad_lines='skip',delimiter='`')
+                    df.fillna(method='pad', inplace=True)
+                    st.table(df)
+                #with st.expander("Raw sources:"):
+                st.write(answer["source"])