import streamlit as st
from streamlit_extras import buy_me_a_coffee
from streamlit_extras.add_vertical_space import add_vertical_space
from streamlit_extras.badges import badge
from streamlit_lottie import st_lottie
import openai
from utils import load_lottie_url
from api_key import load_api_key, reset_key_button
from internet_search import *
from assistant import *
from gpt_api import find_top_similar_results
from auth import authenticate_user, logout_button
from database import delete_user_button

st.set_page_config(page_title='Quest🔍')
st.title("Quest🔍")
st.markdown('Tired of sifting through search results to find the \
    information you need? The Assistant can take care of it for you! \
        This open source AI-powered personal assistant can access the internet, \
            providing both quick and accurate answers to your questions.')

# Create Sidebar
with st.sidebar:
    lottie_image1 = load_lottie_url('https://assets10.lottiefiles.com/packages/lf20_ofa3xwo7.json')
    st_lottie(lottie_image1)

#authenticate_user()

openai.api_key = load_api_key()

if 'settings' not in st.session_state:
    st.session_state['settings'] = {}

# App layout
tab1, tab2, tab3, tab4 = st.tabs(["Have a conversation", "Internet search", "Create your Assistant", "Settings"])

# Have a conversation tab
with tab1:
    response = st.container()
    chat = st.container()
    
# Internet search tab
with tab2:
    st.markdown("<span style='font-size:2em'>\
        Tell the Assistant what to research about.</span>", unsafe_allow_html=True)
    st.markdown("This tab allows you to give information from across the internet to the Assistant AI. \
        Once you've told it all the topics to search for, you can have a conversation with it in the \
            'Have a conversation' tab.")
    with st.spinner("Getting search history..."):
        google_history = get_user_search_history()
    unique_searches = google_history['query'].unique().tolist()
    unique_searches.insert(0,'')
    initial_search = st.selectbox('Search history', unique_searches, index=0)
    search = st.container()

with tab3:
    st.write("<span style='font-size:2em'>Comming soon...</span>", unsafe_allow_html=True)
    st.write("In this page you will be able to create custom Assistant archetypes.")

with tab4:
    #logout_button()
    reset_key_button()
    #delete_history_button()
    #delete_user_button()

# Google search section
with search:
    with st.form('Google'):
        user_query_text = st.text_input(label='Google search',value=initial_search, help="This tab \
            allows you to give information from across the internet to the Assistant AI. Once you've \
                told it all the topics to search for, you can have a conversation with it in the \
                    'Have a conversation' tab.")
        google_submitted = st.form_submit_button("Submit")

        # If the user pressed submit to make a new search or selected an existing one from history
        if (google_submitted and user_query_text != '') or initial_search != '':
            google_findings, links = make_new_internet_search(user_query_text)
            
            display_search_results(user_query_text, google_findings, links)

# Section where user inputs directly to GPT
with chat:
    with st.form('Chat'):
        user_chat_text = st.text_area(label="Ask the Assistant")
        col1, col2 = st.columns(2)
        chat_submitted = col1.form_submit_button("Submit")
        settings = assistant_settings(chat_submitted, col2)
    add_searches(settings)


# User input is used here to process and display GPT's response
with response:
    if 'archetype' not in settings:
        archetypes, default_setting_index = load_assistant_settings()
        default_setting = list(archetypes.keys())[default_setting_index]
        settings['archetype'] = archetypes[default_setting]
    starting_conversation = settings['archetype']['starting_conversation']
    load_conversation(starting_conversation)
    display_chat_history(starting_conversation)
    if chat_submitted:
        submit_user_message(settings, user_chat_text, chat_submitted)

add_vertical_space(4)

col1, col2, col3 = st.columns(3)
with col1:
    buy_me_a_coffee.button('farrael004', floating=False)
with col2:
    st.markdown("By [Rafael Moraes](https://github.com/farrael004)")
    badge(type="github", name="farrael004/Quest")
with col3:
    st.container()