Update app.py

app.py

@@ -228,20 +228,23 @@ def re_rank_candidates(query, candidates, method):
         # first sort by cross_score + bi_score
         filtered_sorted_result = sorted(filtered_sorted_by_encoder, key=lambda x: x[1]['gms'], reverse=True
         )
-    data_dicts = [{'query': item[0], **item[1]} for item in filtered_sorted_result]
-    # Convert the list of dictionaries into a DataFrame
-    df = pd.DataFrame(data_dicts)
-    return df
+    # data_dicts = [{'query': item[0], **item[1]} for item in filtered_sorted_result]
+    # # Convert the list of dictionaries into a DataFrame
+    # df = pd.DataFrame(data_dicts)
+    return filtered_sorted_result
+
 
-candidates = generate_query_expansion_candidates(query = user_query)
 # st.write("## Raw Candidates:")
 if st.button('Generated Expansion'): 
+    candidates = generate_query_expansion_candidates(query = user_query)
+    result = re_rank_candidates(user_query, candidates, method='gms')
+    st.write(result)
     # convert into dataframe
-    data_dicts = [{'query': key, **values} for key, values in candidates.items()]
-    df = pd.DataFrame(data_dicts)
+    # data_dicts = [{'query': key, **values} for key, values in candidates.items()]
+    # df = pd.DataFrame(data_dicts)
     # st.write(list(candidates.keys())[0:maxtags_sidebar])
     # st.write(df)
-    st.dataframe(df)
+    # st.dataframe(df)
     # st.success(raw_candidates)
 
 # if st.button('Rerank By Encoder'):