From 61d1750d1da770f44e77703b1583932181c26036 Mon Sep 17 00:00:00 2001
From: tin2tin <tintwotin@gmail.com>
Date: Tue, 9 Apr 2024 10:17:46 +0200
Subject: [PATCH] Fix: Strip placement of audio strips

---
 __init__.py | 16 +++++++++++-----
 1 file changed, 11 insertions(+), 5 deletions(-)

diff --git a/__init__.py b/__init__.py
index e524583..4dad9eb 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2727,6 +2727,7 @@ class SEQUENCER_OT_generate_audio(Operator):
         movie_num_inference_steps = scene.movie_num_inference_steps
         movie_num_guidance = scene.movie_num_guidance
         audio_length_in_s = scene.audio_length_in_f / (scene.render.fps / scene.render.fps_base)
+        pipe = None
         #try:
         import torch
         import torchaudio
@@ -2751,6 +2752,7 @@ class SEQUENCER_OT_generate_audio(Operator):
             import numpy as np
             try:
                 from whisperspeech.pipeline import Pipeline
+                from resemble_enhance.enhancer.inference import denoise, enhance
             except ModuleNotFoundError:
                 print("Dependencies needs to be installed in the add-on preferences.")
                 self.report(
@@ -2865,6 +2867,7 @@ class SEQUENCER_OT_generate_audio(Operator):
 
             start_time = timer()
 
+            # Find free space for the strip in the timeline.
             if i > 0:
                 empty_channel = scene.sequence_editor.active_strip.channel
                 start_frame = (
@@ -2876,11 +2879,12 @@ class SEQUENCER_OT_generate_audio(Operator):
                 )
             else:
                 empty_channel = find_first_empty_channel(
-                    scene.frame_current,
-                    100000000000000000000,
+                scene.sequence_editor.active_strip.frame_final_start,
+                    (scene.movie_num_batch * (len(prompt) * 4)) + scene.frame_current,
                 )
                 start_frame = scene.frame_current
 
+
             # Bark.
             if addon_prefs.audio_model_card == "bark":
                 print("Generate: Speech (Bark)")
@@ -2976,6 +2980,7 @@ class SEQUENCER_OT_generate_audio(Operator):
 
                 pipe.generate_to_file(filename, prompt, speaker=speaker, lang='en', cps=int(scene.audio_speed))
 
+
             # Musicgen.
             elif addon_prefs.audio_model_card == "facebook/musicgen-stereo-medium":
                 print("Generate: MusicGen Stereo")
@@ -3080,11 +3085,12 @@ class SEQUENCER_OT_generate_audio(Operator):
                 filename = solve_path(str(seed) + "_" + prompt + ".wav")
                 write_wav(filename, rate, audio.transpose())
 
+            # Add Audio Strip
             filepath = filename
             if os.path.isfile(filepath):
-                empty_channel = find_first_empty_channel(
-                    start_frame, start_frame + scene.audio_length_in_f
-                )
+#                empty_channel = find_first_empty_channel(
+#                    start_frame, start_frame + scene.audio_length_in_f
+#                )
                 strip = scene.sequence_editor.sequences.new_sound(
                     name=prompt,
                     filepath=filepath,