Update app.R

app.R

@@ -27,9 +27,44 @@ download_private_csv <- function(repo_id, filename) {
     stop("Failed to download dataset")
   }
 }
-MLB <- download_private_csv("TimStats/StatcastDataAll", "MLB.csv") %>% filter(season == 2024)
-AAA <- download_private_csv("TimStats/StatcastDataAll", "AAA.csv") %>% filter(season == 2024)
-FSL <- download_private_csv("TimStats/StatcastDataAll", "FSL.csv") %>% filter(season == 2024)
+download_private_parquet <- function(repo_id, filename) {
+  library(httr)
+  library(arrow)
+  
+  # Create the direct download URL based on your example
+  url <- paste0("https://huggingface.co/datasets/", repo_id, "/resolve/main/", filename, "?download=true")
+  
+  # Create a temporary file
+  temp_file <- tempfile(fileext = ".parquet")
+  
+  # Download directly to file
+  response <- GET(
+    url,
+    add_headers(Authorization = paste("Bearer", Sys.getenv("GETCSV"))),
+    write_disk(temp_file, overwrite = TRUE)
+  )
+  
+  # Check if download was successful
+  if (status_code(response) == 200) {
+    tryCatch({
+      # Read the parquet file
+      data <- read_parquet(temp_file)
+      file.remove(temp_file)
+      return(data)
+    }, error = function(e) {
+      file.remove(temp_file)
+      stop(paste("Error reading parquet file:", e$message))
+    })
+  } else {
+    file.remove(temp_file)
+    stop(paste("Failed to download file. Status code:", status_code(response)))
+  }
+}
+
+
+MLB <- download_private_parquet("TimStats/StatcastDataAll", "MLB25.parquet")
+AAA <- download_private_parquet("TimStats/StatcastDataAll", "AAA25.parquet")
+FSL <- download_private_parquet("TimStats/StatcastDataAll", "FSL25.parquet")
 
 
 MLB <- MLB %>%