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="Mixtral Search Engine",
    page_icon="📚",
)


init_state(st.session_state, config)

st.write("# Mixtral Search Engine")

# 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)