Update app.py

app.py

@@ -27,6 +27,9 @@ celeb_folder = {
 
 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]
@@ -37,7 +40,9 @@ def Clear(prev_size, photo=celeb_list[1], folder=celeb_folder, next=False):
 
     for curr_size in pixel_sizes:
         if curr_size<prev_size:
-           return pixelate(img, curr_size), photo, curr_size
+            result_img = pixelate(img, curr_size)
+            photo_answer = photo
+            return result_img, photo
 
 
 # # Display the pixelated images progressively
@@ -54,7 +59,7 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
     gr.Markdown(MARKDOWN)
 
     Old_size = curr_size
-    Actual_Answer = answer
+    Actual_Answer = photo_answer
     if_next=True
 
     with gr.Row():
@@ -63,10 +68,12 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
             with gr.Accordion("Answer", open=False):
                 answer = gr.Textbox()
 
+        Start_button = gr.Button(value='Start', variant='huggingface')
         Clear_button = gr.Button(value='Clear', variant='primary')
         Next_button = gr.Button(value='Next', variant='secondary')
 
-    Clear_button.click(fn=Clear, inputs=[Old_size], outputs=[pixelated_image, answer, curr_size])
-    Next_button.click(fn=Clear, inputs=[Old_size, Actual_Answer, if_next], outputs=[pixelated_image, answer, prev_size])
+    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])
 
 demo.launch(debug=False, show_error=True)