Update app.py

app.py

@@ -47,41 +47,40 @@ init()
 
 import numpy
 
-# Function to generate images based on a seed
-def gen(seed):
-    numpy.random.seed(int(seed))
+def gen(seed=None):
+    if seed is None or seed == "Surprise Me":
+        seed = random.randint(0, 1000)
+    else:
+        seed = int(seed)
+    np.random.seed(seed)
     init()  # Initialize and generate samples
     crop()  # Crop the generated images
     imgArr = []
     for i in range(4):
-        img = Image.open(str(i)+".png")
+        img = Image.open(f"{i}.png").convert('RGB')
         img = img.resize((64, 64), Image.NEAREST)
         imgArr.append(img)
     return imgArr
 
-
-# Function to handle input, deciding based on user choice
-def handle_input(seed, action):
-    if action == "Surprise Me":
-        seed = random.randint(0, 1000)
-    return gen(seed)
+    
+# Function to handle the input from the interface
+def handle_input(seed_text, surprise_me):
+    if surprise_me:
+        return gen("Surprise Me")
+    else:
+        return gen(seed_text)
 
 
 
 iface = gr.Interface(
     fn=handle_input,
     inputs=[
-        gr.Text(label="Seed", placeholder="Enter a seed or use 'Surprise Me'"),  # Text input for seed
-        gr.Radio(choices=["Generate", "Surprise Me"], label="Action", default="Generate"),  # Radio buttons to select action
+        gr.Text(label="Seed", placeholder="Enter a seed"),  # Text input for seed
+        gr.Checkbox(label="Surprise Me", value=False),  # Checkbox for 'Surprise Me' functionality
     ],
     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,
 )
-# 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()