Update app.R

app.R

@@ -28,60 +28,61 @@ plot_factor <- function(pi_star_list,
     )
   }))
   
+  # Manual dodging: Create numeric x-positions with offsets
+  df$Level_num <- as.numeric(as.factor(df$Level))  # Convert Level to numeric (1, 2, ...)
+  if (n_strategies == 1) {
+    df$x_dodged <- df$Level_num  # No dodging for single strategy
+  } else {
+    # Apply ±offset for Democrat/Republican
+    df$x_dodged <- df$Level_num + ifelse(df$Strategy == "Democrat", -0.05, 0.05)
+  }
+  
   # Plot with ggplot2
-  p <- ggplot(df, aes(x = Level, y = Probability, color = Strategy)) +
-    # Thin segment from y=0 to y=Probability
+  p <- ggplot(df, aes(x = x_dodged, 
+                      y = Probability, 
+                      color = Strategy)) +
+    # Segment from y=0 to y=Probability
     geom_segment(
-      aes(xend = Level, y = 0, yend = Probability),
-      position = position_dodge(width = 0.7),
+      aes(x = x_dodged, xend = x_dodged, 
+          y = 0, yend = Probability),
       size = 0.3
     ) +
     # Point at the probability
     geom_point(
-      position = position_dodge(width = 0.7),
       size = 2.5
     ) +
-    # Optional text label for the probability value above the point
+    # Text label above the point
     geom_text(
-      aes(label = sprintf("%.2f", Probability)),
-      position = position_dodge(width = 0.7),
+      aes(x = x_dodged, label = sprintf("%.2f", Probability)),
       vjust = -0.7,
       size = 3
     ) +
-    # If you want error bars, re-activate lines below and adjust as desired
-    # geom_errorbar(
-    #   aes(ymin = Probability - zStar * SE, ymax = Probability + zStar * SE),
-    #   position = position_dodge(width = 0.7),
-    #   width = 0.2,
-    #   size = 0.3
-    # ) +
-    
+    # Set x-axis with original Level labels
+    scale_x_continuous(
+      breaks = unique(df$Level_num),
+      labels = unique(df$Level)
+    ) +
     # Labels
     labs(
       title = sprintf("Optimal Distribution for *%s*", 
-                      gsub(factor_name,pattern = "\\.",replace=" ")),
+                      gsub(factor_name, pattern = "\\.", replace = " ")),
       x = "Level",
       y = "Probability"
     ) +
-    
     # Apply Tufte's minimalistic theme
-    theme_minimal(base_size = 16,
-                  base_line_size = 0) +
+    theme_minimal(base_size = 16, base_line_size = 0) +
     theme(
       legend.position = "none",
-      legend.title    = element_blank(),
-      # Remove or soften grid lines
+      legend.title = element_blank(),
       panel.grid.major = element_blank(),
       panel.grid.minor = element_blank(),
-      # Keep a minimal axis line
       axis.line = element_line(color = "black", size = 0.3),
       axis.text.x = element_text(angle = 45, hjust = 1)
     ) +
-    
     # Manual color scale for different strategies
-    scale_color_manual(values = c("Democrat"   = "#89cff0",
+    scale_color_manual(values = c("Democrat" = "#89cff0",
                                   "Republican" = "red",
-                                  "Optimal"    = "black"))
+                                  "Optimal" = "black"))
   
   return(p)
 }