multi-thread didn't work so update to go back

Author: SoFrans

nulrdcscripts/tools/spectrogramgeneration/main.py

@@ -1,64 +1,33 @@
-# Coded with the help of Microsoft CoPilot
-
 import os
 import subprocess
 import progressbar
-from concurrent.futures import ThreadPoolExecutor
 from nulrdcscripts.tools.spectrogramgeneration.params import args
 
 
-def generate_spectrogram(input_path, channel, output_path, spectroname, ffmpegpath):
-    """Creates a spectrogram for a single audio track in the input"""
+def generate_spectrogram(input_path, channels, output_path, spectroname, ffmpegpath):
+    """Creates a spectrogram for each audio track in the input"""
     spectrogram_resolution = "1920x1080"  # Updated resolution
-    output = os.path.join(output_path, f"{spectroname}_spectrogram0{channel + 1}_s.png")
-    spectrogram_args = [ffmpegpath]
-    spectrogram_args += ["-loglevel", "error", "-y"]
-    spectrogram_args += ["-i", input_path, "-lavfi"]
-    if channel > 1:
-        spectrogram_args += [
-            f"[0:a:{channel}]showspectrumpic=mode=separate:s={spectrogram_resolution}"
-        ]
-    else:
-        spectrogram_args += [
-            f"[0:a:{channel}]showspectrumpic=s={spectrogram_resolution}"
-        ]
-    spectrogram_args += [output]
-    subprocess.run(spectrogram_args)
-    print(f"*** Spectrogram for channel {channel + 1} generated ***")
-
-
-def generate_all_spectrograms(
-    input_path, channels, output_path, spectroname, ffmpegpath
-):
-    """Creates spectrograms for all audio channels in parallel"""
-    widgets = [
-        " [",
-        progressbar.Percentage(),
-        "] ",
-        progressbar.Bar(),
-        " (",
-        progressbar.ETA(),
-        ") ",
-    ]
-    bar = progressbar.ProgressBar(max_value=int(channels), widgets=widgets)
+    bar = progressbar.ProgressBar(max_value=int(channels))
     bar.start()
-
-    with ThreadPoolExecutor(max_workers=int(channels)) as executor:
-        futures = [
-            executor.submit(
-                generate_spectrogram,
-                input_path,
-                index,
-                output_path,
-                spectroname,
-                ffmpegpath,
-            )
-            for index in range(int(channels))
-        ]
-        for i, future in enumerate(futures):
-            future.result()
-            bar.update(i + 1)
-
+    for index in range(int(channels)):
+        output = os.path.join(
+            output_path, f"{spectroname}_spectrogram0{index + 1}_s.png"
+        )
+        spectrogram_args = [ffmpegpath]
+        spectrogram_args += ["-loglevel", "error", "-y"]
+        spectrogram_args += ["-i", input_path, "-lavfi"]
+        if int(channels) > 1:
+            spectrogram_args += [
+                f"[0:a:{index}]showspectrumpic=mode=separate:s={spectrogram_resolution}"
+            ]
+        else:
+            spectrogram_args += [
+                f"[0:a:{index}]showspectrumpic=s={spectrogram_resolution}"
+            ]
+        spectrogram_args += [output]
+        subprocess.run(spectrogram_args)
+        bar.update(index + 1)
+        print(f"*** Spectrogram for channel {index + 1} generated ***")
     bar.finish()
 
 
@@ -103,9 +72,7 @@ def checkOrCreateOutput(output_path, input_path):
 def callableSpectrogram(ffprobe_path, input_path, output_path, ffmpeg_path):
     channels = getnumberchannels(ffprobe_path, input_path)
     spectroname, _ = os.path.splitext(os.path.basename(input_path))
-    generate_all_spectrograms(
-        input_path, channels, output_path, spectroname, ffmpeg_path
-    )
+    generate_spectrogram(input_path, channels, output_path, spectroname, ffmpeg_path)
 
 
 def main():