upload app.py

app.py

@@ -0,0 +1,20 @@
+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
+
+if __name__ == "__main__":
+    check_env()
+    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()