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