BingRAG / app.py
vishnun's picture
Update app.py
8ae5128
raw
history blame
519 Bytes
import re
import streamlit as st
from PyBingScrapper.search import BingSearch
query = st.text_input("Enter the query you want to search for:,{PS: please enter the search term in brackets []}")
submit_button = st.button("Submit")
r = "\[(.*?)\]"
re_list = re.findall(r, query)
q_input = (' ').join(re_list)
hf_token = st.secrets["HF_TOKEN"]
if submit_button:
bingsearch = BingSearch(q_input)
b_res = bingsearch.get_results(num=5, max_lines=3)
st.info(bingsearch.rag_output(query, b_res, hf_token, 15))