Update app.py

app.py

@@ -111,6 +111,7 @@ def generate_question(state):
 
 # -----------------------------
 # 6. Check the User's Answer
+#    Returns both the feedback message and a flag indicating correctness.
 # -----------------------------
 def check_answer(state, user_answer):
     correct_answer = state["answer"].lower().strip()
@@ -118,14 +119,16 @@ def check_answer(state, user_answer):
     if user_answer_clean == correct_answer:
         state["score"] += 1
         result_text = "Correct! Nice work!"
+        is_correct = True
     else:
         state["score"] -= 1
         result_text = f"Oops! The correct answer was: {state['answer']}"
+        is_correct = False
     difficulty_update = adjust_difficulty(state)
-    return result_text + "\n" + difficulty_update
+    return result_text + "\n" + difficulty_update, is_correct
 
 # -----------------------------
-# 7. Build the Gradio Interface with a Hip, Rainbow Theme
+# 7. Build the Gradio Interface with a Hip, Rainbow Theme and Fireworks Effects
 # -----------------------------
 custom_css = """
 body {
@@ -153,6 +156,7 @@ body {
 """
 
 with gr.Blocks(css=custom_css) as demo:
+    # Create persistent session state.
     state = gr.State(init_state())
     
     gr.Markdown("# Lingo Quest: The Ultimate Word Adventure")
@@ -171,24 +175,45 @@ with gr.Blocks(css=custom_css) as demo:
     user_answer = gr.Textbox(label="Your Answer")
     submit_button = gr.Button("Submit Answer")
     result_output = gr.Textbox(label="Result", interactive=False)
+    # New HTML output to display fireworks or celebration visuals.
+    fireworks_output = gr.HTML(label="Celebration", value="")
     
     def update_difficulty_label(state):
         return f"**Current Level**: {state['difficulty']} (Score: {state['score']})"
     
+    # Update the difficulty label when the interface loads.
     demo.load(fn=update_difficulty_label, inputs=state, outputs=difficulty_label)
     
     def on_generate_question(state):
         question = generate_question(state)
         difficulty_text = update_difficulty_label(state)
-        return question, difficulty_text
+        # Clear fireworks when a new challenge starts.
+        return question, difficulty_text, ""
     
-    generate_button.click(fn=on_generate_question, inputs=state, outputs=[question_output, difficulty_label])
+    generate_button.click(
+        fn=on_generate_question, 
+        inputs=state, 
+        outputs=[question_output, difficulty_label, fireworks_output]
+    )
     
     def on_submit_answer(user_answer, state):
-        feedback = check_answer(state, user_answer)
+        feedback, is_correct = check_answer(state, user_answer)
         difficulty_text = update_difficulty_label(state)
-        return feedback, difficulty_text
+        if is_correct:
+            # Fireworks effect: using an embedded GIF.
+            fireworks_html = """
+            <div style="text-align: center;">
+                <img src="https://media.giphy.com/media/3o6Zt481isNVuQI1l6/giphy.gif" alt="Fireworks!" style="width: 300px;"/>
+            </div>
+            """
+        else:
+            fireworks_html = ""
+        return feedback, difficulty_text, fireworks_html
     
-    submit_button.click(fn=on_submit_answer, inputs=[user_answer, state], outputs=[result_output, difficulty_label])
+    submit_button.click(
+        fn=on_submit_answer, 
+        inputs=[user_answer, state], 
+        outputs=[result_output, difficulty_label, fireworks_output]
+    )
     
 demo.launch()