Update app.py

app.py

@@ -12,22 +12,41 @@ question = {
     "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 actor_game(text, emotion):
+def actor_game(text, emotion, play_again):
     pred_emotion = pipe(text)
     if pred_emotion[0]['label'] == emotion:
-        return "YOU WON",emotion,pred_emotion[0]['label']
+        result = "YOU WON"
     else:
-        return "YOU LOSE",emotion,pred_emotion[0]['label']
+        result = "YOU LOSE"
+    if play_again.lower() == "yes":
+        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", ""
 
-iface = gr.Interface(
-    fn=actor_game,
-    inputs=[gr.Textbox(label=task[1]), 
-            gr.Textbox(task[0],visible=False)],
-    outputs=[gr.Textbox(label="Result"),gr.Textbox(label="Expected"),gr.Textbox(label="your emotion")],
-    title="Be an Actor",
-    description="Type your dialog",
-)
+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"
+    )
+    response = iface.launch()
+    if response[2] == "yes":  # Check if user wants to play again
+        initialize_interface()  # Restart the game
 
-iface.launch()
+initialize_interface()