Update soni_translate/utils.py

soni_translate/utils.py

@@ -3,6 +3,8 @@ from .logging_setup import logger
 from urllib.parse import urlparse
 from IPython.utils import capture
 import re
+import soundfile as sf
+import numpy as np
 
 VIDEO_EXTENSIONS = [
     ".mp4",
@@ -66,6 +68,31 @@ def run_command(command):
         raise Exception(errors.decode())
 
 
+def write_chunked(
+    file,
+    data,
+    samplerate,
+    subtype=None,
+    endian=None,
+    format=None,
+    closefd=True,
+    chunk_size=0x1000
+):
+
+    data = np.asarray(data)
+    if data.ndim == 1:
+        channels = 1
+    else:
+        channels = data.shape[1]
+    with sf.SoundFile(
+        file, 'w', samplerate, channels,
+        subtype, endian, format, closefd
+    ) as f:
+        num_chunks = (len(data) + chunk_size - 1) // chunk_size
+        for chunk in np.array_split(data, num_chunks, axis=0):
+            f.write(chunk)
+
+
 def print_tree_directory(root_dir, indent=""):
     if not os.path.exists(root_dir):
         logger.error(f"{indent} Invalid directory or file: {root_dir}")
@@ -143,10 +170,6 @@ def manual_download(url, dst):
 
 
 def download_list(text_downloads):
-
-    if os.environ.get("ZERO_GPU") == "TRUE":
-        raise RuntimeError("This option is disabled in this demo.")
-    
     try:
         urls = [elem.strip() for elem in text_downloads.split(",")]
     except Exception as error:
@@ -484,4 +507,4 @@ def rename_file(current_name, new_name):
         return dir_new_name_file
     else:
         logger.error(f"File '{current_name}' does not exist.")
-        return None
+        return None