Update app.py

app.py

@@ -62,9 +62,9 @@ def check_guess(user_guess):
     else:
         return "Incorrect. Keep trying!"
 
-# Reveal the answer
+# Reveal the answer and original image
 def reveal_answer():
-    return f"The correct answer is: {current_name}!"
+    return current_image, f"The correct answer is: {current_name}!"
 
 # Create the Gradio interface
 with gr.Blocks(theme=gr.themes.Ocean()) as demo:
@@ -78,13 +78,13 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
 
     with gr.Row():
         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')
+        clear_button = gr.Button("Clear More", variant='secondary')
+        guess_button = gr.Button("Submit Guess", variant='primary')
+        reveal_button = gr.Button("Reveal Answer", variant='stop')
 
     start_button.click(start_game, inputs=[], outputs=[pixelated_image, result])
     clear_button.click(clear_more, inputs=[], outputs=[pixelated_image, result])
     guess_button.click(check_guess, inputs=answer_input, outputs=result)
-    reveal_button.click(reveal_answer, inputs=[], outputs=result)
+    reveal_button.click(reveal_answer, inputs=[], outputs=[pixelated_image, result])
 
 demo.launch(debug=True, show_error=True)