Update app.py

app.py

@@ -2,6 +2,13 @@ from transformers import pipeline
 import gradio as gr
 import random
 
+def actor_game(text, emotion):
+    pred_emotion = pipe(text)
+    if pred_emotion[0]['label'] == emotion:
+        return "YOU WON", emotion, pred_emotion[0]['label']
+    else:
+        return "YOU LOSE", emotion, pred_emotion[0]['label']
+
 pipe = pipeline("text-classification", model="muhnatha/distilbert-base-uncased-game")
 
 question = {
@@ -11,43 +18,27 @@ question = {
     "anger": "Be a student who angry because he was bullied",
     "fear": "Be a grandpa who tries to run away because there is an alien in his farm",
     "surprise": "Be a child who just got a birthday surprise"
-} 
+}
 
-def actor_game(text, emotion, play_again):
-    pred_emotion = pipe(text)
-    if pred_emotion[0]['label'] == emotion:
-        result = "YOU WON"
-    else:
-        result = "YOU LOSE"
-    if play_again.lower() == "yes":
+task = random.choice(list(question.items()))
+
+def play(click_count=0):
+    if click_count == 0:
         task = random.choice(list(question.items()))
-        return result, emotion, pred_emotion[0]['label'], task[1], task[0]
-    else:
-        return result, emotion, pred_emotion[0]['label'], "Game Over", ""
 
-def initialize_interface():
-    task = random.choice(list(question.items()))
-    iface = gr.Interface(
-        fn=actor_game,
-        inputs=[
-            gr.Textbox(label=task[1], lines=3), 
-            gr.Textbox(label="Expected", value=task[0], visible=False),
-            gr.Radio(["Yes", "No"], label="Do you want to play again?")
-        ],
-        outputs=[
-            gr.Textbox(label="Result"), 
-            gr.Textbox(label="Expected"), 
-            gr.Textbox(label="Your emotion"), 
-            gr.Textbox(label="Next scenario"),
-            gr.Textbox(label="Next emotion", visible=False)
-        ],
-        title="Be an Actor",
-        description="Type your dialog",
-        allow_flagging="never"
-    )
-    iface.launch()
-    response = iface.launch()
-    if response[2] == "yes":  # Check if user wants to play again
-        initialize_interface()  # Restart the game
+    with gr.Blocks() as demo:
+        gr.Markdown(
+            """
+            # Be an Actor!
+            Type your dialog.
+            """
+        )
+        inp_dialog = gr.Textbox(placeholder=task[1])
+        inp_emotion = gr.Textbox(value=task[0], visible=False)
+        out = gr.Textbox()
+        inp_dialog.change(actor_game, [inp_dialog, inp_emotion], out)
+        reset_button = gr.Button()
+        reset_button.click(lambda: play(click_count + 1))
+    demo.launch()
 
-initialize_interface()
+play()