import streamlit as st from pyserini.search.lucene import LuceneSearcher st.set_page_config( page_title="Reddit CMV", page_icon="📄", layout="centered" ) index_locations = {"shoe-sizes": "index/cmv'-shoe-sizes-should-be-the-same-for-both-men-and-women", "social-media": "index/cmv'-social-media-is-the-most-destructive-addiction-in-our-society", "depression": "index/cmv'-the-others-have-it-worse-argument-is-terrible-and-should-never-be-used-in-an-actual-conversation-with-a-depressed-person", "metric-system": "index/cmv'-there-shouldnt-be-anything-other-than-the-metric-system.", "best-time-period": "index/cmv'-today-is-the-best-time-period-in-human-history-to-be-alive-for-the-vast-majority-of-people.", } @st.cache(suppress_st_warning=True, allow_output_mutation=True, show_spinner=False) def search_chat_noir(key, search_query): return search(api_key=key, query=search_query) def result_html(result): return ( f"
{(result.contents)}

" ) discussion = st.selectbox( 'Choose discussion', index_locations.keys()) searcher = LuceneSearcher(index_locations[discussion]) searcher.set_rm3() cola, colb, colc = st.columns([5,4,5]) with colb: st.write(discussion) col1, col2 = st.columns([9, 1]) with col1: search_query = st.text_input(label="", placeholder="Search" ) with col2: st.write('#') button_clicked = st.button("🔎") 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) with st.expander("Instructions", expanded=False): st.markdown( """ INSTRUCTIONS HERE """ )