Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
-from beat_generator.riffusion_cpu import generate_audio_loop
-from beat_generator.midi_tools import generate_midi_pattern
+import numpy as np
+from scipy.io.wavfile import write
+import os
+import uuid
+
 from content_generator.hooks import generate_hooks
 from content_generator.captions import generate_captions
 from content_generator.ideas import generate_content_ideas
@@ -8,23 +11,86 @@ from content_generator.scripts import generate_short_scripts
 from pack_builder.zip_builder import build_creator_pack
 
 
+# -----------------------------------------------------
+#     SIMPLE CPU-SAFE BEAT GENERATOR (NO MODELS)
+# -----------------------------------------------------
+def generate_cpu_safe_beat(style, bpm):
+    sr = 44100  # sample rate
+    seconds = 8  # loop length
+    t = np.linspace(0, seconds, int(sr * seconds))
+
+    beat = np.zeros_like(t)
+
+    # -------------------------------------------------
+    # STYLE PRESETS (lightweight procedural synthesis)
+    # -------------------------------------------------
+    if style == "Country Rap":
+        kick_freq = 55
+        snare_freq = 180
+        hat_freq = 8000
+
+    elif style == "Trap":
+        kick_freq = 45
+        snare_freq = 200
+        hat_freq = 10000
+
+    elif style == "Dark Outlaw":
+        kick_freq = 60
+        snare_freq = 150
+        hat_freq = 6000
+
+    elif style == "Drill":
+        kick_freq = 40
+        snare_freq = 160
+        hat_freq = 9000
+
+    else:  # BoomBap
+        kick_freq = 70
+        snare_freq = 190
+        hat_freq = 7000
+
+    # -------------------------------------------------
+    # BEAT PATTERN (Kick, Snare, Hats)
+    # -------------------------------------------------
+    beat_interval = 60 / bpm
+    hits = int(seconds / beat_interval)
+
+    for i in range(hits):
+        pos = int(i * beat_interval * sr)
+
+        # Kick
+        beat[pos:pos+2000] += np.sin(2 * np.pi * kick_freq * t[:2000]) * 0.9
+
+        # Snare
+        if i % 2 == 1:
+            beat[pos:pos+2000] += np.sin(2 * np.pi * snare_freq * t[:2000]) * 0.5
+
+        # Hats
+        beat[pos:pos+500] += (np.random.randn(500)) * 0.1
+
+    # Normalize
+    beat /= np.max(np.abs(beat) + 1e-9)
+
+    # Save file
+    filename = f"beat_{uuid.uuid4().hex}.wav"
+    filepath = f"/tmp/{filename}"
+    write(filepath, sr, (beat * 32767).astype(np.int16))
+
+    return filepath
+
+
+
 # -------------------------------
 #   BEAT GENERATOR CALLBACK
 # -------------------------------
 def ui_generate_beats(style, bpm, loops):
     audio = None
     loops = int(loops)
-    for _ in range(loops):
-        audio = generate_audio_loop(style, bpm)
-    return audio
-
 
+    for _ in range(loops):
+        audio = generate_cpu_safe_beat(style, bpm)
 
-# -------------------------------
-#   MIDI GENERATOR CALLBACK
-# -------------------------------
-def ui_generate_midi(bpm, pattern_type):
-    return generate_midi_pattern(bpm, pattern_type)
+    return audio
 
 
 
@@ -85,7 +151,6 @@ with gr.Blocks(title="AI Content + Beat Generator (Free Tier)") as demo:
         )
 
 
-
     # ---------------------------
     #     CONTENT GENERATOR
     # ---------------------------
@@ -108,7 +173,6 @@ with gr.Blocks(title="AI Content + Beat Generator (Free Tier)") as demo:
         )
 
 
-
     # ---------------------------
     #       PACK BUILDER
     # ---------------------------
@@ -132,6 +196,4 @@ with gr.Blocks(title="AI Content + Beat Generator (Free Tier)") as demo:
         )
 
 
-
-
 demo.launch()