Load game supports media now?

app.py

@@ -96,8 +96,8 @@ def generate_story_and_timeline(no_ui_timeline_points=10, no_media_timeline_poin
     # Initialize merged timeline with UI and story timelines
     merged_timeline = ui_timeline + story_timeline
     no_media_merged_timeline = ui_timeline + story_timeline
-    print(merged_timeline)
-    print(no_media_merged_timeline)
+    #print(merged_timeline)
+    #print(no_media_merged_timeline)
 
     # Include media-related items if specified
     if include_media:
@@ -676,7 +676,9 @@ with gr.Blocks() as demo:
                         with gr.Group() as wamediagrouping: 
                             gr.HTML("Placeholder to load all media tests")
                         
-                        return wamediagrouping + media_items
+                        if wamedia: 
+                            return wamediagrouping + media_items
+                        else: wamediagrouping
                     
                     wachoices = gr.Radio(label="Your Choices", choices=initgameinfo[1])
                     wasubmit_btn = gr.Button("Make Choice")