Update app.py

app.py

@@ -1,44 +1,22 @@
 from transformers import pipeline
 import gradio as gr
-import random
 
-def actor_game(text, emotion):
+pipe = pipeline("text-classification", model="muhnatha/distilbert-base-uncased-game")
+
+def actor_game(emotion, text):
     pred_emotion = pipe(text)
     if pred_emotion[0]['label'] == emotion:
-        return "YOU WON", emotion, pred_emotion[0]['label']
+        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 = {
-    "sadness": "Be a student who just had a bad grade",
-    "joy": "Be a football player who just won the world cup",
-    "love": "Be a teenager who has a crush but she can only talk to her friend",
-    "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"
-}
-
-task = random.choice(list(question.items()))
-
-def play(click_count=0):
-    if click_count == 0:
-        task = random.choice(list(question.items()))
+        return "YOU LOSE",emotion,pred_emotion[0]['label']
 
-    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()
+iface = gr.Interface(
+    fn=actor_game,
+    inputs=[gr.Textbox(label="What emotion do you want to train?",placeholder = "joy, sadness, fear, surprise, love, anger"), 
+            gr.Textbox(label="your dialog")],
+    outputs=[gr.Textbox(label="Result"),gr.Textbox(label="Expected"),gr.Textbox(label="your emotion")],
+    title="Be an Actor",
+    description="Type your dialog",
+)
 
-play()
+iface.launch()