Lines of Dialogue in The Office

library(tidyverse)
library(schrute)
library(ggtext)

data(theoffice)

theoffice

episodes_per_writer <- theoffice %>%
  select(season, episode, writer) %>% 
  distinct() %>%
  count(writer, sort = TRUE)

episodes_per_writer

episodes <- theoffice %>%
  select(season, episode, episode_name, writer) %>%
  mutate(
    writer = case_when(
      str_detect(writer, "Mindy Kaling") ~ "Mindy Kaling",
      str_detect(writer, "B.J. Novak") ~ "B.J. Novak",
      TRUE ~ "Other writers"
    )
  ) %>%
  distinct()

episodes

dialogue <- theoffice %>%
  select(season, episode, character) %>%
  mutate(character = case_when(
    str_detect(character, "Kelly") ~ "Kelly",
    str_detect(character, "Ryan") ~ "Ryan",
    TRUE ~ "Other characters"
  )) %>%
  group_by(season, episode) %>%
  count(character) %>%
  mutate(prop = n / sum(n)) %>%
  ungroup() %>%
  filter(character != "Other characters")

dialogue

kelly_ryan <- writers %>%
  select(season, episode) %>%
  distinct() %>%
  mutate(Kelly = NA, Ryan = NA) %>%
  pivot_longer(c("Kelly", "Ryan"), names_to = "character", values_to = "dummy") %>%
  select(-dummy) %>%
  left_join(episodes, by = c("season", "episode")) %>%
  left_join(dialogue, by = c("season", "episode", "character")) %>%
  replace_na(list("n" = 0, "prop" = 0)) %>%
  mutate(
    season = paste("Season", as.numeric(season)),
    episode = as.numeric(episode),
    episode_name = case_when(
      character == "Ryan" ~ "",
      episode_name == "Lecture Circuit (Part 1)" ~ "Lecture Circuit (Parts 1&2)",
      episode_name == "Lecture Circuit (Part 2)" ~ "",
      episode_name == "Sexual Harassment" ~ "Sexual\nHarassment",
      TRUE ~ episode_name
    )
  )

kelly_ryan

notes <- tribble(
  ~season, ~episode, ~text, ~character, ~prop,
  "Season 1", 6, "— 20 lines of dialogue for every 100 lines in the episode", "Ryan", 0.2,
  "Season 1", 6, "— 10 lines of dialogue for every 100 lines in the episode", "Ryan", 0.1,
  "Season 1", 6, "— No lines of dialogue for every 100 lines in the episode", "Ryan", 0.0,
)

mk_color <- "#9D02D7"
bn_color <- "#FA8775"

ggplot() +
  geom_rect(data = kelly_ryan, aes(xmin = -Inf, xmax = Inf, ymin = -Inf, ymax = 0.205, color = writer), alpha = 0, size = 0.3) + 
  geom_col(data = kelly_ryan, aes(x = character, y = prop, fill = character), width = 0.7) +
  geom_text(data = filter(df, writer != "Other writers"), aes(x = 1.5, y = 0.235, label = episode_name), family = "Fira Sans Extra Condensed Light", size = 2.5, lineheight = 0.7) +
  geom_text(data = notes, aes(x = character, y = prop, label = text), family = "Fira Sans Extra Condensed Light", size = 3, lineheight = 0.9, hjust = -0.05) +
  facet_grid(season ~ episode, switch = "both") +
  scale_color_manual(values = c(bn_color, mk_color, "#F0F0F0")) +
  scale_fill_manual(values = c(mk_color, bn_color)) +
  guides(color = FALSE, fill = FALSE) +
  labs(
    title = glue::glue("<i>The Office</i> writers <span style='color:{mk_color}'>Mindy Kaling</span> and <span style='color:{bn_color}'>B.J. Novak</span> also play two characters on the show, <span style='color:{mk_color}'>Kelly Kapoor</span> and <span style='color:{bn_color}'>Ryan Howard</span>.<br>Do episodes they have written feature more lines of dialogue from their characters than episodes by other writers? "),
    subtitle = "No, but there are a few notable exceptions. B.J. Novak wrote <i>The Fire</i> (Season 2, Episode 4) where Ryan accidentally starts a fire in the office toaster oven. He also wrote <i>Initiation</i><br>(Season 3, Episode 5) where Ryan is taken to Dwight's beet farm for a sales department hazing. In the following episode <i>Diwali</i> (Season 3, Episode 6), written by Mindy Kaling,<br>Kelly invites the office to a Diwali celebration, which Michael mistakes for a Halloween party. Kaling also wrote <i>Night Out</i> (Season 4, Episode 15) where Michael and Dwight travel<br>to New York City to party with Ryan and it is revealed that Ryan is on drugs, and <i>Lecture Circuit</i> (Season 5, Episodes 16 & 17) where the office forgets to celebrate Kelly's birthday.",
    caption = "Data from the schrute R package (github.com/bradlindblad/schrute)\nCode to recreate this graphic at nsgrantham.com/the-office",
    x = NULL, y = NULL
  ) +
  theme_minimal(base_family = "Fira Sans Extra Condensed Light", base_size = 14) +
  theme(
    plot.title = element_markdown(family = "Fira Sans Extra Condensed", margin = margin(0, 0, 0.8, 0, unit = "line"), size = 22, lineheight = 1.2),
    plot.subtitle = element_markdown(size = 16, margin = margin(0, 0, 1.2, 0, unit = "line"), lineheight = 1.2),
    plot.background = element_rect(fill = "#F8F8FF", color = "#F8F8FF"),
    plot.title.position = "plot",
    plot.margin = margin(1, 1, 1, 1, unit = "line"),
    panel.grid.major.y = element_blank(),
    panel.grid.minor.y = element_blank(),
    panel.grid.major.x = element_blank(),
    strip.text.y.left = element_text(family = "Fira Sans Extra Condensed", angle = 0, margin = margin(0, 1, 0, 0, unit = "line")),
    strip.placement = "bottom",
    axis.text.y = element_blank(),
    axis.ticks.y = element_blank(),
    axis.text.x = element_blank()
  ) +
  coord_cartesian(clip = "off")

ggsave("the-office.png", width = 16, height = 12)