duongthienz commited on
Commit
b40d148
·
verified ·
1 Parent(s): 394825c

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -607,6 +607,7 @@ def build_fig_bar(df2, speakerNames, catColors, speaker_color_map, get_display_n
607
  raw_to_display = {sp: get_display_name_fn(sp, currFile) for sp in df2["names"]}
608
  df2["display"] = df2["names"].map(raw_to_display)
609
  df2["mv_secs"] = df2["names"].map(lambda sp: mv_per_speaker.get(sp, 0.0))
 
610
  df2["sv_secs"] = (df2["values"] - df2["mv_secs"]).clip(lower=0)
611
  df2["time_label"] = df2["values"].apply(_seconds_to_hhmmss)
612
  df2["mv_time_label"] = df2["mv_secs"].apply(_seconds_to_hhmmss)
 
607
  raw_to_display = {sp: get_display_name_fn(sp, currFile) for sp in df2["names"]}
608
  df2["display"] = df2["names"].map(raw_to_display)
609
  df2["mv_secs"] = df2["names"].map(lambda sp: mv_per_speaker.get(sp, 0.0))
610
+ df2["mv_secs"] = df2["mv_secs"].clip(upper=df2["values"]) # cap at total time
611
  df2["sv_secs"] = (df2["values"] - df2["mv_secs"]).clip(lower=0)
612
  df2["time_label"] = df2["values"].apply(_seconds_to_hhmmss)
613
  df2["mv_time_label"] = df2["mv_secs"].apply(_seconds_to_hhmmss)