Update app.py

app.py

@@ -3,16 +3,21 @@ import librosa
 import numpy as np
 import soundfile as sf
 
-def reverse_segments(audio_path):
-    # Load audio and detect BPM
+def reverse_segments_advanced(audio_path):
+    # Load audio with high precision
     y, sr = librosa.load(audio_path, sr=None)
-    tempo, beats = librosa.beat.beat_track(y=y, sr=sr)
 
-    # Reverse the entire audio
-    y_reversed = y[::-1]
+    # Advanced BPM detection
+    onset_env = librosa.onset.onset_strength(y=y, sr=sr, aggregate=np.median)
+    tempo, beats = librosa.beat.beat_track(y=y, sr=sr, onset_envelope=onset_env, trim=False)
 
-    # Split into beat segments
+    # Convert beats to exact sample positions
     beat_frames = librosa.frames_to_samples(beats)
+
+    # Reverse the entire audio first
+    y_reversed = y[::-1]
+
+    # Cut into beat-aligned segments
     segments = [y_reversed[beat_frames[i]:beat_frames[i+1]] for i in range(len(beat_frames)-1)]
 
     # Reverse each segment back to forward
@@ -25,11 +30,11 @@ def reverse_segments(audio_path):
     return output_path
 
 iface = gr.Interface(
-    fn=reverse_segments,
+    fn=reverse_segments_advanced,
     inputs=gr.Audio(type="filepath"),
     outputs=gr.Audio(),
-    title="Beat-Reversed Music",
-    description="Reverses the entire song, then splits it by detected beats and plays each segment forward."
+    title="High-Precision Beat-Reversed Music",
+    description="More accurate BPM detection for reversing music and playing beat-aligned segments forward."
 )
 
 iface.launch()