File size: 877 Bytes
fe93e75
 
 
479f67b
 
4117eaa
fe93e75
 
 
7f45547
fe93e75
479f67b
fe93e75
479f67b
 
fe93e75
 
 
 
 
479f67b
 
 
fe93e75
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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()