Fake seeds

app.py

@@ -7,12 +7,14 @@ from collections import defaultdict
 from functools import partial
 from PIL import Image
 
-block = gr.Blocks(css=".container { max-width: 800px; margin: auto; }")
-
 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):
@@ -21,8 +23,8 @@ def infer_grid(prompt):
         response["images"].append(Image.open(f"sample_outputs/{i}.png"))
         response["seeds"].append(random.randint(0, 2 ** 32 -1))
 
-    # TODO: only run when a selection exists
-    # TODO: get seed to reuse
+    global seeds
+    seeds = response["seeds"]
     return response["images"]
 
 def infer(prompt):
@@ -37,8 +39,9 @@ def infer(prompt):
     image_with_seed = None
     visible = (False, False)
 
-    if current_selection() > -1:
-        image_with_seed = infer_seeded_image(prompt, 7667)
+    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)