Update app.py

app.py

@@ -8,7 +8,7 @@ embedding_path = "abokbot/wikipedia-embedding"
 
 st.header("Wikipedia Search Engine app")
 
-st_model_load = st.text('Loading encoders, embeddings and dataset (takes about 5min)')
+st_model_load = st.text('Loading embeddings, encoders and dataset (takes about 5min)')
 
 @st.cache_resource
 def load_embedding():
@@ -41,7 +41,7 @@ def load_wikipedia_dataset():
     return dataset
     
 dataset = load_wikipedia_dataset()
-st.success('Loading done')
+st.success('App ready')
 st_model_load.text("")
     
 if 'text' not in st.session_state:
@@ -82,14 +82,17 @@ def search():
     print("\n-------------------------\n")
     print("Top-3 Cross-Encoder Re-ranker hits")
     st.subheader("Top-3 Search results")
-    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"]
-        }
+    results = []
+    for hit in hits[:3]:
+        results.append
+        (
+            {
+                "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
 
 
@@ -102,7 +105,7 @@ print(st.session_state.results)
 if len(st.session_state.results) > 0:
     with st.container():
         st.subheader("Search results")
-        for i, result in st.session_state.results:
+        for result in st.session_state.results:
             st.markdown("score: " + result["score"])
             st.markdown("title: " + result["title"])
             st.markdown("abstract: " + result["abstract"])