Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from pyserini.search.lucene import LuceneSearcher
 import json
+import time
 
 st.set_page_config(page_title="Pyserini x Datasets", page_icon='🌸', layout="centered")
 searcher = LuceneSearcher('index')
@@ -22,9 +23,11 @@ with col2:
 if search_query or button_clicked:
     num_results = None
     #search_results = searcher.search(myquery, limit=num_results)
-    search_results = searcher.search(search_query, k=10)
+    t_0 = time.time()
+    search_results = searcher.search(search_query)
+    search_time = time.time() - t_0
     #st.write(dir(search_results[0]))
-    #st.write(f"<p align=\"right\" style=\"color:grey;\">Retrieved {len(search_results):,.0f} documents in {search_results.runtime*1000:.2f} ms</p>", unsafe_allow_html=True)
+    st.write(f"<p align=\"right\" style=\"color:grey;\">Retrieved {len(search_results):,.0f} documents in {search_time*1000:.2f} ms</p>", unsafe_allow_html=True)
     for result in search_results[:10]:
         #keywords = ', '.join(result.key_terms('text'))
         #meta = result['meta']
@@ -32,12 +35,10 @@ if search_query or button_clicked:
         result = json.loads(result.raw)
         doc = result["contents"]
         result_id = result["id"]
-        st.write(f"<b>Document ID</b>: {result_id}", unsafe_allow_html=True)
         try:
-            st.write(
-            doc[:1000],
-            unsafe_allow_html=True
-            )
+            st.write(doc[:1000], unsafe_allow_html=True)
+            st.write(f"<div align="right"><b>Document ID</b>: {result_id}</div>", unsafe_allow_html=True)
+
         except:
             pass
         #with st.expander("Document Text", expanded=False):