import collections
from numpy.core.defchararray import lower
import streamlit as st
import numpy as np
import pandas as pd
import streamlit as st
import pandas as pd
import numpy as np
import os, random, time
from utils import footer, generate, drawGrid, generate2
from PIL import Image

mode = "ai"
#mode = "dummy"

def app():
    global _prompt

    st.title('AI-Generated Architecture')

    #st.subheader('(beta v1.1)')

    #st.text('This is a working beta version with bugs. Known issues are:\n-Some images will grey out when you change the input parameters')

    st.subheader('Describe a building, interior, or other architecture you would like to see. You can change the prompt and input parameters on the fly.')

    #Modern architecture museum with black brick and large windows.
    print("Prompt: "+st.session_state.prompt)
    prompt = st.text_input(label="",value=st.session_state.prompt)

    st.text("")

    #with st.expander("Having trouble thinking of something? Click here to view examples."):
    #    st.write("""
    #        • Modern architecture museum with black brick and large windows.\n
    #        • A prosaic, simple architecture.\n
    #        • An urban, post-modern architecture with concrete and steel.\n
    #        • A sleek urban interior design.
    #    """)

    st.text("")

    crazy = st.slider('Temperature. This controls how "crazy" generated images are, where 0 is the least crazy.', 0.0, 1.0, 0.75)
    k = st.slider('Top K. The higher the value, the higher quality the results tend to be at the cost of extra processing time.', 1, 10, 5)
    #k = k*400

    if( 'results' not in st.session_state ):
        st.session_state.results = []


    c1,c2 = st.columns(2)

    with c1:
        holder = st.empty()
    with c2:
        holder2 = st.empty()
    
    startButton = holder.button("Start")

    already = []

    if startButton or hasattr(st.session_state, 'load_state') or st.session_state.prompt is not None:

        with st.spinner("Generating..."):

            holder.empty()

            st.session_state.load_state = True


            
            placeholder = st.empty()
            second = st.empty()

            nextButton = False
            randomButton = False
            f = True

            ii = 0
            while len(st.session_state.results) <= 64:

                ii += 1

                if(f and len(st.session_state.results) > 0):
                    f = False
                    randomButton = holder2.button("Randomize Prompt")  
                    nextButton = holder.button("Finish Generating Images")    

                with second.container():
                    drawGrid()

                with placeholder.container():

                    st.session_state.bar = placeholder.progress(0)
                    
                    if(len(st.session_state.results) > 0 and nextButton):
                        st.session_state.page = 1
                        break

                if(randomButton):
                    randomButton = False
                    _1 = ["A modern ","A post-modern ","An ultramodern ", "A classical ", "A parametric ", "A contemporary ", "A minimalist "]
                    _2 = ["museum architecture","home architecture","landscape architecture","interior design"]
                    _3 = [""," in the style of I.M. Pei"," in the style of Frank Gehry"," in the style of John Lautner"," in the style of Frank Lloyd Wright"]
                    _4 = [" photograph",", watercolor painting",", oil painting", ", digital art"]

                    prompt = str(random.choice(_1)+random.choice(_2)+random.choice(_3)+random.choice(_4))
                    st.session_state.prompt = prompt

                generate(prompt,crazy,k)

            st.session_state.bar = st.container()