duongthienz commited on
Commit
50419e0
·
verified ·
1 Parent(s): bfa5937

Add new tab Rename Speaker

Browse files
Files changed (1) hide show
  1. state.py +5 -5
state.py CHANGED
@@ -209,16 +209,16 @@ def updateMultiSelect():
209
  # ---------------------------------------------------------------------------
210
 
211
  def store_speaker_clips(fname, annotations, waveform, sample_rate):
212
- """Generate clips & segments and write them into session state."""
213
  clips, segments = utils.build_speaker_clips(annotations, waveform, sample_rate)
214
  st.session_state.speakerClips[fname] = clips
215
  st.session_state.speakerSegments[fname] = segments
216
  st.session_state.speakerWaveforms[fname] = (waveform, sample_rate)
217
- print(f"Generated {len(clips)} speaker clips for {fname}")
218
 
219
 
220
  def randomize_speaker_clip(file_index, speaker):
221
- """Replace a speaker's clip with a freshly randomized one."""
222
  segs = st.session_state.speakerSegments.get(file_index, {}).get(speaker)
223
  waveform_data = st.session_state.speakerWaveforms.get(file_index)
224
  if not segs or waveform_data is None:
@@ -226,7 +226,7 @@ def randomize_speaker_clip(file_index, speaker):
226
  waveform, sample_rate = waveform_data
227
  new_clip = utils.get_randomized_clip(waveform, sample_rate, segs)
228
  st.session_state.speakerClips[file_index][speaker] = new_clip
229
- print(f"Randomized clip for {speaker} in {file_index}")
230
 
231
 
232
  # ---------------------------------------------------------------------------
@@ -324,7 +324,7 @@ def run_analysis_loop(file_names, file_paths_dict, pipeline,
324
  st.session_state.results[fname] = (annotations, totalSeconds)
325
  st.session_state.summaries[fname] = {}
326
  st.session_state.unusedSpeakers[fname] = list(annotations.labels())
327
- with st.spinner(f"Generating clips {i+1}/{totalFiles}"):
328
  store_speaker_clips(fname, annotations, waveform, sample_rate)
329
  del waveform
330
 
 
209
  # ---------------------------------------------------------------------------
210
 
211
  def store_speaker_clips(fname, annotations, waveform, sample_rate):
212
+ """Generate samples & segments and write them into session state."""
213
  clips, segments = utils.build_speaker_clips(annotations, waveform, sample_rate)
214
  st.session_state.speakerClips[fname] = clips
215
  st.session_state.speakerSegments[fname] = segments
216
  st.session_state.speakerWaveforms[fname] = (waveform, sample_rate)
217
+ print(f"Generated {len(clips)} speaker samples for {fname}")
218
 
219
 
220
  def randomize_speaker_clip(file_index, speaker):
221
+ """Replace a speaker's audio sample with a freshly randomized one."""
222
  segs = st.session_state.speakerSegments.get(file_index, {}).get(speaker)
223
  waveform_data = st.session_state.speakerWaveforms.get(file_index)
224
  if not segs or waveform_data is None:
 
226
  waveform, sample_rate = waveform_data
227
  new_clip = utils.get_randomized_clip(waveform, sample_rate, segs)
228
  st.session_state.speakerClips[file_index][speaker] = new_clip
229
+ print(f"Randomized sample for {speaker} in {file_index}")
230
 
231
 
232
  # ---------------------------------------------------------------------------
 
324
  st.session_state.results[fname] = (annotations, totalSeconds)
325
  st.session_state.summaries[fname] = {}
326
  st.session_state.unusedSpeakers[fname] = list(annotations.labels())
327
+ with st.spinner(f"Generating audio samples {i+1}/{totalFiles}"):
328
  store_speaker_clips(fname, annotations, waveform, sample_rate)
329
  del waveform
330