Update app.py

app.py

@@ -41,22 +41,22 @@ def clear_and_start(prev_size=256):
     # Open and pixelate the image
     img = Image.open(image_path)
     result_img = pixelate(img, current_pixel_size)
-    return result_img, celebrity
+    return result_img, celebrity, current_pixel_size
 
-def next_image(prev_size):
+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 = prev_size
+    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
+    return result_img, celebrity, current_pixel_size
 
 def progressive_clear(pixel_size):
     """
@@ -70,11 +70,11 @@ def progressive_clear(pixel_size):
     # Open and pixelate the image
     img = Image.open(image_path)
     result_img = pixelate(img, current_pixel_size)
-    return result_img, celebrity
+    return result_img, celebrity, current_pixel_size
 
 # Gradio App Layout
 MARKDOWN = "## Guess the Celebrity before the Image Clears Up!"
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Ocean()) as demo:
     gr.Markdown(MARKDOWN)
     
     with gr.Row():
@@ -89,8 +89,8 @@ with gr.Blocks() as demo:
             Next_button = gr.Button(value='Next Image', variant='success')
 
     # 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])
+    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)])
 
-demo.launch(debug=True, show_error=True)
+demo.launch(debug=True, show_error=True)