reddit-cmv / app.py
cakiki's picture
Update app.py
eefd1fd
raw
history blame
1.8 kB
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"<div>{(result.contents)}</div><br>"
)
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
"""
)