import streamlit as st
import requests
import streamlit as st
import json

siteUrl = "https://aloqas-aloqas-qa-fastapi.hf.space"

userContext = None
context = None
error = 0
limit = 9000

with st.sidebar:
    st.title("Configuration")
    model = st.selectbox('Which model would you like to use?',
                         ('SqueezeBERT', 'BERT', 'DeBERTa-v2'))
    contextSelect = st.radio("Pick a context mode:", ["Text", "File"])
    if contextSelect == "File":
        userContext = st.file_uploader("Pick a file for the context", accept_multiple_files=True)
        context = "/uploadfile/"
    else:
        userContext = st.text_area("write the context")
        context = "/contextText/"

st.title("ALOQAS")

"""
bla bla bla
"""

if "messages" not in st.session_state:
    st.session_state["messages"] = [
        {"role": "assistant", "content": "Hi, I'm ALOQAS. How can I help you?"}
    ]

if prompt := st.chat_input(placeholder="Write to ALOQAS..."):
    st.session_state.messages.append({"role": "user", "content": prompt})
    params = {'texte': prompt, "model": model}
    if userContext != '' and len(userContext) <= limit:
        if contextSelect == "File" and userContext:
            # Préparation de la liste des fichiers pour l'envoi
            files = [("file", (file.name, file, file.type)) for file in userContext]
            response = requests.post(siteUrl + context, params=params, files=files)
        else:
            params["context"] = userContext
            request_url = f"{siteUrl}/{model.lower()}/?context={userContext}&question={prompt}"
            response = requests.post(request_url)
            print(response.text)
            print(response)
            st.session_state.messages.append({"role": "assistant", "content": json.loads(response.text)["answer"]})

    elif userContext == '' :
        error = 1
    else : 
        error = 2

    # st.write("Statut de la requête:", response.status_code)
    # st.write("Réponse du serveur:", response.text)

    if userContext != '' and len(userContext) <= limit:
        with st.sidebar:
            st.title("Statistics on the last answer")
            st.write(f"Score: {round(json.loads(response.text)['score'], 3)}")
            st.write(f"Start: {json.loads(response.text)['start']}")
            st.write(f"End: {json.loads(response.text)['end']}")

for msg in st.session_state.messages:
    st.chat_message(msg["role"]).write(msg['content'])

if error == 1:
    message_rouge = "⚠️ Please provide a context via the menu on your left."
    st.markdown(f'<div style="color: white; background-color: #ff4444; padding: 10px; border-radius: 5px;">{message_rouge}</div>', unsafe_allow_html=True)
    error = 0
elif error == 2:
    message_rouge = f"⚠️ The message you submitted was too long, please reload the conversation and submit something shorter than {limit} caracter."
    st.markdown(f'<div style="color: white; background-color: #ff4444; padding: 10px; border-radius: 5px;">{message_rouge}</div>', unsafe_allow_html=True)
    error = 0