Update app.py

app.py

@@ -1,11 +1,11 @@
-#Importing required libraries
-import spaces
+# Importing required libraries
 import gradio as gr
-import random
 from PIL import Image
+import random
 
 # Define pixel sizes for different levels
 pixel_sizes = [128, 96, 64, 32, 24, 16, 12, 8, 4, 2]
+
 # Function to pixelate an image
 def pixelate(image, pixel_size):
     # Reduce the image size
@@ -13,67 +13,84 @@ def pixelate(image, pixel_size):
     # Scale back to original size
     return small.resize(image.size, Image.Resampling.NEAREST)
 
-# dictionary of photo name and url
+# 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",
-    "Kajol"   : "./Celebs/Kajol.png",
-    "Oscar Isaac" : "./Celebs/OscarIsaac.jpg",
-    "Nayanthara" : "./Celebs/Nayanthara.jpg",
-    "Dhanush" : "./Celebs/Dhanush.jpg",    
+    "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"
 }
 
-curr_size = 256
-def Clear(prev_size, photo=celeb_list[1], folder=celeb_folder, next=False):
-    global photo_answer
-    global curr_size
-    
-    if next:
-        index = celeb_list.index(photo) + 1
-        photo = celeb_list[index]
-        
-    print(f"Processing {photo}")
-    image_path = folder[photo]
-    img = Image.open(BytesIO(image_path))    
-
-    for curr_size in pixel_sizes:
-        if curr_size<prev_size:
-            result_img = pixelate(img, curr_size)
-            photo_answer = photo
-            return result_img, photo
+# Initialize global variables
+current_index = 0
+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
 
-# # Display the pixelated images progressively
-# for idx, img in enumerate(images):
-#     plt.figure(figsize=(6, 6))
-#     plt.imshow(img)
-#     plt.axis("off")
-#     plt.title(f"Guess the Portrait - Level {idx + 1}")
-#     plt.show()
+def next_image(prev_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 = 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
 
+def progressive_clear(pixel_size):
+    """
+    Progressively clears the pixelation of the current image.
+    """
+    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
 
-MARKDOWN = "Guess the Cartoon Character before the Image becomes fully clear"
-with gr.Blocks(theme=gr.themes.Ocean()) as demo:
+# Gradio App Layout
+MARKDOWN = "## Guess the Celebrity before the Image Clears Up!"
+with gr.Blocks() as demo:
     gr.Markdown(MARKDOWN)
-
-    Old_size = curr_size
-    Actual_Answer = photo_answer
-    if_next=True
-
+    
     with gr.Row():
         with gr.Column(scale=1):
-            pixelated_image = gr.Image(type='pil', image_mode='RGB', label='Pixelated Image')
-            with gr.Accordion("Answer", open=False):
-                answer = gr.Textbox()
+            pixelated_image = gr.Image(type='pil', label='Pixelated Image')
+            with gr.Accordion("Reveal Answer", open=False):
+                answer = gr.Textbox(label="Current Celebrity")
 
-        Start_button = gr.Button(value='Start', variant='huggingface')
-        Clear_button = gr.Button(value='Clear', variant='primary')
-        Next_button = gr.Button(value='Next', variant='secondary')
+        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')
 
-    Start_button.click(fn=Clear, inputs=[256], outputs=[pixelated_image, answer])
-    Clear_button.click(fn=Clear, inputs=[Old_size], outputs=[pixelated_image, answer])
-    Next_button.click(fn=Clear, inputs=[Old_size, Actual_Answer, if_next], outputs=[pixelated_image, answer])
+    # Define button actions
+    Start_button.click(fn=clear_and_start, inputs=[], outputs=[pixelated_image, answer])
+    Clear_button.click(fn=progressive_clear, inputs=[gr.Number(value=current_pixel_size)], outputs=[pixelated_image, answer])
+    Next_button.click(fn=next_image, inputs=[gr.Number(value=current_pixel_size)], outputs=[pixelated_image, answer])
 
-demo.launch(debug=False, show_error=True)
+demo.launch(debug=True, show_error=True)