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, intro


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

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

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

if( hasattr(st.session_state, 'prompt') == False):
    st.session_state.prompt = None

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


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



footer()

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

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