Update app.py

app.py

@@ -60,29 +60,26 @@ def gen(seed):
     return imgArr
 
 
-# Function to generate a random seed and then generate images
-def surprise_me():
-    import random
-    seed = random.randint(0, 1000)
+# New function to handle text input or surprise me functionality
+def handle_input(seed=None):
+    if seed is None or seed == "":
+        seed = random.randint(0, 1000)
     return gen(seed)
 
 
 iface = gr.Interface(
-    fn=gen, 
+    fn=handle_input,
     inputs=[
-        gr.Text(label="Seed", default="500"),  # Allow manual seed input
-        gr.Button("Surprise Me")  # Add a "Surprise Me" button
+        gr.Text(label="Seed"),  # Text input for seed
     ],
     outputs=gr.Gallery(label="Generated Skins"),
     title = "Minecraft Skin Generator <style>img{image-rendering: pixelated;}</style>", #<-- EWW GROSS IK IK IM SORRY, BUT THAT'S THE ONLY WAY I FOUND IT TO WORK
     debug = True,
 )
-# Define what happens when the "Surprise Me" button is clicked
-iface.add_component(
-    component=gr.Button("Surprise Me"),
-    fn=surprise_me,
-    inputs=[],
-    outputs=[gr.Gallery(label="Generated Skins")]
-)
+# Adding a 'Surprise Me' button that does not require input and calls the same function without a seed
+iface.add_component(gr.Button("Surprise Me"), 
+                    fn=lambda: handle_input(), 
+                    inputs=[], 
+                    outputs=iface.outputs)
 
-iface.launch(width=64,height=64,enable_queue=True)
+iface.launch()