import streamlit as st
import requests
import json
import st_auth 
authenticator=st_auth.get_authenticator()

st_auth.st_authenticate(authenticator)
st.header("Chatbot Editais")
url="https://i103tx5wac-vpce-03f49579eed9be4bf.execute-api.us-east-1.amazonaws.com/dev"
payload=[]
message_history=[]
if "user_prompt_history" not in st.session_state:
    st.session_state["user_prompt_history"]=[]
if "chat_answer_history" not in st.session_state:
    st.session_state["chat_answer_history"]=[]
if "chat_history" not in st.session_state:
    st.session_state["chat_history"] = []
prompt=st.chat_input(placeholder="Digite uma mensagem...",key="prompt")
for generated_response, user_query in zip(st.session_state["chat_answer_history"],st.session_state["user_prompt_history"]):
        st.chat_message("user").write(user_query)
        st.chat_message("assistant").write(generated_response)
if prompt:
    st.chat_message("user").write(prompt)
    with st.spinner("Generating response.."):
        payload=[{"role":"user","content":prompt}]
        content={"message":payload,"chat_history":st.session_state["chat_history"],"username":st.session_state["username"]}
        headers={"Content-type":"application/json","x-api-key":json.loads(st_auth.get_secret())['api-gateway-api-key']}
        generated_response=json.loads(requests.post(url,json=content,headers=headers).text)
        if 'chat_history' in generated_response:
            if st.session_state["chat_history"] == []:
                st.session_state["chat_history"] = generated_response['chat_history']
        if 'json' in generated_response: 
            generated_response=generated_response['json']
        if 'message' in generated_response and generated_response['message']=="Endpoint request timed out":
            generated_response="Falta de dados: Por favor encaminhe a conversa para a equipe desenvolvedora"
        #generated_response=[{"role":"user","content":prompt}]
        # sources= set([doc.metadata["source"] for doc in generated_response['context']])
        #formatted_response=f"{generated_response['answer']} \n\n {create_sources_string(sources)}"
        formatted_response=generated_response
        st.chat_message("assistant").write(formatted_response)
        st.session_state["user_prompt_history"].append(prompt)
        st.session_state["chat_answer_history"].append(formatted_response)
        st.session_state["chat_history"]=st.session_state["chat_history"]+[{"role":"user","content":prompt}]+[{"role":"assistant","content":formatted_response}]