Update app.py

app.py

@@ -46,7 +46,7 @@ if name and model_name:
         query2outputs[query] = [x['hits'] for x in output]
 
     search_query = st.selectbox(
-        "",
+        "Choose a query from the list",
         sorted(query2outputs),
         # index=None,
         # placeholder="Choose a query from the list..."
@@ -98,10 +98,9 @@ if name and model_name:
     
         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)
+
+    def write_ranking(search_results, text):
+        st.write(f'<p align=\"right\" style=\"color:grey;\"> {text} ms</p>', unsafe_allow_html=True)
 
         qid = {result["qid"] for result in search_results}
         assert len(qid) == 1
@@ -142,8 +141,8 @@ if name and model_name:
     if search_query:
         with col1:
             if search_query or button_clicked:
-                write_ranking(search_results=query2outputs[search_query][0])
+                write_ranking(search_results=query2outputs[search_query][0], "w/o PSC")
         
         with col2:
             if search_query or button_clicked:
-                write_ranking(search_results=aggregated_ranking)
+                write_ranking(search_results=aggregated_ranking, "w/ PSC")