Update app.py

app.py

@@ -160,7 +160,7 @@ def resample_waveform(waveform, original_sample_rate, target_sample_rate):
 def split_audio(waveform, sample_rate):
     segment_samples = segment_duration * sample_rate
     total_samples = waveform.size(0)
-
+    
     segments = []
     # If the audio is shorter than the segment duration, just use the entire audio
     if total_samples <= segment_samples:
@@ -172,16 +172,41 @@ def split_audio(waveform, sample_rate):
             segment = waveform[start:end]
             segments.append(segment)
     
-    # Ensure we have at least one segment
-    if len(segments) == 0:
-        segments.append(waveform)
-
+    # Ensure we have at least one segment with a minimum length
+    if len(segments) == 0 or all(len(segment) < 100 for segment in segments):
+        # Create a padded segment if audio is too short
+        padded_segment = torch.zeros(segment_samples)
+        if total_samples > 0:
+            padded_segment[:total_samples] = waveform
+        segments = [padded_segment]
+        
     return segments
 
 # def split_audio(waveform, sample_rate):
 #     segment_samples = segment_duration * sample_rate
 #     total_samples = waveform.size(0)
 
+#     segments = []
+#     # 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)
+    
+#     # 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