Update app.py

app.py

@@ -5,10 +5,10 @@ import random
 import os
 
 # Define the folder containing images
-Character_FOLDER = "./Cartoon_Characters"
+CELEB_FOLDER = "./Celebs"
 
 # Get the list of celebrity images from the folder
-celebs = {os.path.splitext(file)[0]: os.path.join(Character_FOLDER, file) for file in os.listdir(Character_FOLDER)}
+celebs = {os.path.splitext(file)[0]: os.path.join(CELEB_FOLDER, file) for file in os.listdir(CELEB_FOLDER)}
 
 # Define pixel sizes for different levels of clarity
 PIXEL_SIZES = [64, 48, 32, 24, 16, 12, 8, 4, 2]
@@ -58,18 +58,18 @@ def clear_more():
 # Check the user's guess
 def check_guess(user_guess):
     if user_guess.strip().lower() == current_name.strip().lower():
-        return "Correct! 🎉 You guessed it right!"
+        return f"Correct! 🎉 You guessed it right! This character's name is {current_name}"
     else:
         return "Incorrect. Keep trying!"
 
 # Reveal the answer and original image
 def reveal_answer():
-    return current_image, 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:
-    gr.Markdown("# 🎭 Guess the Chartoon Character Game!")
-    gr.Markdown("Can you identify the Cartoon Characters as the image becomes clearer?")
+    gr.Markdown("# 🎭 Guess the Celebrity Game!")
+    gr.Markdown("Can you identify the celebrity as the image becomes clearer?")
 
     with gr.Row():
         pixelated_image = gr.Image(type="pil", label="Pixelated Image")
@@ -80,11 +80,11 @@ with gr.Blocks(theme=gr.themes.Ocean()) as demo:
         start_button = gr.Button("Start Game", variant='huggingface')
         clear_button = gr.Button("Clear More", variant='secondary')
         guess_button = gr.Button("Submit Guess", variant='primary')
-        reveal_button = gr.Button("Reveal Answer", variant='stop')
+        reveal_button = gr.Button("Reveal Answer", variant='success')
 
     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=[pixelated_image, result])
 
-demo.launch(debug=True, show_error=True)
+demo.launch(debug=True, show_error=True)