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))