Create beat_generator/riffusion_cpu.py

beat_generator/riffusion_cpu.py

@@ -0,0 +1,31 @@
+import numpy as np
+import scipy.io.wavfile as wav
+import uuid
+import os
+
+def generate_audio_loop(style, bpm):
+    # fake riffusion-style sinewave for CPU-budget
+    sr = 44100
+    sec = 10
+    t = np.linspace(0, sec, sr*sec)
+
+    if style == "Country Rap":
+        freq = 90 + (bpm / 3)
+    elif style == "Trap":
+        freq = 120 + (bpm / 2)
+    elif style == "Dark Outlaw":
+        freq = 55 + (bpm / 4)
+    elif style == "Drill":
+        freq = 140 + (bpm / 1.5)
+    else:
+        freq = 100
+
+    wave = 0.4 * np.sin(2*np.pi*freq*t)
+    wave = (wave * 32767).astype(np.int16)
+
+    filename = f"generated_{uuid.uuid4().hex}.wav"
+    path = f"assets/generated_beats/{filename}"
+    os.makedirs("assets/generated_beats", exist_ok=True)
+    wav.write(path, sr, wave)
+
+    return path