fix beep detection retry

app.py

@@ -98,10 +98,22 @@ def detect_beeps(video_path, event_length=30):
     # top_q = np.max(corr) - 0.1
     # mean = np.mean(corr)
     # print(top_q, mean)
-
-    peaks, _ = find_peaks(corr, height=0.7, distance=fs)
-    event_start = int(peaks[0] / fs * fps)
-    event_end = int(peaks[-1] / fs * fps)
+    beep_height = 0.8
+    event_start = length
+    while length - event_start < fps * event_length:
+        peaks, _ = find_peaks(corr, height=beep_height, distance=fs)
+        event_start = int(peaks[0] / fs * fps)
+        event_end = int(peaks[-1] / fs * fps)
+        if event_end == event_start:
+            event_end = event_start + fps * event_length
+        beep_height -= 0.1
+        if beep_height <= 0.1:
+            event_start = 0
+            event_end = length
+            break
+    #peaks, _ = find_peaks(corr, height=0.7, distance=fs)
+    #event_start = int(peaks[0] / fs * fps)
+    #event_end = int(peaks[-1] / fs * fps)
     # plt.plot(corr)
     # plt.plot(peaks, corr[peaks], "x")
     # plt.savefig('beep.png')
@@ -123,10 +135,11 @@ def inference(stream_url, start_time, end_time, beep_detection_on, event_length,
         has_access = pbkdf2_sha256.verify(os.environ['DEV_API_TOKEN'], api_key)
         if not has_access:
             return "Invalid API Key"
+    event_length = int(event_length)
     if beep_detection_on:
         event_start, event_end = detect_beeps(in_video, event_length)
     print(event_start, event_end)
-    event_length = int(event_length)
+    
     cap = cv2.VideoCapture(in_video)
     length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
     frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))