Update app.py

app.py

@@ -120,33 +120,45 @@ def aggregate(list_of_hits):
 aggregated_ranking = aggregate(query2outputs[search_query])
 qrels = load_qrels("dl19")
 
-if search_query or button_clicked:
+col1, col2 = st.columns([5, 5])
 
-    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"]
-
-        style = "style=\"color:grey;\"" if qrels[qid].get(result_id, 0) else  "style=\"color:red;\""
-        # 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('---')
+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('---')