import sys
sys.path.append("./src")
import streamlit as st
from app_pages import button_interface, step_by_step_generation, one_click_generation, homepage
from app_pages.locale import _
from utils.hash import check_env, check_embedding

if __name__ == "__main__":
    check_env()
    # check_embedding() 
    backend = button_interface.Backend()
    # backend = None
    st.set_page_config(layout="wide")
    if "language" not in st.session_state:
        st.session_state["language"] = "zh"
    def fn1():
        one_click_generation.one_click_generation(backend)
    def fn2():
        step_by_step_generation.step_by_step_generation(backend)
    pg = st.navigation([
        st.Page(homepage.home_page, title=_("🏠️ Homepage")),
        st.Page(fn1, title=_("💧 One-click Generation")),
        st.Page(fn2, title=_("💦 Step-by-step Generation")),
    ])
    pg.run()