Load game supports media now?

app.py

@@ -503,13 +503,18 @@ class GameSession:
         state = all_states[self.current_location][self.current_state]
         choices = [f"{idx + 1}. {choice}" for idx, choice in enumerate(state['choices'])]
         return state['description'], choices, "\n".join(self.game_log)
+    
+    def get_current_state_media(self):
+        media = all_states[self.current_location][self.current_state]['media']
+        return media
+
 
 def start_game(starting_location='village', starting_state='start'):
     game_session = GameSession(starting_location, starting_state)
     description, choices, game_log = game_session.get_current_state_info()
     return description, choices, game_log, game_session
 
-def make_choice(choice, game_session): #Calls the nested make choice function in the game session class
+def make_choice(choice, game_session, with_media=False): #Calls the nested make choice function in the game session class
     if not choice:
         description, choices, game_log = game_session.get_current_state_info()
         return description, choices, "Please select a choice before proceeding.", game_session
@@ -517,7 +522,11 @@ def make_choice(choice, game_session): #Calls the nested make choice function in
     choice_index = int(choice.split('.')[0]) - 1 
     result = game_session.make_choice(choice_index)
 
-    return result[0], gr.update(choices=result[1]), result[2], game_session
+    if with_media:
+        media = game_session.get_current_state_media()
+        return result[0], gr.update(choices=result[1]), result[2], game_session, media
+    else:
+        return result[0], gr.update(choices=result[1]), result[2], game_session
 
 # def load_game(custom_config=None, with_media=False):
 #     global all_states
@@ -739,8 +748,8 @@ with gr.Blocks() as demo:
                     def dynamic_with_media(media_items):
                         print(media_items)
                         with gr.Group() as wamediagrouping: 
-                            gr.HTML("Placeholder to load all media tests")
-                            if not media_items:
+                            gr.HTML("Placeholder to load all media tests - still need to test clearing media on ")
+                            if media_items == []:
                                 gr.Markdown("No media items to display.")
                             else:
                                 for item in media_items:
@@ -754,8 +763,8 @@ with gr.Blocks() as demo:
                     wagame_session = gr.State(value=initgameinfo[3])
                     wasubmit_btn.click(
                         make_choice,
-                        inputs=[wachoices, wagame_session],
-                        outputs=[wadescription, wachoices, wagame_log, wagame_session]
+                        inputs=[wachoices, wagame_session, wamediabool],
+                        outputs=[wadescription, wachoices, wagame_log, wagame_session, wamedia]
                     )
                 with gr.Column(scale=1):
                     gr.Markdown("# Debugging")