Update app.R

app.R

@@ -561,7 +561,9 @@ server <- function(input, output, session) {
     # random coefficients
     beta <- rnorm(ncol(Xval), 0, 1)
     linear_part <- Xval %*% beta
-    Ysim <- as.numeric(linear_part + obsW * input$true_tau + rnorm(nunits, 0, input$noise_sd))
+    Ysim <- as.numeric(linear_part + 
+                         obsW * input$true_tau + 
+                         rnorm(nunits, 0, input$noise_sd))
     
     Y_data(Ysim)
   })
@@ -585,7 +587,8 @@ server <- function(input, output, session) {
     req(Y_data())  # Make sure Y_data is not NULL
     
     # Convert to data frame for DT
-    dfy <- data.frame(Y = Y_data())
+    dfy <- data.frame(obsW = RerandResult()$randomizations[1, ],
+                      Y = Y_data())
     
     # Optionally round numeric data
     dfy[] <- lapply(dfy, function(col) {