import yaml
import streamlit as st
from components.sidebar import sidebar
from components.chat_box import chat_box
from components.chat_loop import chat_loop
from components.init_state import init_state
from components.prompt_engineering_dashboard import prompt_engineering_dashboard



with open("config.yaml", "r") as file:
    config = yaml.safe_load(file)

st.set_page_config(
    page_title="NCTC OSINT AGENT",
    page_icon="📚",
)


init_state(st.session_state, config)
html_title = '''
<style>
.stTitle {
  color: #00008B;  /* Deep blue color */
  font-size: 36px;  /* Adjust font size as desired */
  font-weight: bold;  /* Add boldness (optional) */
  /* Add other font styling here (optional) */
}
</style>
<h1 class="stTitle">NCTC OSINT AGENT</h1>
'''

st.write(html_title, unsafe_allow_html=True)

# st.write("# NCTC OSINT AGENT ")

# Prompt Engineering Dashboard is working but not for production, works great for testing.
prompt_engineering_dashboard(st.session_state, config)


sidebar(st.session_state, config)

chat_box(st.session_state, config)

chat_loop(st.session_state, config)