Update app.py

app.py

@@ -187,8 +187,6 @@ def generate_query_expansion_candidates(query):
         value = final_candidates[query_candidate]
         value['gms'] = query_gms_dict.get(query_candidate, 0)
         final_candidates[query_candidate] = value
-    # Total Results
-    st.write("E-Commerce Query Expansion Candidates: \n")
     return final_candidates
 
 def re_rank_candidates(query, candidates, method):
@@ -232,8 +230,10 @@ def re_rank_candidates(query, candidates, method):
         )
     return filtered_sorted_result
 
-col1, col2 = st.columns(2)
+
 if st.button('Generated Expansion'): 
+    st.write("E-Commerce Query Expansion Candidates: \n")
+    col1, col2 = st.columns(2)
     candidates = generate_query_expansion_candidates(query = user_query)
     with col1:
         st.subheader('Raw Candidates:')
@@ -243,5 +243,5 @@ if st.button('Generated Expansion'):
     with col2:
         st.subheader('Rerank By GMS:')
         candidates_rerank = re_rank_candidates(user_query, candidates, method='gms')[:maxtags_sidebar]
-        result = [{item[0]: item[1]['gms']} for item in candidates_rerank]
+        result = [(item[0], item[1]['gms']) for item in candidates_rerank]
         st.write(result)