library(tidyverse)
library(rio)

setwd("C:/Users/leoto/OneDrive/Documents/598/Final Lab")

mood <- import("sleo mood daylio.csv")
screen_time <- import("sleo screen time data.csv")
sleep <- import("sleo sleep data.csv")

mood$mood <- factor(mood$mood,levels = c("bad", "meh", "good", "rad"))

#mood chart
ggplot(data=mood) + 
    geom_bar(aes(x = Treat, fill = factor(mood)), position = position_dodge(preserve = 'single')) +
    geom_vline(xintercept=.5,linetype="solid", color="black") +
  labs(title = "Daily Mood" ,  x = "Mood Before and After Treatment" , y = "Count") +
    theme(axis.text.x = element_blank()) +
  theme_minimal()

#sleep chart
ggplot(data=sleep, aes(x=Day, y=minutes)) +
  geom_line(aes(y=minutes), color="darkblue") +
  labs(title="Sleep Over Time",
       subtitle="Sleep Before and After Treatment",
       caption="Sleep as measured by a daily app",
       y="# Minutes") +
  geom_smooth(method = "lm", color="red", fullrange=F, data=subset(sleep,Treat==0),se=FALSE,linetype="dotted") +
  geom_smooth(method = "lm", color="red", fullrange=F, data=subset(sleep,Treat==1),se=FALSE,linetype="dotted") +
   geom_vline(xintercept=14,linetype="solid", color="black") +
  geom_text(x = 10, y= 425, label="pre:495.71") +
  geom_text(x = 18 , y= 425, label="post:579.21") +
  theme_minimal()

#Screentime line chart
ggplot(data=screen_time, aes(x=Day, y=total_screentime)) +
  geom_line(aes(y=total_screentime), color="darkblue") +
  labs(titlesleep="Phone usage Over Time",
       subtitle="Phone Usage Before and After Treatment",
       caption="phone usage as measured by a daily app",
       y="# Minutes") +
  geom_smooth(method = "lm", color="red", fullrange=F, data=subset(screen_time,Treat==0),se=FALSE,linetype="dotted") +
  geom_smooth(method = "lm", color="red", fullrange=F, data=subset(screen_time,Treat==1),se=FALSE,linetype="dotted") +
   geom_vline(xintercept=14,linetype="solid", color="black") +
  geom_text(x = 10, y= 60, label="pre:103.71") +
  geom_text(x = 18 , y= 60, label="post:89.96") +
  theme_minimal()