Update app.py

app.py

@@ -66,7 +66,7 @@ def search():
 
     ##### Re-Ranking #####
     # Now, score all retrieved passages with the cross_encoder
-    cross_inp = [[query, dataset["text"][hit['corpus_id']]] for hit in hits]
+    cross_inp = [[query, dataset[hit['corpus_id']]["text"]] for hit in hits]
     cross_scores = cross_encoder.predict(cross_inp)
 
     # Sort results by the cross-encoder scores
@@ -79,13 +79,30 @@ def search():
     print("\n-------------------------\n")
     print("Top-3 Cross-Encoder Re-ranker hits")
     st.subheader("Top-3 Search results")
-    for hit in hits[:3]:
-        st.markdown("score: " +  round(hit['cross-score'], 3)),
-        st.markdown("title: " + dataset["title"][hit['corpus_id']]), 
-        st.markdown("substract: " + dataset["text"][hit['corpus_id']].replace("\n", " ")), 
-        st.markdown("link: " + dataset["url"][hit['corpus_id']])
+    results: dict[str, dict] = {}
+    for i, hit in enumerate(hits[:3]):
+        results[i] = {
+            "score": round(hit['cross-score'], 3),
+            "title": dataset[hit['corpus_id']]["title"],
+            "abstract": dataset[hit['corpus_id']]["text"].replace("\n", " "),
+            "link": dataset[hit['corpus_id']]["url"]
+        }
+    st.session_state.results = results
 
 
 # search button
 st_search_button = st.button('Search', on_click=search)
 
+if 'results' not in st.session_state:
+    st.session_state.results = {}
+
+if len(st.session_state.results) > 0:
+    with st.container():
+        st.subheader("Search results")
+        for result in st.session_state.questions:
+            for k,v in result.items():
+                st.markdown("score: " + results["score"])
+                st.markdown("title: " + results["title"])
+                st.markdown("abstract: " + results["abstract"])
+                st.markdown("link: " + results["link"])
+                st.text("")