BingRAG / app.py
vishnun's picture
Update app.py
3087fbb
raw
history blame
602 Bytes
import re
import streamlit as st
from PyBingScrapper.search import BingSearch
st.title("Bing - RAG")
st.subheader('Implementation of RAG over search results bing search', divider='rainbow')
query = st.text_input("Enter the query you want to search for:")
st.caption("Example: Why is there a huge conflict between India and Pakistan?")
submit_button = st.button("Submit")
hf_token = st.secrets["HF_TOKEN"]
if submit_button:
bingsearch = BingSearch(query)
b_res = bingsearch.get_results(num=5, max_lines=3)
#st.text(b_res)
st.info(bingsearch.rag_output(query, b_res, hf_token, 15))