Updated with: [removal] of display_segment

app.py

@@ -336,33 +336,7 @@ def plot_alignments(trellis, segments, word_segments, waveform=np.random.randn(1
 #plot_alignments(trellis, segments, word_segments, waveform, sample_rate)
 st.pyplot(plot_alignments(trellis, segments, word_segments, waveform=np.random.randn(1024), sample_rate=16000))
 
-# Part N: Display Segment
-sample_rate = 16000
-
-def display_segment(i):
-    word = word_segments[i]
-    x0 = int(word.start * sample_rate)
-    x1 = int(word.end * sample_rate)
-    print(f"{word.label} ({word.score:.2f}): {x0 / sample_rate:.3f} - {x1 / sample_rate:.3f} sec")
-    segment = waveform[x0:x1]
-    # Ensure the segment is a 1D array
-    segment = segment.squeeze()
-    return segment.numpy()  # Return the audio data as a numpy array
-
-# Display the full audio file
-st.audio(waveform.numpy(), format="audio/wav", sample_rate=sample_rate)
-
-# Display the audio for each segment
-for i in range(len(word_segments)):
-    segment_audio = display_segment(i)
-    # Ensure the segment_audio is a 1D array before passing to st.audio
-    if segment_audio.ndim == 2:
-        segment_audio = segment_audio.squeeze()
-    st.audio(segment_audio.astype('float32'), format="audio/wav", sample_rate=sample_rate)
-
-st.write(display_segment(3))
-
-# Part O: Audio generation for each segment
+# Part N: Audio generation 
 st.write('Abby Cadabby Transcript:')
 # Display the audio in the Streamlit app
 st.audio(SPEECH_FILE, format="audio/wav")