updating app

app.py

@@ -3,37 +3,37 @@ import random, os
 from PIL import Image
 
 def open_ims(choice):
-    print(choice)
-    imnames = os.listdir('images/'+'Photo_portrait_of_a_'+choice.replace(' ',' ')+'/Seed_46267/')
-    images = [(Image.open(name)) for name in imnames]
-    return images
+    dirname='images/'+'Photo_portrait_of_a_'+choice.replace(' ',' ')+'/Seed_'+ str(seed_choice)+'/'
+    imnames= os.listdir(dirname)
+    images = [(Image.open(dirname+name)) for name in imnames]
+    return images[:9]
 
 def random_image():
     cat = random.choice(categories)
-    print(cat)
-    imnames = os.listdir('images/'+'Photo_portrait_of_a_'+choice.replace(' ',' ')+'/Seed_46267/')
-    print(imnames)
-    ims = [(Image.open(name)) for name in imnames]
-    print(ims)
-    return ims
+    dirname = 'images/'+'Photo_portrait_of_a_'+cat.replace(' ',' ')+'/Seed_'+str(seed_choice)+'/'
+    imnames = os.listdir(dirname)
+    images = [(Image.open(dirname+name)) for name in imnames]
+    return images[:9]
 
 categories=os.listdir('images/')
 categories = [c.replace('Photo_portrait_of_a_','').replace('_',' ') for c in categories]
-print(categories)
+seeds = [46267, 48040, 51237, 54325, 60884, 64830, 67031, 72935, 92118, 93109]
 
 with gr.Blocks() as demo:
     gr.Markdown("# Stable Diffusion Explorer")
     gr.Markdown("## TODO")
-
+    seed_choice = gr.State(0)
+    seed_choice = random.choice(seeds)
+    print(seed_choice)
     with gr.Row():
         with gr.Column():
-              choice = gr.Dropdown(categories, label = "Choose a first category", interactive=True)
-              images = gr.Gallery(label="Images"),
+            choice1 = gr.Dropdown(categories, label = "Choose a first category", interactive=True)
+            images1 = gr.Gallery(label="Images").style(grid=[3], height="auto")
         with gr.Column():
-              gr.Textbox(label="Text prompt"),
-              gr.Image(optional=True, label="Image prompt", type='filepath'),
-
-    demo.load(random_image, None, images)          
-    choice.change(open_ims, [prompt], [images])
+            choice2 = gr.Dropdown(categories, label = "Choose a second category", interactive=True)
+            images2 = gr.Gallery(label="Images").style(grid=[3], height="auto")
 
+    #demo.load(random_image, None, [images])
+    choice1.change(open_ims, [choice1], [images1])
+    choice2.change(open_ims, [choice2], [images2])
 demo.launch()