Update app.py

app.py

@@ -18,8 +18,9 @@ def reset_game():
     # 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)
-    current_emoji = response[0]['generated_text']
+    current_emoji = response[0]['generated_text'].strip()
     
+    # Return introduction and the emoji puzzle
     return introduction + f"Here's your first puzzle: {current_emoji}"
 
 # Function to handle the game logic and user interactions
@@ -46,7 +47,7 @@ def emoji_game(user_input=""):
     # Check if the user made a guess; LLM evaluates the guess
     guess_prompt = f"User guessed '{user_input}'. Was the guess correct for this emoji puzzle: {current_emoji}?"
     response = pipe([{"role": "user", "content": guess_prompt}], max_new_tokens=30)
-    response_text = response[0]['generated_text'].lower()
+    response_text = response[0]['generated_text'].strip().lower()
     
     # Process response: if it's correct, increase points; otherwise, decrease points
     if "correct" in response_text:  # Correct guess
@@ -57,7 +58,7 @@ def emoji_game(user_input=""):
             # 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)
-            current_emoji = response[0]['generated_text']
+            current_emoji = response[0]['generated_text'].strip()
             return f"✅ Correct! Your current points: {points}. Here's your next puzzle: {current_emoji}"
     else:  # Incorrect guess
         points -= 1