Update app.py

app.py

@@ -14,30 +14,6 @@ sys.path.append(str(path_root))
 st.set_page_config(page_title="PSC Runtime",
                    page_icon='🌸', layout="centered")
 
-# cola, colb, colc = st.columns([5, 4, 5])
-
-# colaa, colbb, colcc = st.columns([1, 8, 1])
-# with colbb:
-#     runtime = st.select_slider(
-#         'Select a runtime type',
-#         options=['PyTorch', 'ONNX Runtime'])
-#     st.write('Now using: ', runtime)
-
-
-# colaa, colbb, colcc = st.columns([1, 8, 1])
-# with colbb:
-#     encoder = st.select_slider(
-#         'Select a query encoder',
-#         options=['uniCOIL', 'SPLADE++ Ensemble Distil', 'SPLADE++ Self Distil'])
-#     st.write('Now Running Encoder: ', encoder)
-
-# if runtime == 'PyTorch':
-#     runtime = 'pytorch'
-#     runtime_index = 1
-# else:
-#     runtime = 'onnx'
-#     runtime_index = 0
-
 
 col1, col2 = st.columns([9, 1])
 with col1:
@@ -51,9 +27,8 @@ with col2:
 import torch
 fn = "dl19-gpt-3.5.pt"
 object = torch.load(fn)
-# for x for x in object:
-    
-# outputs = [x[2] for x in object]
+
+
 outputs = object[2]
 query2outputs = {}
 for output in outputs:
@@ -62,8 +37,11 @@ for output in outputs:
     query = list(all_queries)[0]
     query2outputs[query] = [x['hits'] for x in output]
 
-search_query = sorted(query2outputs)[0]
-
+# search_query = sorted(query2outputs)[0]
+search_query = st.selectbox(
+    "Choose a query from the list",
+    sorted(query2outputs)
+)
 
 def preferences_from_hits(list_of_hits):
     docid2id = {}
@@ -109,12 +87,6 @@ def aggregate(list_of_hits):
     y_optimal = KemenyOptimalAggregator().aggregate(preferences)
     # y_optimal = BordaRankAggregator().aggregate(preferences)
 
-    # print("-------------------------------------")
-    # print("preference:")
-    # print(preferences)
-    # print("preferences shape: ", preferences.shape)
-    # print("y_optimal: ", y_optimal)
-
     return [id2doc[id] for id in y_optimal]
 
 
@@ -130,7 +102,7 @@ def write_ranking(search_results):
         result_id = result["docid"]
         contents = result["content"]
 
-        label = qrels[qid].get(result_id, 0)
+        label = qrels[str(qid)].get(str(result_id), 0)
         if label == 3:
             style = "style=\"color:blue;\""
         elif label == 2:
@@ -155,13 +127,13 @@ def write_ranking(search_results):
 
 aggregated_ranking = aggregate(query2outputs[search_query])
 qrels = load_qrels("dl19")
-
 col1, col2 = st.columns([5, 5])
 
-with col1:
-    if search_query or button_clicked:
-        write_ranking(search_results=query2outputs[search_query][0])
-
-with col2:
-    if search_query or button_clicked:
-        write_ranking(search_results=aggregated_ranking)
+if search_query:
+    with col1:
+        if search_query or button_clicked:
+            write_ranking(search_results=query2outputs[search_query][0])
+    
+    with col2:
+        if search_query or button_clicked:
+            write_ranking(search_results=aggregated_ranking)