Update app.py

app.py

@@ -154,24 +154,48 @@ def resample_waveform(waveform, original_sample_rate, target_sample_rate):
         return resampler(waveform), target_sample_rate
     return waveform, original_sample_rate
 
+
+
+
 def split_audio(waveform, sample_rate):
     segment_samples = segment_duration * sample_rate
     total_samples = waveform.size(0)
 
     segments = []
-    for start in range(0, total_samples, segment_samples):
-        end = start + segment_samples
-        if end <= total_samples:
+    # If the audio is shorter than the segment duration, just use the entire audio
+    if total_samples <= segment_samples:
+        segments.append(waveform)
+    else:
+        # Split the audio into segments of the specified duration
+        for start in range(0, total_samples, segment_samples):
+            end = min(start + segment_samples, total_samples)
             segment = waveform[start:end]
             segments.append(segment)
     
-    # In case audio length is shorter than segment length.
-    if len(segments) == 0: 
-        segment = waveform
-        segments.append(segment)
+    # Ensure we have at least one segment
+    if len(segments) == 0:
+        segments.append(waveform)
 
     return segments
 
+# def split_audio(waveform, sample_rate):
+#     segment_samples = segment_duration * sample_rate
+#     total_samples = waveform.size(0)
+
+#     segments = []
+#     for start in range(0, total_samples, segment_samples):
+#         end = start + segment_samples
+#         if end <= total_samples:
+#             segment = waveform[start:end]
+#             segments.append(segment)
+    
+#     # In case audio length is shorter than segment length.
+#     if len(segments) == 0: 
+#         segment = waveform
+#         segments.append(segment)
+
+#     return segments
+
 
 def safe_remove_dir(directory):
     """