app.py

import sys
import os
sys.path.append("./src")
import streamlit as st
from pages import button_interface, step_by_step_generation, one_click_generation
from utils.hash import check_env, check_embedding

if __name__ == "__main__":
    check_env()
    check_embedding() 
    backend = button_interface.Backend()
    st.set_page_config(layout="wide")
    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(fn1, title="💧 One-click Generation"),
        st.Page(fn2, title="💦 Step-by-step Generation"),
    ])
    pg.run()