Update page/generate.py

page/generate.py

@@ -47,7 +47,14 @@ def app():
     if( 'results' not in st.session_state ):
         st.session_state.results = []
 
-    holder = st.empty()
+
+    c1,c2 = st.columns(2)
+
+    with c1:
+        holder = st.empty()
+    with c2:
+        holder2 = st.empty()
+    
     startButton = holder.button("Start")
 
     already = []
@@ -60,20 +67,23 @@ def app():
 
             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) <= 4*64:
+            while len(st.session_state.results) <= 128:
 
                 ii += 1
 
                 if(f and len(st.session_state.results) > 0):
                     f = False
+                    randomButton = holder2.button("randomize prompt")  
                     nextButton = holder.button("finished generating images")    
 
                 with second.container():
@@ -87,24 +97,16 @@ def app():
                         st.session_state.page = 1
                         break
 
-                    
-                
-                #if(ii % 2 == 1):
-                generate2(prompt,crazy,k)
-                #generate(prompt ,crazy,k)
-
-
-                if ii % 6 == 0:
+                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","structure architecture"]
                     _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))
-                    print("swap")
+                    st.session_state.prompt = prompt
 
-                    crazy = random.random()*2
-                    k = random.randrange(1,2500)
-                
+                generate2(prompt,crazy,k)
 
             st.session_state.bar = st.container()