rerank model

pages/Semantic_Search.py

@@ -674,47 +674,45 @@ if(search_all_type == True or 1==1):
 
 
 def write_user_message(md,ans):
-    ans = ans["answer"][0]
-    col1, col2, col3 = st.columns([3,40,20])
-    
-    with col1:
-        st.image(USER_ICON, use_column_width='always')
-    with col2:
-        st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>Input Text: </div><div style='fontSize:25px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;font-style: italic;color:#e28743'>"+md['question']+"</div>", unsafe_allow_html = True)
-        if('query_sparse' in ans):
-            with st.expander("Expanded Query:"):
-                query_sparse = dict(sorted(ans['query_sparse'].items(), key=lambda item: item[1],reverse=True))
-                filtered_query_sparse = dict()
-                for key in query_sparse:
-                    filtered_query_sparse[key] = round(query_sparse[key], 2)
-                st.write(filtered_query_sparse)
-        if(st.session_state.input_is_rewrite_query == "enabled" and st.session_state.input_rewritten_query !=""):
-            with st.expander("Re-written Query:"):
-                st.json(st.session_state.input_rewritten_query,expanded = True)
+    if(len(ans["answer"])>0)
+        ans = ans["answer"][0]
+        col1, col2, col3 = st.columns([3,40,20])
+        
+        with col1:
+            st.image(USER_ICON, use_column_width='always')
+        with col2:
+            st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>Input Text: </div><div style='fontSize:25px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;font-style: italic;color:#e28743'>"+md['question']+"</div>", unsafe_allow_html = True)
+            if('query_sparse' in ans):
+                with st.expander("Expanded Query:"):
+                    query_sparse = dict(sorted(ans['query_sparse'].items(), key=lambda item: item[1],reverse=True))
+                    filtered_query_sparse = dict()
+                    for key in query_sparse:
+                        filtered_query_sparse[key] = round(query_sparse[key], 2)
+                    st.write(filtered_query_sparse)
+            if(st.session_state.input_is_rewrite_query == "enabled" and st.session_state.input_rewritten_query !=""):
+                with st.expander("Re-written Query:"):
+                    st.json(st.session_state.input_rewritten_query,expanded = True)
+                    
                 
-            
-    with col3:   
-        st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>Input Image: </div>", unsafe_allow_html = True)
-   
-        if(st.session_state.input_imageUpload == 'yes'):
-
-            if(st.session_state.input_rad_1 is not None and st.session_state.input_rad_1!=""):
-                num_str = str(int(st.session_state.input_rad_1.strip())-1)
-                img_file = parent_dirname+"/gen_images/"+st.session_state.image_prompt+"_gen_"+num_str+"-resized_display.jpg"
-            else:
-                img_file = parent_dirname+"/uploaded_images/"+st.session_state.img_doc.name.split(".")[0]+"-resized_display."+st.session_state.img_doc.name.split(".")[1]
+        with col3:   
+            st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>Input Image: </div>", unsafe_allow_html = True)
     
-            st.image(img_file)
-            if(st.session_state.input_rekog_label !=""):
-                with st.expander("Enriched Query Metadata:"):
-                        st.markdown('<p>'+json.dumps(st.session_state.input_rekog_directoutput)+'<p>',unsafe_allow_html=True)
-        else:
-            st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>None</div>", unsafe_allow_html = True)
-            
-            
-   
+            if(st.session_state.input_imageUpload == 'yes'):
 
-    st.markdown('---')
+                if(st.session_state.input_rad_1 is not None and st.session_state.input_rad_1!=""):
+                    num_str = str(int(st.session_state.input_rad_1.strip())-1)
+                    img_file = parent_dirname+"/gen_images/"+st.session_state.image_prompt+"_gen_"+num_str+"-resized_display.jpg"
+                else:
+                    img_file = parent_dirname+"/uploaded_images/"+st.session_state.img_doc.name.split(".")[0]+"-resized_display."+st.session_state.img_doc.name.split(".")[1]
+        
+                st.image(img_file)
+                if(st.session_state.input_rekog_label !=""):
+                    with st.expander("Enriched Query Metadata:"):
+                            st.markdown('<p>'+json.dumps(st.session_state.input_rekog_directoutput)+'<p>',unsafe_allow_html=True)
+            else:
+                st.markdown("<div style='fontSize:15px;padding:3px 7px 3px 7px;borderWidth: 0px;borderColor: red;borderStyle: solid;width: fit-content;height: fit-content;border-radius: 10px;'>None</div>", unsafe_allow_html = True)
+                
+        st.markdown('---')
         
 
 # def stem_(sentence):