Update app.py

app.py

@@ -1,96 +1,67 @@
 # Importing required libraries
 import gradio as gr
 from PIL import Image
-import random
+import os
 
-# Define pixel sizes for different levels
-pixel_sizes = [128, 96, 64, 32, 24, 16, 12, 8, 4, 2]
+# Define the folder containing images
+CELEB_FOLDER = "./Celebs"
 
-# Function to pixelate an image
-def pixelate(image, pixel_size):
-    # Reduce the image size
-    small = image.resize((image.size[0] // pixel_size, image.size[1] // pixel_size), Image.Resampling.NEAREST)
-    # Scale back to original size
-    return small.resize(image.size, Image.Resampling.NEAREST)
+# Get the list of celebrity images from the folder
+celebs = {os.path.splitext(file)[0]: os.path.join(CELEB_FOLDER, file) for file in os.listdir(CELEB_FOLDER)}
 
-# List of celebrities and folder paths
-celeb_list = ["Tom Cruise", "Jake Gyllenhal", "Natalie Portman", "Aubrey Plaza", "Oscar Isaac", "Kate Winslet", "Ellen DeGeneres"]
-celeb_folder = {
-    "Tom Cruise": "./Celebs/TomCruise.jpeg",
-    "Jake Gyllenhal": "./Celebs/JakeGyllenhal.jpg",
-    "Natalie Portman": "./Celebs/NataliePortman.png",
-    "Aubrey Plaza": "./Celebs/AubreyPlaza.jpg",
-    "Oscar Isaac": "./Celebs/OscarIsaac.jpg",
-    "Kate Winslet": "./Celebs/KateWinslet.jpg",
-    "Ellen DeGeneres": "./Celebs/EllenDeGeneres.jpg"
-}
+# Define pixel sizes for different levels of clarity
+PIXEL_SIZES = [128, 96, 64, 48, 32, 24, 16, 12, 8, 4, 2]
 
 # Initialize global variables
-current_index = 0
+current_image = None
+current_name = None
 current_pixel_size = 256
 
-def clear_and_start(prev_size=256):
-    """
-    Resets the current image and returns the first level of pixelation.
-    """
-    global current_index, current_pixel_size
-    current_pixel_size = prev_size
-    celebrity = celeb_list[current_index]
-    image_path = celeb_folder[celebrity]
-    
-    # Open and pixelate the image
-    img = Image.open(image_path)
-    result_img = pixelate(img, current_pixel_size)
-    return result_img, celebrity, current_pixel_size
+# Function to pixelate an image
+def pixelate(image, pixel_size):
+    # Reduce the image size
+    small = image.resize((image.size[0] // pixel_size, image.size[1] // pixel_size), Image.Resampling.NEAREST)
+    # Scale back to the original size
+    return small.resize(image.size, Image.Resampling.NEAREST)
 
-def next_image(pixel_size):
-    """
-    Moves to the next celebrity and pixelates the image.
-    """
-    global current_index, current_pixel_size
-    current_index = (current_index + 1) % len(celeb_list)  # Loop through the list
-    current_pixel_size = pixel_size
-    celebrity = celeb_list[current_index]
-    image_path = celeb_folder[celebrity]
-    
-    # Open and pixelate the image
-    img = Image.open(image_path)
-    result_img = pixelate(img, current_pixel_size)
-    return result_img, celebrity, current_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()))
+    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!"
 
-def progressive_clear(pixel_size):
-    """
-    Progressively clears the pixelation of the current image.
-    """
+# Reveal more of the image
+def clear_more():
     global current_pixel_size
-    current_pixel_size = max(pixel_size - 32, 2)  # Decrease pixel size for better clarity
-    celebrity = celeb_list[current_index]
-    image_path = celeb_folder[celebrity]
-    
-    # Open and pixelate the image
-    img = Image.open(image_path)
-    result_img = pixelate(img, current_pixel_size)
-    return result_img, celebrity, current_pixel_size
+    if current_pixel_size > PIXEL_SIZES[-1]:
+        current_pixel_size = PIXEL_SIZES[PIXEL_SIZES.index(current_pixel_size) + 1]
+    pixelated_image = pixelate(current_image, current_pixel_size)
+    return pixelated_image, "Getting clearer! Keep guessing!"
 
-# Gradio App Layout
-MARKDOWN = "## Guess the Celebrity before the Image Clears Up!"
+# Reveal the answer
+def reveal_answer():
+    return f"The correct answer is: {current_name}!"
+
+# Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Ocean()) as demo:
-    gr.Markdown(MARKDOWN)
-    
+    gr.Markdown("# 🎭 Guess the Celebrity Game!")
+    gr.Markdown("Can you identify the celebrity as the image becomes clearer?")
+
     with gr.Row():
-        with gr.Column(scale=1):
-            pixelated_image = gr.Image(type='pil', label='Pixelated Image')
-            with gr.Accordion("Reveal Answer", open=False):
-                answer = gr.Textbox(label="Current Celebrity")
+        pixelated_image = gr.Image(type="pil", label="Pixelated Image")
+        answer = gr.Textbox(label="Your Guess", placeholder="Type your guess here...")
+        result = gr.Label(label="Game Status")
 
-        with gr.Column(scale=1):
-            Start_button = gr.Button(value='Start', variant='primary')
-            Clear_button = gr.Button(value='Clear More', variant='secondary')
-            Next_button = gr.Button(value='Next Image', variant='success')
+    with gr.Row():
+        start_button = gr.Button("Start Game", variant='primary')
+        clear_button = gr.Button("Clear More", variant='secondary')
+        reveal_button = gr.Button("Reveal Answer", variant='success')
 
-    # Define button actions
-    Start_button.click(fn=clear_and_start, inputs=[], outputs=[pixelated_image, answer, gr.State(current_pixel_size)])
-    Clear_button.click(fn=progressive_clear, inputs=[gr.State(current_pixel_size)], outputs=[pixelated_image, answer, gr.State(current_pixel_size)])
-    Next_button.click(fn=next_image, inputs=[gr.State(current_pixel_size)], outputs=[pixelated_image, answer, gr.State(current_pixel_size)])
+    start_button.click(start_game, inputs=[], outputs=[pixelated_image, result])
+    clear_button.click(clear_more, inputs=[], outputs=[pixelated_image, result])
+    reveal_button.click(reveal_answer, inputs=[], outputs=result)
 
-demo.launch(debug=True, show_error=True)
+demo.launch(debug=True, show_error=True)