Update app.py

app.py

@@ -499,15 +499,8 @@ def draw(state, drawpad):
 
 ### Load examples
 
-
-root = pathlib.Path(__file__).parent
-print(root)
-example_root = os.path.join(root, 'examples')
-example_images = glob.glob(os.path.join(example_root, '*.png'))
-example_images = [Image.open(i) for i in example_images]
-
-# with open(os.path.join(example_root, 'prompt_background_advanced.txt')) as f:
-#     prompts_background = [l.strip() for l in f.readlines() if l.strip() != '']
+with open(os.path.join(example_root, 'prompt_background.txt')) as f:
+    prompts_background = [l.strip() for l in f.readlines() if l.strip() != '']
 
 # with open(os.path.join(example_root, 'prompt_girl.txt')) as f:
 #     prompts_girl = [l.strip() for l in f.readlines() if l.strip() != '']
@@ -515,14 +508,14 @@ example_images = [Image.open(i) for i in example_images]
 # with open(os.path.join(example_root, 'prompt_boy.txt')) as f:
 #     prompts_boy = [l.strip() for l in f.readlines() if l.strip() != '']
 
-# with open(os.path.join(example_root, 'prompt_props.txt')) as f:
-#     prompts_props = [l.strip() for l in f.readlines() if l.strip() != '']
-#     prompts_props = {l.split(',')[0].strip(): ','.join(l.split(',')[1:]).strip() for l in prompts_props}
+with open(os.path.join(example_root, 'prompt_props.txt')) as f:
+    prompts_props = [l.strip() for l in f.readlines() if l.strip() != '']
+    prompts_props = {l.split(',')[0].strip(): ','.join(l.split(',')[1:]).strip() for l in prompts_props}
 
-# prompt_background = lambda: random.choice(prompts_background)
+prompt_background = lambda: random.choice(prompts_background)
 # prompt_girl = lambda: random.choice(prompts_girl)
 # prompt_boy = lambda: random.choice(prompts_boy)
-# prompt_props = lambda: np.random.choice(list(prompts_props.keys()), size=(opt.max_palettes - 2), replace=False).tolist()
+prompt_props = lambda: np.random.choice(list(prompts_props.keys()), size=(opt.max_palettes - 1), replace=False).tolist()
 
 
 ### Main application
@@ -658,19 +651,19 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, head=head) as demo:
         state.active_palettes = 5
 
         # Front-end initialized to the default values.
-        # prompt_props_ = prompt_props()
+        prompt_props_ = prompt_props()
         state.prompt_names = [
             '🌄 Background',
             '👧 Girl',
-            '🐶 Dog',
-            '💐 Garden',
-        ] + ['🎨 New Palette' for _ in range(opt.max_palettes - 3)]
+            # '🐶 Dog',
+            # '💐 Garden',
+        ] + prompt_props_ + ['🎨 New Palette' for _ in range(opt.max_palettes - 3)]
         state.prompts = [
             '',
-           'A girl smiling at viewer',
-            'Doggy body part',
-            'Flower garden',
-        ] + ['' for _ in range(opt.max_palettes - 3)]
+            'A girl smiling at viewer',
+            # 'Doggy body part',
+            # 'Flower garden',
+        ] + [prompts_props[k] for k in prompt_props_] + ['' for _ in range(opt.max_palettes - 3)]
         state.neg_prompts = [
             opt.default_negative_prompt
             + (', humans, humans, humans' if i == 0 else '')