some upgrades and downgrades to test app

app.py

@@ -45,20 +45,18 @@ class DocumentSearch:
         return sorted(results, key=lambda x: x['score'], reverse=True)[:k]
 
 
-enc_path = "ivan-savchuk/msmarco-distilbert-dot-v5-tuned-full-v1"
-idx_path = "idx_vectors.index"
-cross_enc_path = "ivan-savchuk/cross-encoder-ms-marco-MiniLM-L-12-v2-tuned_mediqa-v1"
-docs_path = "docs.json"
-# get instance of DocumentSearch class
-surfer = DocumentSearch(
-    labels_path=docs_path,
-    encoder_path=enc_path,
-    index_path=idx_path,
-    cross_encoder_path=cross_enc_path
-)
-
-
 if __name__ == "__main__":
+    enc_path = "ivan-savchuk/msmarco-distilbert-dot-v5-tuned-full-v1"
+    idx_path = "idx_vectors.index"
+    cross_enc_path = "ivan-savchuk/cross-encoder-ms-marco-MiniLM-L-12-v2-tuned_mediqa-v1"
+    docs_path = "docs.json"
+    # get instance of DocumentSearch class
+    surfer = DocumentSearch(
+        labels_path=docs_path,
+        encoder_path=enc_path,
+        index_path=idx_path,
+        cross_encoder_path=cross_enc_path
+    )
     # streamlit part starts here with title
     st.title('Medical Search')
     # here we have input space
@@ -79,13 +77,50 @@ if __name__ == "__main__":
         with st.container():
             # show which query was entered, and what was searching time
             st.write(f"**Results Related to:** {query} ({elapsed_time} sec.)")
-            # then we use loop to show results
-            for i, answer in enumerate(results):
-                # answer starts with header
-                st.subheader(f"Answer {i+1}")
-                # cropped answer
-                doc = answer["doc"][:150] + "..."
-                # and url to the full answer
-                url = answer["url"]
-                # then we display it
-                st.markdown(f"{doc}\n[**Read More**]({url})\n")
+            # answer starts with header
+            st.subheader(f"Answer 1")
+            # cropped answer
+            doc = results[0]["doc"][:150] + "..."
+            # and url to the full answer
+            url = answer[0]["url"]
+            # then we display it
+            st.markdown(f"{doc}\n[<span style=\"color:blue\">**Read More**</span>]({url})\n")
+            
+            # answer starts with header
+            st.subheader(f"Answer 2")
+            # cropped answer
+            doc = results[1]["doc"][:150] + "..."
+            # and url to the full answer
+            url = answer[1]["url"]
+            # then we display it
+            st.markdown(f"{doc}\n[<span style=\"color:blue\">**Read More**</span>]({url})\n")
+            
+            # answer starts with header
+            st.subheader(f"Answer 3")
+            # cropped answer
+            doc = results[2]["doc"][:150] + "..."
+            # and url to the full answer
+            url = answer[2]["url"]
+            # then we display it
+            st.markdown(f"{doc}\n[<span style=\"color:blue\">**Read More**</span>]({url})\n")
+            
+            # answer starts with header
+            st.subheader(f"Answer 4")
+            # cropped answer
+            doc = results[3]["doc"][:150] + "..."
+            # and url to the full answer
+            url = answer[3]["url"]
+            # then we display it
+            st.markdown(f"{doc}\n[<span style=\"color:blue\">**Read More**</span>]({url})\n")
+            
+            # answer starts with header
+            st.subheader(f"Answer 5")
+            # cropped answer
+            doc = results[4]["doc"][:150] + "..."
+            # and url to the full answer
+            url = answer[4]["url"]
+            # then we display it
+            st.markdown(f"{doc}\n[<span style=\"color:blue\">**Read More**</span>]({url})\n")
+            
+            st.markdown("---")
+            st.markdown("Author: Ivan Savchuk. 2022")