please work

app.py

@@ -1,70 +1,71 @@
 import gradio as gr
 import random
 
-def flappy_bird(_):
+def initialize_game():
     # Initialize or reset the game state
-    return get_game_state(2, 2, 0, 0), 0, 2, 2, 0, False
+    game_state = {
+        'height': 0,
+        'gap_position': 2,
+        'bird_position': 2,
+        'score': 0,
+        'game_over': False
+    }
+    return get_game_state(game_state), game_state
 
-def update_game(command, height, gap_position, bird_position, score, game_over):
+def update_game(command, game_state):
     PIPE_WIDTH = 3
     GAME_HEIGHT = 5
     GAP_SIZE = 2
 
-    if game_over or command == "start":
-        return get_game_state(2, 2, 0, 0), 0, 2, 2, 0, False
+    if game_state['game_over'] or command == "start":
+        return initialize_game()
 
     # Move bird based on command
-    bird_position += 1 if command.lower() != "flap" else -1
+    game_state['bird_position'] += 1 if command.lower() != "flap" else -1
 
     # Check bounds
-    if bird_position < 0 or bird_position >= GAME_HEIGHT:
-        return "You hit the top/bottom! Game Over! Your score: {}".format(score), 0, 2, 2, 0, True
+    if game_state['bird_position'] < 0 or game_state['bird_position'] >= GAME_HEIGHT:
+        game_state['game_over'] = True
+        return f"You hit the top/bottom! Game Over! Your score: {game_state['score']}", game_state
 
     # Update pipe position
-    height = (height + 1) % (PIPE_WIDTH + GAME_HEIGHT)
+    game_state['height'] = (game_state['height'] + 1) % (PIPE_WIDTH + GAME_HEIGHT)
 
     # Check for collision or passing through the gap
-    if height == PIPE_WIDTH:
-        if bird_position not in range(gap_position, gap_position + GAP_SIZE):
-            return "You hit a pipe! Game Over! Your score: {}".format(score), 0, 2, 2, 0, True
+    if game_state['height'] == PIPE_WIDTH:
+        if game_state['bird_position'] not in range(game_state['gap_position'], game_state['gap_position'] + GAP_SIZE):
+            game_state['game_over'] = True
+            return f"You hit a pipe! Game Over! Your score: {game_state['score']}", game_state
         else:
-            score += 1
-            gap_position = random.randint(0, GAME_HEIGHT - GAP_SIZE)
+            game_state['score'] += 1
+            game_state['gap_position'] = random.randint(0, GAME_HEIGHT - GAP_SIZE)
 
-    return get_game_state(height, gap_position, bird_position, score), height, gap_position, bird_position, score, False
+    return get_game_state(game_state), game_state
 
-def get_game_state(height, gap_position, bird_position, score):
+def get_game_state(game_state):
     PIPE_WIDTH = 3
     GAME_HEIGHT = 5
     GAP_SIZE = 2
-    game_state = "Score: {}\n".format(score)
+    display = "Score: {}\n".format(game_state['score'])
     for i in range(GAME_HEIGHT):
         line = ""
-        if height < PIPE_WIDTH and (i < gap_position or i >= gap_position + GAP_SIZE):
+        if game_state['height'] < PIPE_WIDTH and (i < game_state['gap_position'] or i >= game_state['gap_position'] + GAP_SIZE):
             line += "|"
         else:
             line += " "
-        line += "B" if i == bird_position else "_"
+        line += "B" if i == game_state['bird_position'] else "_"
         line += "\n"
-        game_state += line
-    return game_state
+        display += line
+    return display
 
 iface = gr.Interface(
     fn=update_game,
     inputs=[
         gr.Dropdown(choices=["flap", "start"], label="Control"),
-        gr.State(),
-        gr.State(),
-        gr.State(),
-        gr.State(),
         gr.State()
     ],
     outputs=[
         gr.Textbox(label="Game State"),
-        gr.State(),
-        gr.State(),
-        gr.State(),
-        gr.State(),
         gr.State()
     ],
     title="Simple Text-based Flappy Bird Game",