Update app.py

app.py

@@ -117,48 +117,51 @@ def aggregate(list_of_hits):
 
     return [id2doc[id] for id in y_optimal]
 
+
+def write_ranking(search_results):
+    st.write(
+        f'<p align=\"right\" style=\"color:grey;\"> Before aggregation for query [{search_query}] ms</p>', unsafe_allow_html=True)
+
+    qid = {result["qid"] for result in search_results}
+    assert len(qid) == 1
+    qid = list(qid)[0]
+
+    for i, result in enumerate(search_results):
+        result_id = result["docid"]
+        contents = result["content"]
+
+        label = qrels[qid].get(result_id, 0)
+        if label == 3:
+            style = "style=\"color:blue;\""
+        elif label == 2:
+            style = "style=\"color:green;\""
+        elif label == 1:
+            style = "style=\"color:red;\""
+        else:
+            style = "style=\"color:grey;\""
+
+        print(qid, result_id, label, style)
+        # output = f'<div class="row"> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id} | <b>Score</b>:{result_score:.2f}</div>'
+        output = f'<div class="row" {style}> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id}'
+
+        try:
+            st.write(output, unsafe_allow_html=True)
+            st.write(
+                f'<div class="row" {style}>{contents}</div>', unsafe_allow_html=True)
+
+        except:
+            pass
+        st.write('---')
+
 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:
-    
-        num_results = None
-        t_0 = time.time()
-        # search_results = query2outputs[search_query][0] # first from the 20
-        search_results = aggregated_ranking
-    
-        st.write(
-            f'<p align=\"right\" style=\"color:grey;\"> Before aggregation for query [{search_query}] ms</p>', unsafe_allow_html=True)
-    
-        qid = {result["qid"] for result in search_results}
-        assert len(qid) == 1
-        qid = list(qid)[0]
-    
-        for i, result in enumerate(search_results):
-            result_id = result["docid"]
-            contents = result["content"]
-
-            label = qrels[qid].get(result_id, 0)
-            if label == 3:
-                style = "style=\"color:blue;\""
-            elif label == 2:
-                style = "style=\"color:green;\""
-            elif label == 1:
-                style = "style=\"color:red;\""
-            else:
-                style = "style=\"color:grey;\""
-
-            # output = f'<div class="row"> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id} | <b>Score</b>:{result_score:.2f}</div>'
-            output = f'<div class="row" {style}> <b>Rank</b>: {i+1} | <b>Document ID</b>: {result_id}'
-    
-            try:
-                st.write(output, unsafe_allow_html=True)
-                st.write(
-                    f'<div class="row" {style}>{contents}</div>', unsafe_allow_html=True)
-    
-            except:
-                pass
-            st.write('---')
+        write_ranking(search_results=aggregated_ranking):