Analyze Data with AI

library(tidyverse)
library(elmer)

# Analyze data ------------------------------------------------------------

survey_spanish <-
  read_tsv("https://raw.githubusercontent.com/rstudio/r-community-survey/refs/heads/master/2020/data/2020-spanish-survey-final.tsv")

# * Translate text --------------------------------------------------------

translate_text <- function(text) {
  chat <- chat_openai(
    system_prompt = "You are a translator, taking Spanish responses and converting them to English.",
  )
  
  chat$chat(text)
}

translate_text("Mi nombre es David")

survey_translated <-
  survey_spanish |>
  drop_na(Qlike_best) |>
  select(Qlike_best) |>
  slice(1:50) |> 
  mutate(english = map_chr(Qlike_best, translate_text))

# * Identify themes -------------------------------------------------------

identify_themes <- function(text) {
  
  chat <- chat_openai(
    system_prompt = "You are a sociologist, looking for the top three themes in the responses to a survey.
  Each response is separated by the \n character."
  )
  
  chat$chat(text)
}

survey_translated |>
  mutate(english_combined = paste(english, collapse = "\n")) |> 
  distinct(english_combined) |> 
  pull(english_combined) |>
  identify_themes()