import streamlit as st
import requests

st.title("LLM-based RAG Search")

# Input for user query
query = st.text_input("Enter your query:")

if st.button("Search"):
    with st.spinner("Searching and generating response..."):
        # Make a POST request to the Flask API
        print("accessing ", "<Flask app string>", " with query ", query)
        api_url = "https://omkar008-api.hf.space/query"
        response = requests.post(api_url, json={"query": query})
    
        # implement the flask call here
        
        if response.status_code == 200:
            # Display the generated answer
            answer = response.json().get('answer', "No answer received.")
            st.write("Answer:", answer)
        else:
            st.error(f"Error: {response.status_code}")