Update app.py

app.py

@@ -16,7 +16,7 @@ def get_segment(audio, sr):
 
     # If no onsets are detected, return a segment from the beginning
     if len(onset_frames) == 0:
-        return audio[:2048]  # Return the first segment of 1 second
+        return audio[:2048], sr  # Return the first segment of 1 second
 
     # Calculate energy over time
     energy = np.array([np.sum(np.abs(audio[i:i + 2048]**2)) for i in range(0, len(audio), 2048)])
@@ -36,14 +36,14 @@ def get_segment(audio, sr):
     end_sample = min(start_sample + segment_length, len(audio))
 
     # Return the selected segment
-    return audio[start_sample:end_sample]
+    return audio[start_sample:end_sample], sr
 
 # Function to extend music by adding silence
 def extend_music(file, added_minutes):
     audio, sr = load_audio(file)
     
     # Get a relevant segment from the audio
-    segment = get_segment(audio, sr)
+    segment, sr = get_segment(audio, sr)
 
     # Calculate the number of samples to add based on the duration in minutes
     additional_samples = int(added_minutes * 60 * sr)
@@ -51,7 +51,8 @@ def extend_music(file, added_minutes):
 
     # Normalize audio to the range of [-1, 1]
     extended_audio = extended_audio / np.max(np.abs(extended_audio))  # Avoid overflow
-    return extended_audio.astype(np.float32)
+
+    return extended_audio.astype(np.float32), sr  # Return audio and sample rate
 
 # Gradio UI setup
 with gr.Blocks() as app: