Update app.py

app.py

@@ -153,11 +153,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                         value=3.5
                     )
                 
+                def get_random_seed():
+                    return torch.randint(0, 1000000, (1,)).item()
+                
                 seed = gr.Number(
-                    label="Seed (for reproducibility)",
-                    value=3413,
+                    label="Seed (random by default, set for reproducibility)",
+                    value=get_random_seed(),
                     precision=0
                 )
+                
+                randomize_seed = gr.Button("🎲 Randomize Seed", elem_classes=["generate-btn"])
             
             generate_btn = gr.Button(
                 "✨ Generate Image",
@@ -229,11 +234,26 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             return generated_image, load_gallery()
     
     # Connect the generation button to both the image output and gallery update
+    def update_seed():
+        return get_random_seed()
+
     generate_btn.click(
         process_and_save_image,
         inputs=[height, width, steps, scales, prompt, seed],
         outputs=[output, gallery]
     )
+    
+    # Add randomize seed button functionality
+    randomize_seed.click(
+        update_seed,
+        outputs=[seed]
+    )
+    
+    # Also randomize seed after each generation
+    generate_btn.click(
+        update_seed,
+        outputs=[seed]
+    )
 
 if __name__ == "__main__":
     demo.launch()