'''
    This is the main file of the app. This file contains the code for the streamlit app.
'''

import time
import streamlit as st
from streamlit_chat import message

from job_description_generator import predict_job_description, get_job_description_conversation
from job_description_fixer import fix_job_description, get_job_description_fixer_conversation
from interview_questions_generator import (predict_interview_question,
                                           get_interview_questions_conversation)
from cover_letter_generator import get_cover_letter

conversation = get_job_description_conversation()
if 'generator_conversation' not in st.session_state:
    st.session_state['generator_conversation'] = conversation

fixer_conversation = get_job_description_fixer_conversation()
if 'fixer_conversation' not in st.session_state:
    st.session_state['fixer_conversation'] = fixer_conversation
    
interview_questions_conversation = get_interview_questions_conversation()
if 'interview_questions_conversation' not in st.session_state:
    st.session_state['interview_questions_conversation'] = interview_questions_conversation

def message_writer(input_text, response):
    '''
        Writes the messages to the chat window.
    '''
    messages = []
    current_message = ""
    current_is_user = True
    for historical_message in response['history']:
        if "human" in historical_message.lower():
            messages.append([current_message, current_is_user])
            current_message = historical_message.replace("Human:", "")
            current_is_user = True
        elif "JobGPT" in historical_message:
            messages.append([current_message, current_is_user])
            current_message = historical_message.replace("JobGPT:", "")
            current_is_user = False
        else:
            current_message = current_message + "\n" + historical_message
    messages.append([current_message, current_is_user])
    for message_to_send, is_user in messages:
        if message_to_send.strip() != "":
            message(message_to_send, is_user=is_user)
    message(input_text, is_user=True)
    message(response['prediction'], is_user=False)
    return 0


def setup():
    """
    Streamlit related setup. This has to be run for each page.
    """
    hide_streamlit_style = """
    
    <style>
    #MainMenu {visibility: hidden;}
    footer {visibility: hidden;}
    </style>
    """
    st.markdown(hide_streamlit_style, unsafe_allow_html=True)

def main():
    '''
     Main function of the app.
    '''
    setup()

    #create a sidebar where you can select your page
    st.sidebar.title("JobGPT")
    st.sidebar.markdown("---")
    #selector
    page = st.sidebar.selectbox(
        "Select a page", ["Home",
                          "Job Description Generator",
                          "Job Description Fixer",
                          "Cover Letter Generator",
                          "interview questions generator"])
    if page == "Home":
        st.title("JobGPT")
        st.write("Select a page in the sidebar to get started.")
        st.write("### Available options:")
        st.write("1. Job Description Generator")
        st.write("2. Job Description Fixer")
        st.write("3. Cover Letter Generator")
        st.write("4. Interview Questions Generator")
        st.markdown("---")

    elif page == "Job Description Generator":
        container_one = st.container()
        container_one.title("A Job Description Generating Chatbot")
        container_one.markdown(
            "JobGPT is a chatbot that generates job descriptions. \
                This is built just for demo purpose."
        )
        input_text = container_one.text_input(
            "Prompt",
            "Hi, can you please help me generate an unbiased job description?")
        button = container_one.button("Send")

        st.sidebar.markdown("---")
        st.sidebar.markdown("Click on `new chat` to start a new chat. \
                History will be cleared and you'll lose access to current chat."
                           )
        clear_session = st.sidebar.button("New Chat")

        if clear_session:
            st.session_state['generator_conversation'] = conversation
        container_one.markdown("---")

        initial_message = "Hello, how can I help you?"
        message(initial_message)

        if button:
            response = predict_job_description(input_text,
                                               st.session_state['generator_conversation'])
            message_writer(input_text, response)
    elif page == "Job Description Fixer":
        container_two = st.container()
        container_two.title("A Job Description Fixing Chatbot")
        container_two.markdown(
            "JobGPT is a chatbot that fixes job descriptions. This is built just for demo purpose."
        )
        input_text = container_two.text_area(
            "Prompt",
            "Hi, can you please help me fix my job description? It's biased.")
        button = container_two.button("Send")

        st.sidebar.markdown("---")
        st.sidebar.markdown("Click on `new chat` to start a new chat. \
                History will be cleared and you'll lose access to current chat."
                           )
        clear_session = st.sidebar.button("New Chat")

        if clear_session:
            st.session_state['fixer_conversation'] = fixer_conversation
        container_two.markdown("---")

        initial_message = "Hello, how can I help you?"
        message(initial_message)

        if button:
            response = fix_job_description(
                input_text, st.session_state['fixer_conversation'])
            message_writer(input_text, response)

    elif page == "Cover Letter Generator":
        container_three = st.container()
        container_three.title("A Cover Letter Generating Chatbot")
        container_three.markdown( "JobGPT is a chatbot that generates cover letters. \
                                 This is built just for demo purpose.")
        container_three.markdown("---")
        uploaded_file = container_three.file_uploader("Upload your resume", type=["pdf"])
        if uploaded_file is not None:
            with open("resume.pdf", "wb") as file_io:
                file_io.write(uploaded_file.getbuffer())

            with st.spinner('Uploading...'):
                time.sleep(1)
            container_three.success('Uploaded!')
            container_three.markdown("---")
            form = container_three.form(key='my_form')
            title = form.text_input("Job Title (required)", placeholder="VP of Engineering")
            company = form.text_input("Company Name (required)", placeholder="Google")
            more_info = form.text_area("More Info",
                                       help="Add more info about you or the job in natural language",
                                       placeholder="I am a software engineer with 5 years of experience. The job focuses on building a new product in healthcare sector.")
            submit_button = form.form_submit_button(label='Submit')
            if submit_button:
                if title == "":
                    st.error("Please enter a job title")
                elif company == "":
                    st.error("Please enter a company name")
                else:
                    with st.spinner('Generating...'):
                        cover_letter = get_cover_letter(title, company, more_info, "resume.pdf")
                        container_three.markdown("---")
                        container_three.markdown("### Cover Letter:")
                        container_three.write(cover_letter)

    elif page == "interview questions generator":
        container_two = st.container()
        container_two.title("A Interview Questions Generating Chatbot")
        container_two.markdown(
            "JobGPT is a chatbot that generates interview questions.\
                This is built just for demo purpose."
        )
        input_text = container_two.text_area(
            "Prompt",
            "Hi, can you please help me generate interview questions?")
        button = container_two.button("Send")

        st.sidebar.markdown("---")
        st.sidebar.markdown("Click on `new chat` to start a new chat. \
                History will be cleared and you'll lose access to current chat."
                           )
        clear_session = st.sidebar.button("New Chat")

        if clear_session:
            st.session_state['interview_questions_conversation'] = interview_questions_conversation
        container_two.markdown("---")

        initial_message = "Hello, how can I help you?"
        message(initial_message)

        if button:
            response = predict_interview_question(
                input_text, st.session_state['interview_questions_conversation'])
            message_writer(input_text, response)

main()