Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
-from whoosh.index import create_in, open_dir
-from whoosh.qparser import QueryParser
+from pyserini.search.lucene import LuceneSearcher
+import json
 
 st.set_page_config(page_title="Pyserini x Datasets", page_icon='🌸', layout="centered")
-
+searcher = LuceneSearcher('index')
 
 
 cola, colb, colc = st.columns([5,4,5])
@@ -23,17 +23,21 @@ if search_query or button_clicked:
     myquery = parser.parse(search_query)
     num_results = None
     #search_results = searcher.search(myquery, limit=num_results)
-    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)
+    search_results = searcher.search(myquery, k=1000)
+    #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)
     for result in search_results[:10]:
         #keywords = ', '.join(result.key_terms('text'))
         #meta = result['meta']
         #st.write(f"<b>Document Keywords</b>: {keywords}", unsafe_allow_html=True)
         #st.write(f"<b>Meta Field</b>: {meta}", unsafe_allow_html=True)
         try:
-            st.write(result.highlights('text'), unsafe_allow_html=True)
+            st.write(
+            json.loads(result.raw["contents"]),
+            unsafe_allow_html=True
+            )
         except:
             pass
-        with st.expander("Document Text", expanded=False):
-            st.write(result['text'][:1600])
+        #with st.expander("Document Text", expanded=False):
+            #st.write(result['text'][:1600])
         st.write('---')