Aryan Jain commited on
Commit
c54e39d
·
1 Parent(s): 12473a4

define color incident map

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +6 -0
src/streamlit_app.py CHANGED
@@ -345,6 +345,7 @@ def render_chart(
345
 
346
  if color_col:
347
  df_plot_copy = df_plot.copy()
 
348
 
349
  if color_col in incident_cols:
350
  df_plot_copy = _add_incident_category(df_plot_copy, color_col)
@@ -354,6 +355,9 @@ def render_chart(
354
  df_plot_copy["_incident_color"].tolist(),
355
  )
356
  )
 
 
 
357
 
358
  if active_incident_col is not None:
359
  x_col = "_incident_label"
@@ -373,6 +377,8 @@ def render_chart(
373
  )
374
  if active_incident_col is not None:
375
  bar_kwargs["x"] = "_incident_label"
 
 
376
  if color_incident_color_map and color_col in incident_cols:
377
  bar_kwargs["x"] = (
378
  "_incident_label" if active_incident_col is not None else x_col
 
345
 
346
  if color_col:
347
  df_plot_copy = df_plot.copy()
348
+ color_incident_color_map = None
349
 
350
  if color_col in incident_cols:
351
  df_plot_copy = _add_incident_category(df_plot_copy, color_col)
 
355
  df_plot_copy["_incident_color"].tolist(),
356
  )
357
  )
358
+
359
+ if not color_incident_color_map:
360
+ color_incident_color_map = incident_color_map
361
 
362
  if active_incident_col is not None:
363
  x_col = "_incident_label"
 
377
  )
378
  if active_incident_col is not None:
379
  bar_kwargs["x"] = "_incident_label"
380
+ color_incident_color_map = incident_color_map
381
+ bar_kwargs["color_discrete_map"] = color_incident_color_map
382
  if color_incident_color_map and color_col in incident_cols:
383
  bar_kwargs["x"] = (
384
  "_incident_label" if active_incident_col is not None else x_col