Update app.py

app.py

@@ -17,6 +17,7 @@ PIXEL_SIZES = [128, 96, 64, 48, 32, 24, 16, 12, 8, 4, 2]
 current_image = None
 current_name = None
 current_pixel_size = 256
+used_images = set()  # Track used images
 
 # Function to pixelate an image
 def pixelate(image, pixel_size):
@@ -27,11 +28,23 @@ def pixelate(image, pixel_size):
 
 # Start a new game
 def start_game():
-    global current_image, current_name, current_pixel_size
-    current_name, image_path = random.choice(list(celebs.items()))
+    global current_image, current_name, current_pixel_size, used_images
+
+    # Check if all images have been used; if so, reset
+    if len(used_images) == len(celebs):
+        used_images.clear()
+
+    # Select a random image that has not been used
+    available_images = list(set(celebs.keys()) - used_images)
+    current_name = random.choice(available_images)
+    used_images.add(current_name)
+
+    # Load the selected image
+    image_path = celebs[current_name]
     current_image = Image.open(image_path)
     current_pixel_size = PIXEL_SIZES[0]
     pixelated_image = pixelate(current_image, current_pixel_size)
+
     return pixelated_image, "Image Loaded. Guess the Celebrity!"
 
 # Reveal more of the image
@@ -54,7 +67,7 @@ def reveal_answer():
     return f"The correct answer is: {current_name}!"
 
 # Create the Gradio interface
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Ocean()) as demo:
     gr.Markdown("# 🎭 Guess the Celebrity Game!")
     gr.Markdown("Can you identify the celebrity as the image becomes clearer?")
 
@@ -64,10 +77,10 @@ with gr.Blocks() as demo:
         result = gr.Label(label="Game Status")
 
     with gr.Row():
-        start_button = gr.Button("Start Game")
-        clear_button = gr.Button("Clear More")
-        guess_button = gr.Button("Submit Guess")
-        reveal_button = gr.Button("Reveal Answer")
+        start_button = gr.Button("Start Game", variant='huggingface')
+        clear_button = gr.Button("Clear More", variant='primary')
+        guess_button = gr.Button("Submit Guess", variant='secondary')
+        reveal_button = gr.Button("Reveal Answer", variant='success')
 
     start_button.click(start_game, inputs=[], outputs=[pixelated_image, result])
     clear_button.click(clear_more, inputs=[], outputs=[pixelated_image, result])