Update app.py

app.py

@@ -1,23 +1,28 @@
 import numpy as np
 import soundfile as sf
 import gradio as gr
+import librosa
 
-def binauralize(audio_file, simulate_rotation, rotation_speed):
+def binauralize(audio_file, simulate_rotation, rotation_speed, auto_rotation):
     """
     Simulate a binaural (stereo) effect by applying a dynamic panning effect 
     to an input audio file. No HRIR files are required.
+
+    If auto_rotation is enabled, AI beat detection (via librosa) is used to 
+    determine the rotation speed based on the tempo of the audio.
     
     Parameters:
         audio_file (str): Path to input audio file (mono or stereo).
         simulate_rotation (bool): If True, apply a dynamic rotation (panning) effect.
         rotation_speed (float): Speed of the rotation effect (in Hz).
+        auto_rotation (bool): If True, auto-detect rotation speed using AI beat detection.
         
     Returns:
         output_file (str): Path to the output stereo audio file.
         status (str): Status message.
     """
     try:
-        # Load input audio file
+        # Load input audio file using soundfile
         audio, sr = sf.read(audio_file)
     except Exception as e:
         return None, f"Error reading input audio file: {e}"
@@ -26,6 +31,19 @@ def binauralize(audio_file, simulate_rotation, rotation_speed):
     if audio.ndim > 1:
         audio = np.mean(audio, axis=1)
     
+    # If auto_rotation is enabled, use librosa to detect tempo and adjust rotation speed.
+    if auto_rotation:
+        try:
+            # librosa expects float32 input.
+            audio_float = audio.astype(np.float32)
+            tempo, _ = librosa.beat.beat_track(y=audio_float, sr=sr)
+            rotation_speed = tempo / 60.0  # Convert BPM to Hz (approx.)
+            status_msg = f"Auto rotation enabled: Detected tempo = {tempo:.1f} BPM, setting rotation speed = {rotation_speed:.3f} Hz."
+        except Exception as e:
+            status_msg = f"Auto rotation failed, using user provided rotation speed. Error: {e}"
+    else:
+        status_msg = "Using user provided rotation speed."
+    
     # Create a time vector for the audio length
     t = np.arange(len(audio)) / sr
     
@@ -55,7 +73,7 @@ def binauralize(audio_file, simulate_rotation, rotation_speed):
     except Exception as e:
         return None, f"Error writing output audio file: {e}"
     
-    return output_file, "Binaural conversion complete!"
+    return output_file, f"Binaural conversion complete! {status_msg}"
 
 # Create an enhanced UI using Gradio Blocks and Tabs.
 with gr.Blocks(title="SonicOrbit", css="""
@@ -66,7 +84,7 @@ with gr.Blocks(title="SonicOrbit", css="""
     """) as demo:
     
     gr.Markdown("<div class='title'>SonicOrbit</div>")
-    gr.Markdown("<div class='subtitle'>Binaural 360 Audio Converter with Dynamic Rotation</div>")
+    gr.Markdown("<div class='subtitle'>Binaural 360 Audio Converter with Dynamic Rotation & AI Beat Detection</div>")
     
     with gr.Tabs():
         with gr.Tab("Converter"):
@@ -74,7 +92,8 @@ with gr.Blocks(title="SonicOrbit", css="""
                 input_audio = gr.Audio(type="filepath", label="Upload Audio (Mono or Stereo)")
             with gr.Row():
                 simulate_rotation = gr.Checkbox(label="Simulate Rotation", value=True)
-                rotation_speed = gr.Slider(0.01, 1.0, value=0.1, step=0.01, label="Rotation Speed (Hz)")
+                rotation_speed = gr.Slider(0.01, 5.0, value=0.1, step=0.01, label="Rotation Speed (Hz)")
+                auto_rotation = gr.Checkbox(label="Auto Detect Rotation Speed (AI)", value=False)
             convert_button = gr.Button("Convert Audio")
             with gr.Row():
                 output_audio = gr.Audio(type="filepath", label="Binaural Audio Output")
@@ -82,7 +101,7 @@ with gr.Blocks(title="SonicOrbit", css="""
             
             convert_button.click(
                 fn=binauralize,
-                inputs=[input_audio, simulate_rotation, rotation_speed],
+                inputs=[input_audio, simulate_rotation, rotation_speed, auto_rotation],
                 outputs=[output_audio, status_text]
             )
             
@@ -94,8 +113,10 @@ with gr.Blocks(title="SonicOrbit", css="""
             2. **Simulate Rotation:**  
                Enable this option to apply a dynamic panning effect that simulates a rotating sound source.
             3. **Rotation Speed:**  
-               Adjust the slider to set the speed of the rotation effect (in Hertz). A higher value rotates the audio field faster.
-            4. **Convert Audio:**  
+               Adjust the slider to set the speed of the rotation effect (in Hertz).
+            4. **Auto Detect Rotation Speed (AI):**  
+               Enable this option to let SonicOrbit analyze your audio and automatically set the rotation speed based on the detected tempo.
+            5. **Convert Audio:**  
                Click the **Convert Audio** button to process your audio file. The output is a binaural (stereo) audio file with the simulated 360° effect.
                
             Enjoy your immersive 3D audio experience!