show previous seed used (#5)

- show previous seed used (7fe459f90cba7c089af300c45433c7905df6378a)


Co-authored-by: Krebzonide <[email protected]>

app.py

@@ -1,4 +1,5 @@
 import functools
+import random
 
 import gradio as gr
 import torch
@@ -95,7 +96,8 @@ def generate_fn(
 ):
   try:
     if seed < 0:
-      seed = None
+      seed = random.randint(1,9999999999999999) #16 digits is an arbitrary limit
+    print("seed: ", seed)
 
     max_feedback_imgs = max(0, int(max_feedback_imgs))
     total_images = (len(liked) if liked else 0) + (len(disliked) if disliked else 0)
@@ -141,7 +143,7 @@ def generate_fn(
       if 'out of memory' in str(err):
         generator.reload()
       raise
-    return [(img, f"Image {i+1}") for i, img in enumerate(images)], images
+    return [(img, f"Image {i+1}") for i, img in enumerate(images)], images, seed
   except Exception as err:
     raise gr.Error(str(err))
 
@@ -164,6 +166,8 @@ def remove_img_from_list(event: gr.SelectData, imgs):
     imgs.pop(event.index)
   return imgs, imgs
 
+def duplicate_seed_value(seed): #I don't like the progress bar showing on the previous seed box and this is how I hide it
+    return seed
 
 with gr.Blocks(css=css) as demo:
 
@@ -225,6 +229,9 @@ with gr.Blocks(css=css) as demo:
           dislike_btn = gr.Button(f"👎 Image {i+1}", elem_classes="btn-red")
           dislike_btns.append(dislike_btn)
 
+      prev_seed = gr.Number(-1, label="Previous seed", interactive=False)
+      prev_seed_hid = gr.Number(-1, visible=False)
+
   generate_params = [
       feedback_enabled,
       max_feedback_imgs,
@@ -242,8 +249,9 @@ with gr.Blocks(css=css) as demo:
       batch_size,
       seed,
   ]
-  submit_btn.click(generate_fn, generate_params, [gallery, curr_imgs], queue=True)
-
+  submit_btn.click(generate_fn, generate_params, [gallery, curr_imgs, prev_seed_hid], queue=True)
+  prev_seed_hid.change(duplicate_seed_value, prev_seed_hid, prev_seed, queue=False)
+    
   for i, like_btn in enumerate(like_btns):
     like_btn.click(functools.partial(add_img_from_list, i), [curr_imgs, liked_imgs], [like_gallery, liked_imgs], queue=False)
   for i, dislike_btn in enumerate(dislike_btns):