Update app.py

app.py

@@ -167,19 +167,19 @@ example_images = [Image.open(i) for i in example_images]
 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() != '']
+with open(os.path.join(example_root, 'prompt_girl.txt')) as f:
+    prompts_girl = [l.strip() for l in f.readlines() if l.strip() != '']
 
-# 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_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}
 
 prompt_background = lambda: random.choice(prompts_background)
-# prompt_girl = lambda: random.choice(prompts_girl)
-# prompt_boy = lambda: random.choice(prompts_boy)
+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 - 1), replace=False).tolist()
 
 
@@ -243,11 +243,14 @@ def select_palette(state, button, idx):
 
 
 def random_palette(state):
-    girlboy = random.choice(['A girl smiling at viewer', 'A boy smiling at viewer'])
-    girlboy_name = '👧 Girl' if 'girl' in girlboy else '👦 Boy'
+    # girlboy = random.choice(['A girl smiling at viewer', 'A boy smiling at viewer'])
+    girlboy = random.choice(['girl', 'boy'])
+    girlboy_name = '👧 Girl' if girlboy == 'girl' else '👦 Boy'
+    girlboy_prompt = prompt_girl() if girlboy == 'girl' else prompt_boy()
+
     prompt_props_ = prompt_props()
     state.prompt_names = ['🌄 Background'] + [girlboy_name] + prompt_props_
-    state.prompts = ['']  + [girlboy] + [prompts_props[k] for k in prompt_props_]
+    state.prompts = ['']  + [girlboy_prompt] + [prompts_props[k] for k in prompt_props_]
     state.neg_prompts = [
         opt.default_negative_prompt
         + (', humans, humans, humans' if i == 0 else '')