Update app.R

app.R

@@ -26,6 +26,8 @@ team_meta <- tryCatch({
   NULL
 })
 
+
+
 # -------------------- GLOBAL CSS --------------------
 app_css <- "
   body { background-color: #f5f5f5; font-family: 'Segoe UI', Arial, sans-serif; }
@@ -160,8 +162,32 @@ app_css <- "
 "
 
 
-  
+download_private_rds <- function(repo_id, filename) {
+  url <- paste0("https://huggingface.co/datasets/", repo_id, "/resolve/main/", filename)
+  
+  api_key <- Sys.getenv("HF_Token")
+  
+  if (api_key == "") {
+    stop("API key is not set.")
+  }
+  
+  response <- GET(url, add_headers(Authorization = paste("Bearer", api_key)))
+  
+  if (status_code(response) == 200) {
+    temp_file <- tempfile(fileext = ".rds")
+    
+    writeBin(content(response, "raw"), temp_file)
+    
+    data <- readRDS(temp_file)
+    
+    return(data)
+  } else {
+    stop(paste("Failed to download dataset. Status code:", status_code(response)))
+  }
+}
 
+stuffplus_recipe <- download_private_rds("CoastalBaseball/PitcherAppFiles", "stuffplus_recipe.rds")
+stuffplus_model <- xgb.load("stuffplus_xgb.json")
 
 process_dataset <- function(df) {
   if ("Batter" %in% names(df)) {
@@ -3360,13 +3386,22 @@ standardize_stuffplus_to_league <- function(data, league_comparison_data) {
       league_comparison_data[[col]] <- as.character(league_comparison_data[[col]])
     }
   }
-  
-  data$raw_stuff <- predict(stuffplus_model, data, type = "numeric")$.pred
+
+  df_processed <- bake(stuffplus_recipe, new_data = data)
+
+  df_matrix <- as.matrix(df_processed)
+                                 
+  data$raw_stuff <- predict(stuffplus_model, df_matrix)
 
   data <- data %>%
     mutate(data_ind = 1)
-  
-  league_comparison_data$raw_stuff <- predict(stuffplus_model, league_comparison_data, type = "numeric")$.pred
+
+  df_processed <- bake(stuffplus_recipe, new_data = league_comparison_data)
+
+  df_matrix <- as.matrix(df_processed)
+                                 
+  league_comparison_data$raw_stuff <- predict(stuffplus_model, df_matrix)
+
 
   league_comparison_data <- league_comparison_data %>%
     mutate(data_ind = 0)
@@ -3598,7 +3633,9 @@ create_advanced_pitch_characteristics <- function(data, player_name) {
     } else {
       message("WARNING: No reference data available. Using local standardization.")
       pitcher_data$raw_stuff <- tryCatch({
-        predict(stuffplus_model, pitcher_data, type = "numeric")$.pred
+          df_processed <- bake(stuffplus_recipe, new_data = pitcher_data)
+          df_matrix <- as.matrix(df_processed)                     
+          pitcher_data$raw_stuff <- predict(stuffplus_model, df_matrix)
       }, error = function(e) {
         message("Stuff+ prediction error: ", e$message)
         rep(NA_real_, nrow(pitcher_data))