import streamlit as st
import requests
import json

# Load data.json
with open('data.json') as f:
    data = json.load(f)

client_endpoint = "https://olivier-truong-mistral-super-fast.hf.space/chat"

def ask_question(question):
    if question in data:
        return data[question]
    else:
        response = requests.post(client_endpoint, json={"context": question})
        if response.status_code == 200:
            answer = response.json()['response']
            data[question] = answer
            with open('data.json', 'w') as f:
                json.dump(data, f)
            return answer
        else:
            return "Error: Failed to retrieve an answer."

def main():
    st.title("Question Answering System")

    question = st.text_input("Ask your question:")

    if st.button("Ask"):
        if question:
            answer = ask_question(question)
            st.write("Answer:", answer)
        else:
            st.write("Please enter a question.")

if __name__ == "__main__":
    main()