Update app.py

app.py

@@ -21,6 +21,18 @@ titles = {"shoe-sizes": "Shoes sizes should be the same for both men and women",
            "best-time-period": "Today is the best time period in human history to be alive for the vast majority of people",
           }
 
+num_docs = {"shoe-sizes": 623,
+           "social-media": 1133,
+           "depression": 982,
+           "metric-system": 4679,
+           "best-time-period": 2516,
+          }
+
+cmv'-the-others-have-it-worse-argument-is-terrible-and-should-never-be-used-in-an-actual-conversation-with-a-depressed-person 982
+cmv'-today-is-the-best-time-period-in-human-history-to-be-alive-for-the-vast-majority-of-people. 2516
+cmv'-there-shouldnt-be-anything-other-than-the-metric-system. 4679
+cmv'-shoe-sizes-should-be-the-same-for-both-men-and-women 623
+cmv'-social-media-is-the-most-destructive-addiction-in-our-society 1133
 
 def result_html(result):
     return (
@@ -51,9 +63,14 @@ with col2:
 
 
 if search_query or button_clicked:
-    search_results = searcher.search(search_query, k=10)
-    for result in search_results:
-        st.write(result_html(result), unsafe_allow_html=True)
+    k=num_docs[discussion]
+    search_results = searcher.search(search_query, k=k)
+    if len(search_results) == 0:
+        st.write("Retrieved 0 documents")
+    else:
+        st.write(f"retrieved {len(search_results)} out of {k} total posts")
+        for result in search_results:
+            st.write(result_html(result), unsafe_allow_html=True)
     
 with st.expander("Instructions", expanded=False):
     st.markdown(