hackathon_specme/app.py at main · carolzy/hackathon_specme · GitHub

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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
# app.py
import streamlit as st
from pages import generate_api_options, landing_page, generate_backend, generate_design, one_page

# Define a function to render the current page
###### page config ###############################################################################################################################
st.set_page_config(
        page_title="specme",
        page_icon="static/yobo_icon.png",
        layout="wide",
        initial_sidebar_state="collapsed"
)

def render_page(page):
    pages = {
        "Landing Page": landing_page.app,
        "Choose Tech Stack Page": generate_api_options.app,
        "Generate Design": generate_design.app,
        "Generate Backend": generate_backend.app,
    }
    page_function = pages[page]
    page_function()

PAGES = {
    "Landing Page": one_page.app,
    "Choose Tech Stack Page": generate_api_options.app,
    "Generate Design": generate_design.app,
    "Generate Backend": generate_backend.app,
}

#Initialize the session state for the current page if it's not already set
if 'current_page' not in st.session_state:
    st.session_state['current_page'] = "Landing Page"

# Page loop
while True:
    page_show_func = PAGES[st.session_state['current_page']]
    next_page = page_show_func()

    if next_page is None:
        # If no navigation is required, break the loop
        break
    else:
        # Update the page_name based on the navigation
        page_name = next_page


# # Render the current page based on the session state
# render_page(st.session_state['current_page'])