Update app.py

app.py

@@ -34,14 +34,10 @@ def update_scenario(choice_index):
 with gr.Blocks() as app:
     with gr.Row():
         story_display = gr.Textbox(value=scenario, show_label=False, lines=7)
-        choices_buttons = [gr.Button(choice) for choice in choices]
     
-    def handle_choice(choice_index):
-        new_story = update_scenario(choice_index)
-        story_display.update(value=new_story)
-        return gr.update()
-
-    for i, button in enumerate(choices_buttons):
-        button.click(fn=handle_choice, inputs=[i], outputs=story_display)
+    with gr.Row():
+        for i, choice in enumerate(choices):
+            button = gr.Button(choice)
+            button.click(fn=update_scenario, data=i, outputs=story_display)
 
 app.launch()