Update app.py

app.py

@@ -44,15 +44,18 @@ def get_labels(audio_fp, min_speech_duration_ms, min_silence_duration_ms, auto_m
         else:
             new_labels.append(list(labels[i]))
 
+    extended = [False] * (len(new_labels) - 1)
     if extend_small_segments:
         for i, nl in enumerate(new_labels[:-1]):
             if nl[1] - nl[0] <= 1.2 and nl[0] + 1.2 < new_labels[i+1][0]:
                 nl[1] = nl[0] + 1.2
+                extended[i] = True
 
     if end_extension:
         for i, nl in enumerate(new_labels[:-1]):
-            if nl[0] + end_extension < new_labels[i+1][0]:
-                nl[1] = nl[0] + end_extension
+            if not extended[i]:
+                if nl[1] + end_extension < new_labels[i+1][0]:
+                    nl[1] = nl[1] + end_extension
 
     translate_labels = list(map(lambda x: f"{x[0]}\t{x[1]}\t{x[2]}", new_labels))