update draw codes

app.py

@@ -1,5 +1,35 @@
 import gradio as gr
+import random
+from PIL import Image
 
-demo = gr.Interface(gr.Image(type="pil"))
+# Define the initial options
+options = ["pink"] * 15 + ["blue"] * 10
 
+
+def spin_roulette():
+    # Randomly select an option
+    selected_option = random.choice(options)
+
+    # Remove the selected option from the list
+    options.remove(selected_option)
+
+    return selected_option
+
+
+def draw():
+    for _ in range(25):
+        if options:
+            option = spin_roulette()
+            if option == "pink":
+                pink = Image.open("images/pink.png")
+                return pink
+            else:
+                blue = Image.open("images/blue.png")
+                return blue
+
+
+with gr.Blocks() as demo:
+    result = draw()
+    draw_btn = gr.Button("Draw")
+    draw_btn.click(fn=draw, inputs="", outputs=result)
 demo.launch()