prasadnu commited on
Commit
8fd9bdf
·
1 Parent(s): 0b90a76

rerank model

Browse files
pages/Multimodal_Conversational_Search.py CHANGED
@@ -345,55 +345,55 @@ def render_answer(question,answer,index,res_img):
345
  with st.expander("Relevant Sources:"):
346
  with st.container():
347
  if(len(res_img)>0):
348
- with st.expander("Images:"):
349
 
350
- idx = 0
351
- print(res_img)
352
- for i in range(0,len(res_img)):
353
-
354
- if(st.session_state.input_is_colpali):
355
- if(st.session_state.show_columns == True):
356
- cols_per_row = 3
357
- st.session_state.image_placeholder=st.empty()
358
- with st.session_state.image_placeholder.container():
359
- row = st.columns(cols_per_row)
360
- for j, item in enumerate(res_img[i:i+cols_per_row]):
361
- with row[j]:
362
- st.image(item['file'])
363
-
364
- else:
365
- st.session_state.image_placeholder = st.empty()
366
- with st.session_state.image_placeholder.container():
367
- col3_,col4_,col5_ = st.columns([33,33,33])
368
- with col3_:
369
- st.image(res_img[i]['file'])
370
-
371
-
372
-
373
-
374
 
375
  else:
376
- if(res_img[i]['file'].lower()!='none' and idx < 2):
377
- col3,col4,col5 = st.columns([33,33,33])
378
- cols = [col3,col4]
379
- img = res_img[i]['file'].split(".")[0]
380
- caption = res_img[i]['caption']
381
 
382
- with cols[idx]:
383
-
384
- st.image(parent_dirname+"/figures/"+st.session_state.input_index+"/"+img+".jpg")
385
- #st.write(caption)
386
- idx = idx+1
387
- if(st.session_state.show_columns == True):
388
- st.session_state.show_columns = False
 
 
 
 
 
 
 
 
 
 
 
389
  #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)
390
  if(len(answer["table"] )>0):
391
- with st.expander("Table:"):
392
- df = pd.read_csv(answer["table"][0]['name'],skipinitialspace = True, on_bad_lines='skip',delimiter='`')
393
- df.fillna(method='pad', inplace=True)
394
- st.table(df)
395
- with st.expander("Raw sources:"):
396
- st.write(answer["source"])
397
 
398
 
399
 
 
345
  with st.expander("Relevant Sources:"):
346
  with st.container():
347
  if(len(res_img)>0):
348
+ #with st.expander("Images:"):
349
 
350
+ idx = 0
351
+ print(res_img)
352
+ for i in range(0,len(res_img)):
353
+
354
+ if(st.session_state.input_is_colpali):
355
+ if(st.session_state.show_columns == True):
356
+ cols_per_row = 3
357
+ st.session_state.image_placeholder=st.empty()
358
+ with st.session_state.image_placeholder.container():
359
+ row = st.columns(cols_per_row)
360
+ for j, item in enumerate(res_img[i:i+cols_per_row]):
361
+ with row[j]:
362
+ st.image(item['file'])
 
 
 
 
 
 
 
 
 
 
 
363
 
364
  else:
365
+ st.session_state.image_placeholder = st.empty()
366
+ with st.session_state.image_placeholder.container():
367
+ col3_,col4_,col5_ = st.columns([33,33,33])
368
+ with col3_:
369
+ st.image(res_img[i]['file'])
370
 
371
+
372
+
373
+
374
+
375
+ else:
376
+ if(res_img[i]['file'].lower()!='none' and idx < 2):
377
+ col3,col4,col5 = st.columns([33,33,33])
378
+ cols = [col3,col4]
379
+ img = res_img[i]['file'].split(".")[0]
380
+ caption = res_img[i]['caption']
381
+
382
+ with cols[idx]:
383
+
384
+ st.image(parent_dirname+"/figures/"+st.session_state.input_index+"/"+img+".jpg")
385
+ #st.write(caption)
386
+ idx = idx+1
387
+ if(st.session_state.show_columns == True):
388
+ st.session_state.show_columns = False
389
  #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)
390
  if(len(answer["table"] )>0):
391
+ #with st.expander("Table:"):
392
+ df = pd.read_csv(answer["table"][0]['name'],skipinitialspace = True, on_bad_lines='skip',delimiter='`')
393
+ df.fillna(method='pad', inplace=True)
394
+ st.table(df)
395
+ #with st.expander("Raw sources:"):
396
+ st.write(answer["source"])
397
 
398
 
399