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

# from components.show_source import show_source
# from components.stream_handler import stream_handler
# from components.generate_chat_stream import generate_chat_stream


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

st.set_page_config(
    page_title="Mixtral Playground",
    page_icon="📚",
)


init_state(st.session_state, config)

st.write("# Mixtral Playground")

prompt_engineering_dashboard(st.session_state, config)

sidebar(st.session_state, config)

chat_box(st.session_state, config)

chat_loop(st.session_state, config)