main.py

import streamlit as st
from utils.config import load_config
from deepchecks_llm_client.client import dc_client
from utils.general_utils import initialize_app, initialize_deepchecks_client, initialize_session_state
from menu.deepy import create_ask_deepy_bot
from menu.settings import create_settings
from utils.api_call import fetch_application_names_with_versions


# Render the sidebar on the UI and add the styling to various components
initialize_app()

# Load the configuration
config = load_config()

initialize_session_state()

deepchecks_llm_app_name = config['DEEPCHECKS_LLM_APP_NAME']
deepchecks_llm_version_name = config['DEEPCHECKS_LLM_APP_VERSION_NAME']
response = fetch_application_names_with_versions(config)

if response['status_code'] != 200:
    st.error({'status_code': response['status_code'], 'text': response['text'], 'solution': 'Make sure that your API keys are correct.'})

elif (deepchecks_llm_app_name not in list(response['application_details'].keys()) or deepchecks_llm_version_name not in response['application_details'][deepchecks_llm_app_name]) and st.session_state.current_page != 'Settings':
    st.title('🦜🔗 Deepy Bot')
    st.error('Your Deepchecks LLM app name and version names are not correct. Please update it from the Settings section.')
else:
    if dc_client.api is None and st.session_state.current_page != 'Settings':
        initialize_deepchecks_client(config)

    if st.session_state.current_page == 'Settings':
        st.session_state.llm_response = ""
        create_settings(config, response['application_details'])
    else:
        create_ask_deepy_bot(config['GPT_MODEL'])