Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 from transformers import pipeline
+import emoji
+import random
 
 # Load the text generation model
 pipe = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct", trust_remote_code=True)
@@ -8,6 +10,14 @@ pipe = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct", trus
 points = 0
 current_emoji = ""
 
+# Predefined emoji puzzles
+emoji_puzzles = [
+    emoji.emojize(":snake:") + emoji.emojize(":wizard:") + emoji.emojize(":lightning:"),
+    emoji.emojize(":apple:") + emoji.emojize(":woman_farmer:") + emoji.emojize(":door:") + emoji.emojize(":mirror:"),
+    emoji.emojize(":dog:") + emoji.emojize(":house:"),
+    emoji.emojize(":pizza:") + emoji.emojize(":slice:"),
+]
+
 # Function to reset the game and initialize introduction
 def reset_game():
     global points, current_emoji
@@ -15,12 +25,8 @@ def reset_game():
     # LLM introduces the game and asks the first emoji puzzle
     introduction = "Welcome to 'Guess the Word from Emojis'! 🎮 Here's how it works:\n\n- You'll get an emoji-based puzzle.\n- For each correct guess, you'll earn 1 point.\n- Wrong guesses will deduct 1 point.\n- Win by reaching 10 points, or lose if your score drops to -10 points.\n\nLet's begin!\n"
     
-    # Ask LLM for the first emoji puzzle
-    emoji_prompt = "Give me an emoji-based puzzle for a word or phrase guessing game."
-    response = pipe([{"role": "user", "content": emoji_prompt}], max_new_tokens=30)
-    
-    # Ensure to get the generated text correctly
-    current_emoji = response[0]['generated_text'] if response and isinstance(response, list) and len(response) > 0 else "❓"
+    # Randomly select an emoji puzzle
+    current_emoji = random.choice(emoji_puzzles)
 
     # Return introduction and the emoji puzzle
     return introduction + f"Here's your first puzzle: {current_emoji}"
@@ -60,11 +66,7 @@ def emoji_game(user_input=""):
             return f"🎉 Correct! You've reached 10 points! You win the game! 🏆"
         else:
             # Get next emoji puzzle for the user
-            next_emoji_prompt = "Give me another emoji-based puzzle."
-            response = pipe([{"role": "user", "content": next_emoji_prompt}], max_new_tokens=30)
-            
-            # Ensure to get the generated text correctly
-            current_emoji = response[0]['generated_text'] if response and isinstance(response, list) and len(response) > 0 else "❓"
+            current_emoji = random.choice(emoji_puzzles)
             return f"✅ Correct! Your current points: {points}. Here's your next puzzle: {current_emoji}"
     else:  # Incorrect guess
         points -= 1