SonicOrbit / app.py
Bils's picture
Update app.py
3736599 verified
import numpy as np
import soundfile as sf
import gradio as gr
import librosa
def binauralize(audio_file, simulate_rotation, rotation_speed, auto_rotation):
try:
audio, sr = sf.read(audio_file)
except Exception as e:
return None, f"Error reading input audio file: {e}"
if audio.ndim > 1:
audio = np.mean(audio, axis=1)
if auto_rotation:
try:
audio_float = audio.astype(np.float32)
tempo, _ = librosa.beat.beat_track(y=audio_float, sr=sr)
tempo = float(tempo)
rotation_speed = tempo / 60.0
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."
t = np.arange(len(audio)) / sr
if simulate_rotation:
angle = 2 * np.pi * rotation_speed * t
left = np.cos(angle) * audio
right = np.sin(angle) * audio
else:
left = audio
right = audio
binaural_audio = np.stack((left, right), axis=-1)
max_val = np.max(np.abs(binaural_audio))
if max_val > 0:
binaural_audio = binaural_audio / max_val
output_file = "output_binaural.wav"
try:
sf.write(output_file, binaural_audio, sr)
except Exception as e:
return None, f"Error writing output audio file: {e}"
return output_file, f"Binaural conversion complete! {status_msg}"
with gr.Blocks(title="SonicOrbit", css="""
/* Custom CSS to enhance spacing and font styling */
.title { font-size: 2.5em; font-weight: bold; text-align: center; margin-bottom: 0.5em; }
.subtitle { font-size: 1.2em; text-align: center; margin-bottom: 1em; }
.footer { text-align: center; font-size: 0.9em; margin-top: 2em; color: #555; }
""") as demo:
gr.Markdown("<div class='title'>SonicOrbit</div>")
gr.Markdown("<div class='subtitle'>Binaural 360 Audio Converter with Dynamic Rotation & AI Beat Detection</div>")
with gr.Tabs():
with gr.Tab("Converter"):
with gr.Row():
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, 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")
status_text = gr.Textbox(label="Status", interactive=False)
convert_button.click(
fn=binauralize,
inputs=[input_audio, simulate_rotation, rotation_speed, auto_rotation],
outputs=[output_audio, status_text]
)
with gr.Tab("Instructions"):
gr.Markdown("""
### How to Use SonicOrbit
1. **Upload Audio:**
Upload a mono or stereo audio file. If you upload a stereo file, it will be converted to mono by averaging the channels.
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).
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!
""")
gr.Markdown("""
<div class='footer'>
© 2025 SonicOrbit. All rights reserved.
<br>
Created with ❤️ by <a href="https://bilsimaging.com" target="_blank" style="color: #88aaff;">bilsimaging.com</a>
</div>
""")
gr.HTML("""
<div style="text-align: center; margin-top: 1rem;">
<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2FSonicOrbit" target="_blank">
<img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2FSonicOrbit&countColor=%23263759" alt="visitor badge" /></a>
</div>
""")
if __name__ == "__main__":
demo.launch()