more demo defaults

dp_wizard/shiny/__init__.py

@@ -5,14 +5,15 @@
 from shiny import ui, reactive, Inputs, Outputs, Session
 
 from dp_wizard.utils.argparse_helpers import CLIInfo
-from dp_wizard.utils.csv_helper import read_csv_names
+from dp_wizard.utils.csv_helper import read_csv_names, name_to_id
 from dp_wizard.shiny import (
     about_panel,
     analysis_panel,
     dataset_panel,
     results_panel,
     feedback_panel,
 )
+from dp_wizard.utils.code_generators.analyses import histogram
 
 
 app_ui = ui.page_bootstrap(
@@ -89,27 +90,37 @@ def _clip(n: float, lower_bound: float, upper_bound: float) -> float:
 
 def make_server_from_cli_info(cli_info: CLIInfo):
     def server(input: Inputs, output: Outputs, session: Session):  # pragma: no cover
-        if cli_info.is_demo:
-            initial_contributions = 10
-            initial_private_csv_path = Path(__file__).parent.parent / "tmp" / "demo.csv"
+        demo = cli_info.is_demo
+
+        initial_contributions = 1 if not demo else 10
+        initial_private_csv_path = (
+            Path() if not demo else Path(__file__).parent.parent / "tmp" / "demo.csv"
+        )
+
+        if demo:
             _make_demo_csv(initial_private_csv_path, initial_contributions)
-            initial_column_names = read_csv_names(Path(initial_private_csv_path))
+            initial_column_names = read_csv_names(initial_private_csv_path)
+            demo_col = "grade"
+            assert (
+                demo_col in initial_column_names
+            ), f"{demo_col} not in {initial_column_names}"
+            id = name_to_id(demo_col)
         else:
-            initial_contributions = 1
-            initial_private_csv_path = ""
             initial_column_names = []
+            id = "typing placeholder: shouldn't be used"
 
-        contributions = reactive.value(initial_contributions)
-        private_csv_path = reactive.value(str(initial_private_csv_path))
-        column_names = reactive.value(initial_column_names)
-
+        contributions = reactive.value(1 if not demo else initial_contributions)
         public_csv_path = reactive.value("")
-        analysis_types = reactive.value({})
-        lower_bounds = reactive.value({})
-        upper_bounds = reactive.value({})
-        bin_counts = reactive.value({})
+        private_csv_path = reactive.value(
+            "" if not demo else str(initial_private_csv_path)
+        )
+        column_names = reactive.value(initial_column_names)
+        analysis_types = reactive.value({} if not demo else {id: histogram.name})
+        lower_bounds = reactive.value({} if not demo else {id: 0.0})
+        upper_bounds = reactive.value({} if not demo else {id: 100.0})
+        bin_counts = reactive.value({} if not demo else {id: 10})
         groups = reactive.value([])
-        weights = reactive.value({})
+        weights = reactive.value({} if not demo else {id: str(1)})
         epsilon = reactive.value(1.0)
         released = reactive.value(False)
 

dp_wizard/shiny/results_panel.py

@@ -47,7 +47,7 @@ def button(
 
 def _strip_ansi(e):
     """
-    >>> e = Exception('\x1B[0;31mValueError\x1B[0m: ...')
+    >>> e = Exception('\x1b[0;31mValueError\x1b[0m: ...')
     >>> _strip_ansi(e)
     'ValueError: ...'
     """