Update page/generate.py

page/generate.py

@@ -14,11 +14,12 @@ mode = "ai"
 #mode = "dummy"
 
 def app():
-    
-    
+
 
     st.title('AI-Generated Architecture')
 
+    st.subheader('(beta v1.0)')
+
     st.subheader('Describe a building, interior, or other architecture you would like to see.')
 
     #Modern architecture museum with black brick and large windows.
@@ -48,13 +49,9 @@ def app():
 
     already = []
 
-    print("-0-") 
-
     if startButton or hasattr(st.session_state, 'load_state'):
 
-        with st.spinner("Generating..."): 
-
-            print("-1-")    
+        with st.spinner("Generating..."):   
 
             holder.empty()
 
@@ -64,12 +61,13 @@ def app():
             placeholder = st.empty()
             second = st.empty()
 
-            with second.container():
-                drawGrid()
+            #with second.container():
+            #    drawGrid()
 
-            while len(st.session_state.results) <= 15:
+            while len(st.session_state.results) <= 64:
 
-                print("Length "+str(len(st.session_state.results)))
+                with second.container():
+                    drawGrid()
 
                 with placeholder.container():
 
@@ -80,18 +78,9 @@ def app():
                         st.session_state.page = 1
                         break
 
-                    generate(prompt,crazy,k)
-
-                with second.container():
-                    drawGrid()
-
-
                     
+                generate( prompt ,crazy,k)
 
-                #placeholder.empty()
                 
-                #st.session_state.bar = placeholder.progress(0)
-                #drawGrid(placeholder)
-
 
-    
+            st.session_state.bar = st.container()