Spaces:
Running on CPU Upgrade
Running on CPU Upgrade
Add new tab Rename Speaker
Browse files
state.py
CHANGED
|
@@ -209,16 +209,16 @@ def updateMultiSelect():
|
|
| 209 |
# ---------------------------------------------------------------------------
|
| 210 |
|
| 211 |
def store_speaker_clips(fname, annotations, waveform, sample_rate):
|
| 212 |
-
"""Generate
|
| 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
|
| 218 |
|
| 219 |
|
| 220 |
def randomize_speaker_clip(file_index, speaker):
|
| 221 |
-
"""Replace a speaker's
|
| 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
|
| 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
|
| 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 |
|