Update infer/lib/audio.py

infer/lib/audio.py

@@ -6,7 +6,6 @@ import av
 import traceback
 from io import BytesIO
 
-
 def wav2(i, o, format):
     inp = av.open(i, "rb")
     if format == "m4a":
@@ -29,7 +28,6 @@ def wav2(i, o, format):
     out.close()
     inp.close()
 
-
 def load_audio(file, sr):
     try:
         # Clean the file path
@@ -47,10 +45,23 @@ def load_audio(file, sr):
         traceback.print_exc()
         raise RuntimeError(f"Failed to load audio: {e}")
 
-    return np.frombuffer(out, np.float32).flatten()
-
+    audio_data = np.frombuffer(out, np.float32).flatten()
+    
+    # Ensure the function always returns a numpy array, even if it's empty
+    if audio_data is None or len(audio_data) == 0:
+        raise RuntimeError("Failed to load audio: no data returned.")
+    
+    return audio_data
 
 def clean_path(path_str):
     if platform.system() == "Windows":
         path_str = path_str.replace("/", "\\")
     return path_str.strip().strip('"').strip("\n").strip('"').strip()
+
+# Example usage
+if __name__ == "__main__":
+    try:
+        audio_data = load_audio("example.wav", 16000)
+        print("Audio data loaded successfully")
+    except RuntimeError as e:
+        print(e)