Update app.py

app.py

@@ -129,12 +129,19 @@ def trim_audio(intervals, input_file_path, output_file_path):
     for i, (start_time, end_time) in enumerate(intervals):
         # extract the segment of the audio
         segment = audio[start_time*1000:end_time*1000]
-
-        # construct the output file path
         output_file_path_i = f"{output_file_path}_{i}.wav"
-
-        # export the segment to a file
-        segment.export(output_file_path_i, format='wav')
+        
+        if len(segment) < 3000:
+            # Calculate how many times to repeat the audio to make it at least 2 seconds long
+            repeat_count = (3000 // len(segment)) + 2
+            # Repeat the audio
+            longer_audio = segment * repeat_count
+            # Save the extended audio
+            print(f"Audio was less than 3 seconds. Extended to {len(longer_audio)} milliseconds.")
+            longer_audio.export(output_file_path_i, format='wav')
+        else:
+            print("Audio is already 3 seconds or longer.")
+            segment.export(output_file_path_i, format='wav')
 
 import re