update app.py

app.py

@@ -15,9 +15,36 @@ import subprocess
 import traceback
 import time
 import shutil
+import numpy as np
 
 path = ""  # Update with your path
 
+# Cross-correlation function
+def cross_correlation(x, y):
+    """
+    Compute the cross-correlation of two signals.
+    """
+    n = len(x)
+    correlation = np.correlate(x - np.mean(x), y - np.mean(y), mode='valid') / (n * np.std(x) * np.std(y))
+    return correlation
+
+# Modified getTrigger function using cross-correlation
+def getTrigger(ad, a, ref_length=100, max=1024):
+    """
+    Find the trigger point in the audio signal using waveform correlation.
+    """
+    ref_signal = a[ad:ad + ref_length]  # Reference signal segment
+
+    max_corr = 0
+    best_idx = ad
+    for i in range(ad, len(a) - ref_length):
+        corr = cross_correlation(ref_signal, a[i:i + ref_length])
+        if corr > max_corr:
+            max_corr = corr
+            best_idx = i
+
+    return best_idx
+
 def getRenderCords(ta: list, idx: int, res: int = 1024, size: tuple = (1280, 720)) -> list:
     i = idx - res // 2
     x, y = size[0] * .9 / -2, (ta[i] - 128) * (size[1] / 2000) + (size[1] * .7 / -2)
@@ -38,11 +65,6 @@ def center_to_top_left(coords, width=1280, height=720):
 def totopleft(coord, width=1280, height=720):
     return coord[0] + width / 2, height / 2 - coord[1]
 
-def getTrigger(ad: int, a: list, max: int = 1024) -> int:
-    i = ad
-    while not (a[i] < 128 and not a[i + 2] < 128 or i - ad > max):
-        i += 1
-    return i
 
 def extract_cover_image(mp3_file):
     audio = MP3(mp3_file, ID3=ID3)