Spaces:
Runtime error
Runtime error
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.", | |
} | |
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( | |
'How would you like to be contacted?', | |
indexes.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(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 | |
""" | |
) |