Update process_data.R

process_data.R

@@ -154,25 +154,26 @@ if (grepl("\\.xlsx$", input_file)) {
 } else if (grepl("\\.csv$", input_file)) {
   data_driver <- read.csv(input_file, stringsAsFactors = FALSE)
 }
+trust_driver_analysis <- function(model_formula, data, output_text_file, csv_file) {
+  tryCatch({
+    data <- data[complete.cases(data), ]  # ✅ <---- ADD THIS LINE HERE
 
-# Trust model
-trust_driver_analysis(
-  Trust ~ Stability + Development + Relationship + Benefit + Vision + Competence,
-  data_driver,
-  output_text_file,
-  csv_output_path_trust)
-
-# Conditional models
-if (nps_present) trust_driver_analysis(NPS ~ Stability + Development + Relationship + Benefit + Vision + Competence, data_driver, output_text_file, csv_output_path_nps)
-if (loyalty_present) trust_driver_analysis(Loyalty ~ Stability + Development + Relationship + Benefit + Vision + Competence, data_driver, output_text_file, csv_output_path_loyalty)
-if (consideration_present) trust_driver_analysis(Consideration ~ Stability + Development + Relationship + Benefit + Vision + Competence, data_driver, output_text_file, csv_output_path_consideration)
-if (satisfaction_present) trust_driver_analysis(Satisfaction ~ Stability + Development + Relationship + Benefit + Vision + Competence, data_driver, output_text_file, csv_output_path_satisfaction)
-
-# TrustBuilder only for Excel files
-if (trustbuilder_present && grepl("\\.xlsx$", input_file)) {
-  data_builder_headers <- read_excel(input_file, sheet = "Builder", skip = 3, n_max = 2)
-  data_builder_rows <- read_excel(input_file, sheet = "Builder", skip = 5)
-  trust_builder_analysis(data_builder_rows, data_builder_headers, output_text_file, csv_output_path_trustbuilder)
-}
+    model <- lm(model_formula, data = data)
+    calc_relaimpo <- calc.relimp(model, type = "lmg", rela = TRUE)
+    average_importance <- mean(calc_relaimpo$lmg)
 
-log_message("Trust Driver and Builder Analysis Script Completed.", output_text_file)
+    file_conn <- file(output_text_file, open = "a")
+    full_output <- capture.output({
+      print("Trust Driver Analysis:\n")
+      print(calc_relaimpo)
+      cat("\nAverage Importance: ", average_importance, "\n")
+    })
+    writeLines(full_output, file_conn)
+    close(file_conn)
+
+    results <- data.frame(Predictor = names(calc_relaimpo$lmg), Importance = calc_relaimpo$lmg)
+    write.csv(results, file = csv_file, row.names = FALSE)
+  }, error = function(e) {
+    log_message(paste("Error in trust_driver_analysis:", e$message), output_text_file)
+  })
+}