| import json | |
| import streamlit as st | |
| from pyserini.search.lucene import LuceneSearcher | |
| class SearchApplication: | |
| def __init__(self): | |
| self.title = "Awesome ChatGPT repositories search" | |
| self.set_page_config() | |
| self.searcher = self.set_searcher() | |
| st.header(self.title) | |
| col1, col2 = st.columns(2) | |
| with col1: | |
| self.query = st.text_input("Search English words", value="") | |
| with col2: | |
| st.write("#") | |
| self.search_button = st.button("🔎") | |
| st.caption( | |
| "You can search for open-source software from [500+ " | |
| " repositories](https://github.com/taishi-i/awesome-ChatGPT-repositories)." | |
| ) | |
| st.write("#") | |
| self.show_popular_words() | |
| self.show_search_results() | |
| def set_page_config(self): | |
| st.set_page_config( | |
| page_title=self.title, | |
| page_icon="😎", | |
| layout="centered", | |
| ) | |
| def set_searcher(self): | |
| searcher = LuceneSearcher("indexes/docs") | |
| return searcher | |
| def show_popular_words(self): | |
| st.caption("Popular words") | |
| word1, word2, word3, word4, _ = st.columns(5) | |
| with word1: | |
| button1 = st.button("Prompt") | |
| if button1: | |
| self.query = "prompt" | |
| with word2: | |
| button2 = st.button("Chatbot") | |
| if button2: | |
| self.query = "chatbot" | |
| with word3: | |
| button3 = st.button("Langchain") | |
| if button3: | |
| self.query = "langchain" | |
| with word4: | |
| button4 = st.button("Extension") | |
| if button4: | |
| self.query = "extension" | |
| def show_search_results(self): | |
| if self.query or self.search_button: | |
| st.write("#") | |
| search_results = self.searcher.search(self.query, k=500) | |
| num_search_results = len(search_results) | |
| st.write(f"{num_search_results} results") | |
| for result in search_results: | |
| data_json = json.loads(result.raw) | |
| description = data_json["description"] | |
| url = data_json["url"] | |
| project_name = data_json["project_name"] | |
| main_topic = data_json["main_topic"] | |
| st.subheader(f"[{project_name}]({url})") | |
| st.markdown(description) | |
| st.caption(f"{main_topic}") | |
| st.write("#") | |
| def main(): | |
| SearchApplication() | |
| if __name__ == "__main__": | |
| main() | |