forgot to edit old function

app.py

@@ -84,7 +84,7 @@ def create_story(timeline):
             story.append(f"The player interacted with {entry[2]}.")
     return " ".join(story)
 
-def generate_story_and_timeline():
+def generate_story_and_timeline(include_media=False):
     # Pick 10 random UI items
     random_ui_items = pick_random_items(player_engagement_items, 10)
 
@@ -92,9 +92,16 @@ def generate_story_and_timeline():
     ui_timeline = generate_timeline(random_ui_items, "UI")
     story_timeline = generate_timeline(story_events, "Story")
 
-    # Merge the timelines
+    # Initialize merged timeline with UI and story timelines
     merged_timeline = ui_timeline + story_timeline
 
+    # Include media-related items if specified
+    if include_media:
+        media_files = generate_media_file_list(5)
+        rendered_media = render_media_with_dropdowns(media_files)
+        media_timeline = generate_timeline(rendered_media, "Media")
+        merged_timeline += media_timeline
+
     # Sort the merged timeline based on the random numbers
     merged_timeline.sort(key=lambda x: x[0])