Fix state scope (it was global!)

app.py

@@ -8,91 +8,92 @@ from functools import partial
 from PIL import Image
 
 SELECT_LABEL = "Select as seed"
-selectors: List[gr.Radio] = []
-seeds = []
-
-block = gr.Blocks(css=".container { max-width: 800px; margin: auto; }")
-
-def infer_seeded_image(prompt, seed):
-    print(f"Prompt: {prompt}, seed: {seed}")
-    return Image.open(f"sample_outputs/seeded_1.png")
-
-def infer_grid(prompt):
-    response = defaultdict(list)
-    for i in range(1, 7):
-        response["images"].append(Image.open(f"sample_outputs/{i}.png"))
-        response["seeds"].append(random.randint(0, 2 ** 32 -1))
-
-    global seeds
-    seeds = response["seeds"]
-    return response["images"]
-
-def infer(prompt):
-    """
-    Outputs:
-    - Grid images (list)
-    - Seeded Image (Image or None)
-    - Grid Box with updated visibility
-    - Seeded Box with updated visibility
-    """
-    grid_images = [None] * 6
-    image_with_seed = None
-    visible = (False, False)
-
-    if seed_index := current_selection() > -1:
-        seed = seeds[seed_index]
-        image_with_seed = infer_seeded_image(prompt, seed)
-        visible = (False, True)
-    else:
-        grid_images = infer_grid(prompt)
-        visible = (True, False)
-
-    boxes = [gr.Box.update(visible=v) for v in visible]
-    return grid_images + [image_with_seed] + boxes
-
-
-def image_block():
-    return gr.Image(
-        interactive=False, show_label=False
-    ).style(
-        # border = (True, True, False, True),
-        rounded = (True, True, False, False),
-    )
 
-selectors_state = [''] * 6
-def did_select(radio: gr.Radio):
-    new_state = list(map(lambda r: SELECT_LABEL if r == radio else '', selectors))
-    return new_state
-
-def update_state(radio: gr.Radio, *state):
-    global selectors_state
-    if list(state) != selectors_state:
-        selectors_state = did_select(radio)
-    return selectors_state
-
-def current_selection():
-    try:
-        return selectors_state.index(SELECT_LABEL)
-    except:
-        return -1
-
-def clear_seed():
-    """Update state of Radio buttons, grid, seeded_box"""
-    global selectors_state
-    selectors_state = [''] * 6
-    return selectors_state + [gr.Box.update(visible=True), gr.Box.update(visible=False)]
-
-def radio_block():
-    radio = gr.Radio(
-        choices=[SELECT_LABEL], interactive=True, show_label=False,
-    ).style(
-        # border = (False, True, True, True),
-        # rounded = (False, False, True, True)
-        container=False
-    )
-    return radio
+with gr.Blocks(css=".container { max-width: 800px; margin: auto; }") as demo:
+    state = gr.Variable({
+        'selectors': [''] * 6,
+        'seeds': []
+    })
+    # selectors_state = [''] * 6
+    # seeds = []
+
+    def infer_seeded_image(prompt, seed):
+        print(f"Prompt: {prompt}, seed: {seed}")
+        return Image.open(f"sample_outputs/seeded_1.png")
+
+    def infer_grid(prompt):
+        response = defaultdict(list)
+        for i in range(1, 7):
+            response["images"].append(Image.open(f"sample_outputs/{i}.png"))
+            response["seeds"].append(random.randint(0, 2 ** 32 -1))
+
+        return response["images"], response["seeds"]
+
+    def infer(prompt, state):
+        """
+        Outputs:
+        - Grid images (list)
+        - Seeded Image (Image or None)
+        - Grid Box with updated visibility
+        - Seeded Box with updated visibility
+        """
+        grid_images = [None] * 6
+        image_with_seed = None
+        visible = (False, False)
+
+        if seed_index := current_selection(state) > -1:
+            seed = state["seeds"][seed_index]
+            image_with_seed = infer_seeded_image(prompt, seed)
+            visible = (False, True)
+        else:
+            grid_images, seeds = infer_grid(prompt)
+            state["seeds"] = seeds
+            visible = (True, False)
+
+        boxes = [gr.Box.update(visible=v) for v in visible]
+        return grid_images + [image_with_seed] + boxes + [state]
+
+
+    def image_block():
+        return gr.Image(
+            interactive=False, show_label=False
+        ).style(
+            # border = (True, True, False, True),
+            rounded = (True, True, False, False),
+        )
+
+    def update_state(selected_index: int, value, state):
+        if value == '':
+            others_value = None
+        else:
+            others_value = ''
+            new_state = [''] * 6
+            new_state[selected_index] = SELECT_LABEL
+            state["selectors"] = new_state
+        others = gr.Radio.update(value=others_value)
+        return [others] * 5 + [state]
+
+    def current_selection(state):
+        try:
+            return state["selectors"].index(SELECT_LABEL)
+        except:
+            return -1
+
+    def clear_seed(state):
+        """Update state of Radio buttons, grid, seeded_box"""
+        state["selectors"] = [''] * 6
+        return state["selectors"] + [gr.Box.update(visible=True), gr.Box.update(visible=False)] + [state]
+
+    def radio_block():
+        radio = gr.Radio(
+            choices=[SELECT_LABEL], interactive=True, show_label=False,
+        ).style(
+            # border = (False, True, True, True),
+            # rounded = (False, False, True, True)
+            container=False
+        )
+        return radio
 
-with block:
     gr.Markdown("<h1><center>Latent Diffusion Demo</center></h1>")
     with gr.Group():
         with gr.Box():
@@ -134,22 +135,39 @@ with block:
                     select6 = radio_block()
 
         images = [image1, image2, image3, image4, image5, image6]
-        selectors += [select1, select2, select3, select4, select5, select6]
+        selectors = [select1, select2, select3, select4, select5, select6]
 
-        for radio in selectors:
-            radio.change(fn=partial(update_state, radio), inputs=selectors, outputs=selectors)
+        for i, radio in enumerate(selectors):
+            others = list(filter(lambda s: s != radio, selectors))
+            radio.change(
+                partial(update_state, i),
+                inputs=[radio, state],
+                outputs=others + [state]
+            )
 
     with (seeded_box := gr.Box()):
         seeded_image = image_block()
         clear_seed_button = gr.Button("Clear Seed")
     seeded_box.visible = False
-    clear_seed_button.click(clear_seed, inputs=[], outputs=selectors + [grid, seeded_box])
+    clear_seed_button.click(
+        clear_seed,
+        inputs=[state],
+        outputs=selectors + [grid, seeded_box] + [state]
+    )
 
     all_images = images + [seeded_image]
     boxes = [grid, seeded_box]
-    infer_outputs = all_images + boxes
+    infer_outputs = all_images + boxes + [state]
 
-    text.submit(infer, inputs=text, outputs=infer_outputs)
-    btn.click(infer, inputs=text, outputs=infer_outputs)
+    text.submit(
+        infer,
+        inputs=[text, state],
+        outputs=infer_outputs
+    )
+    btn.click(
+        infer,
+        inputs=[text, state],
+        outputs=infer_outputs
+    )
 
-block.launch(enable_queue=False)
+demo.launch(enable_queue=False)