import streamlit as st
import utils.user_utils as uu

if 'HR_tickets' not in st.session_state:
    st.session_state['HR_tickets'] =[]
if 'IT_tickets' not in st.session_state:
    st.session_state['IT_tickets'] =[]
if 'Transport_tickets' not in st.session_state:
    st.session_state['Transport_tickets'] =[]
if 'PINECONE_API_KEY' not in st.session_state:
    st.session_state['PINECONE_API_KEY'] = ''

def main():
    st.header("Automatic Ticket Classification Tool")
    st.write("We are here to help you. Please ask a question")
    user_input = st.text_input("🔎")

    st.sidebar.title("😎🗝️")
    st.session_state['PINECONE_API_KEY' ]= st.sidebar.text_input("What's your Pinecone API key?",type="password")

    if user_input:
        embedder = uu.initiate_embedder()

        index = uu.pull_from_pinecone(st.session_state['PINECONE_API_KEY'], 'tickets', embedder)

        similar_docs = uu.get_similar_docs(index, user_input)

        response = uu.get_answer(similar_docs, user_input)
        st.write(response)

        button = st.button("Submit Ticket?")

        if button:
            embedded_query = embedder.embed_query(user_input)
            department = uu.predict(embedded_query)
            st.write("your ticket has been sumbitted to : " + department)
            
            if department == 'HR':
                st.session_state['HR_tickets'].append(user_input)
            elif department == 'IT':
                st.session_state['IT_tickets'].append(user_input)
            else:
                st.session_state['Transport_tickets'].append(user_input)


if __name__ == '__main__':
    main()