Update app.py

app.py

@@ -27,7 +27,7 @@ article_text = """
 </div>
 """
 
-def query(lora_id, prompt, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", seed=-1, width=1024, height=1024):
+def query(lora_id, prompt, steps=28, cfg_scale=3.5, randomize_seed=True, seed=-1, width=1024, height=1024):
     if prompt == "" or prompt == None:
         return None
 
@@ -48,8 +48,8 @@ def query(lora_id, prompt, steps=28, cfg_scale=3.5, sampler="DPM++ 2M Karras", s
     # print(f'\033[1mGeneration {key}:\033[0m {prompt}')
 
     # If seed is -1, generate a random seed and use it
-    if seed == -1:
-        seed = random.randint(1, 1000000000)
+    if randomize_seed:
+        seed = random.randint(1, 4294967296)
     
     payload = {
         "inputs": prompt,
@@ -108,10 +108,12 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
                         with gr.Row():
                             width = gr.Slider(label="Width", value=1024, minimum=64, maximum=1216, step=8)
                             height = gr.Slider(label="Height", value=1024, minimum=64, maximum=1216, step=8)
-                        steps = gr.Slider(label="Sampling steps", value=28, minimum=1, maximum=100, step=1)
-                        cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
-                        method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
-                        seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
+                        seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=4294967296, step=1)
+                        randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                        with gr.Row():
+                            steps = gr.Slider(label="Sampling steps", value=28, minimum=1, maximum=100, step=1)
+                            cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
+                        # method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"])
 
         with gr.Row():
             text_button = gr.Button("Run", variant='primary', elem_id="gen-button")
@@ -128,6 +130,6 @@ with gr.Blocks(theme='Nymbo/Nymbo_Theme', css=css) as app:
         )
 
         
-        text_button.click(query, inputs=[custom_lora, text_prompt, steps, cfg, method, seed, width, height], outputs=[image_output,seed_output])
+        text_button.click(query, inputs=[custom_lora, text_prompt, steps, cfg, randomize_seed, seed, width, height], outputs=[image_output,seed_output])
 
 app.launch(show_api=False, share=False)