Update app.py

app.py

@@ -153,8 +153,35 @@ opt.prep_time = -3
 
 
 ### Shared memory hack for ZeroGPU
-opt.address = ('localhost', 6000)
-opt.authkey = b'secret password'
+# opt.address = ('localhost', 6000)
+# opt.authkey = b'secret password'
+
+
+### Load examples
+
+root = pathlib.Path(__file__).parent
+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.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_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_props = lambda: np.random.choice(list(prompts_props.keys()), size=(opt.max_palettes - 1), replace=False).tolist()
+
 
 
 ### Event handlers
@@ -215,6 +242,43 @@ def select_palette(state, button, idx):
     return updates
 
 
+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'
+    prompt_props_ = prompt_props()
+    state.prompt_names = ['🌄 Background'] + [girlboy_name] + prompt_props_
+    state.prompts = ['']  + girlboy + [prompts_props[k] for k in prompt_props_]
+    state.neg_prompts = [
+        opt.default_negative_prompt
+        + (', humans, humans, humans' if i == 0 else '')
+        for i in range(opt.max_palettes + 1)
+    ]
+    state.prompt_strengths = [opt.default_prompt_strength for _ in range(opt.max_palettes)]
+    state.mask_strengths = [opt.default_mask_strength for _ in range(opt.max_palettes)]
+    state.mask_stds = [opt.default_mask_std for _ in range(opt.max_palettes)]
+
+    updates = [state] + [
+        gr.update(value=state.prompt_names[state.current_palette]),
+        gr.update(value=state.prompts[state.current_palette]),
+        gr.update(value=state.neg_prompts[state.current_palette]),
+        (
+            gr.update(value=state.mask_strengths[state.current_palette - 1], interactive=True) if state.current_palette > 0 else
+            gr.update(value=opt.default_mask_strength, interactive=False)
+        ),
+        (
+            gr.update(value=state.prompt_strengths[state.current_palette - 1], interactive=True) if state.current_palette > 0 else
+            gr.update(value=opt.default_prompt_strength, interactive=False)
+        ),
+        (
+            gr.update(value=state.mask_stds[state.current_palette - 1], interactive=True) if state.current_palette > 0 else
+            gr.update(value=opt.default_mask_std, interactive=False)
+        ),
+    ] + [
+        gr.update(value=v) for v in state.prompt_names[1:]
+    ]
+    return updates
+
+
 def change_prompt_strength(state, strength):
     if state.current_palette == 0:
         return state
@@ -497,31 +561,6 @@ def draw(state, drawpad):
     # conn.send(data)
     # conn.close()
 
-### Load examples
-
-root = pathlib.Path(__file__).parent
-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.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_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_props = lambda: np.random.choice(list(prompts_props.keys()), size=(opt.max_palettes - 1), replace=False).tolist()
-
 
 ### Main application
 
@@ -656,19 +695,18 @@ 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()
         state.prompt_names = [
             '🌄 Background',
             '👧 Girl',
-            # '🐶 Dog',
-            # '💐 Garden',
-        ] + prompt_props_ + ['🎨 New Palette' for _ in range(opt.max_palettes - 3)]
+            '🐶 Dog',
+            '💐 Garden',
+        ] + ['🎨 New Palette' for _ in range(opt.max_palettes - 3)]
         state.prompts = [
             '',
             '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)]
+            'Doggy body part',
+            'Flower garden',
+        ] + ['' for _ in range(opt.max_palettes - 3)]
         state.neg_prompts = [
             opt.default_negative_prompt
             + (', humans, humans, humans' if i == 0 else '')
@@ -760,6 +798,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, head=head) as demo:
                     visible=(state.value.active_palettes < opt.max_palettes),
                 )
 
+                iface.btn_random_palette = gr.Button(
+                    value='Random Semantic Palette',
+                    variant='primary',
+                )
+
             with gr.Accordion(label='Import/Export Semantic Palette', open=True):
                 iface.tbox_state_import = gr.Textbox(label='Put Palette JSON Here To Import')
                 iface.json_state_export = gr.JSON(label='Exported Palette')
@@ -1030,6 +1073,21 @@ Deadline animation originally by <a href="https://codepen.io/jtrancozo/pen/mEoEV
         api_name='create_new',
     )
 
+    iface.btn_random_palette.click(
+        fn=random_palette,
+        inputs=state,
+        output=[
+            state,
+            iface.tbox_name,
+            iface.tbox_prompt,
+            iface.tbox_neg_prompt,
+            iface.slider_alpha,
+            iface.slider_strength,
+            iface.slider_std,
+        ] + iface.btn_semantics[1:],
+        api_name='random_palette',
+    )
+
     run_event = iface.btn_generate.click(
         fn=hide_element,
         inputs=None,