File size: 1,326 Bytes
d4a7025
 
7f2cb09
410e594
d4a7025
f704857
7f2cb09
 
 
 
 
d4a7025
7f2cb09
 
d4a7025
 
 
 
7f2cb09
 
 
17c52e7
7f2cb09
17c52e7
 
 
f79cf2d
7f2cb09
 
 
 
 
17c52e7
7f2cb09
d4a7025
17c52e7
 
d4a7025
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
"""Module doc string"""

from datetime import datetime

import streamlit as st
from .utils import (
    OpenAIFunctions,
    StreamlitFunctions,
    discord_hook,
    logger,
)

datetime_string = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
discord_hook(f"Simple chat bot initiated {datetime_string}")


def main():
    """_summary_"""
    StreamlitFunctions.streamlit_page_config()
    StreamlitFunctions.streamlit_initialize_variables()
    StreamlitFunctions.streamlit_side_bar()
    if st.session_state.start_app:
        logger.info("Application Starting Condition passed")
        if (
            st.session_state.openai_api_key is not None
            and st.session_state.openai_api_key != ""
        ) or st.session_state.provider_select != "OpenAI":
            logger.info("OpenAI key Checking condition passed")
            if OpenAIFunctions.check_openai_api_key():
                logger.info("Inference Started")
                StreamlitFunctions.streamlit_print_messages()
                StreamlitFunctions.streamlit_invoke_model()
            else:
                StreamlitFunctions.reset_history()
        else:
            with st.chat_message("assistant"):
                st.markdown("**'OpenAI API key'** is missing.")


if __name__ == "__main__":
    main()