app.py

import streamlit as st
import pandas as pd
import numpy as np

import os, random, time

from utils import footer
from page import generate, reduce


st.set_page_config(
        page_title="AI Architecture",
)

if( hasattr(st.session_state, 'page') == False):
    st.session_state.page = 0

if( hasattr(st.session_state, 'results') == False):
    st.session_state.results = []

p1 = st.empty()
p2 = st.empty()
p3 = st.empty()


st.session_state.stop = False
st.session_state.progress = 0
st.session_state.images = []
st.session_state.regenerate = False

footer()

if(st.session_state.page == 0):
    p2.empty()
    p3.empty()
    with p1.container():
        generate.app()
        

if(st.session_state.page == 1):
    p1.empty()
    p3.empty()
    with p2.container():
        reduce.app()

if(st.session_state.page == 2):
    p1.empty()
    p2.empty()
    with p3.container():
        st.write("This 333")
        startButton = st.button("S3")
        if startButton:
            st.session_state.page = 0