Update tools/cmd/infer_cli.py

tools/cmd/infer_cli.py

@@ -1,6 +1,10 @@
 import argparse
 import os
 import sys
+from pydub import AudioSegment
+import matplotlib.pyplot as plt
+import numpy as np
+
 
 now_dir = os.getcwd()
 sys.path.append(now_dir)
@@ -63,5 +67,32 @@ def main():
     wavfile.write(args.opt_path, wav_opt[0], wav_opt[1])
 
 
+
+# Load the audio file
+audio = AudioSegment.from_file(wav_opt)  # Replace with your audio file path
+
+# Display basic information about the audio file
+print(f"Channels: {audio.channels}")
+print(f"Sample Width: {audio.sample_width} bytes")
+print(f"Frame Rate (Sample Rate): {audio.frame_rate} Hz")
+print(f"Frame Width: {audio.frame_width} bytes")
+print(f"Length: {len(audio)} ms")
+
+# Convert the audio data to a numpy array for visualization
+samples = np.array(audio.get_array_of_samples())
+
+# If the audio has more than one channel, split the samples into multiple arrays
+if audio.channels == 2:
+    samples = samples.reshape((-1, 2))
+
+# Plot the waveform
+plt.figure(figsize=(15, 5))
+plt.plot(samples[:1000])  # Plotting first 1000 samples for clarity
+plt.title("Waveform of the Audio File")
+plt.xlabel("Sample")
+plt.ylabel("Amplitude")
+plt.show()
+
+
 if __name__ == "__main__":
     main()