Update app.py

app.py

@@ -1,19 +1,28 @@
-import spaces  # <-- DO NOT move this lower
+#!/usr/bin/env python3
+"""
+GhostAI Music Generator — ZeroGPU Space
+• Streams facebook/musicgen-medium from the Hub using your HF_TOKEN
+• Installs audiocraft 1.3.0 (no-deps) at runtime
+• Decorated with @spaces.GPU so each request gets a free transient GPU
+"""
+
+# ── 0. ZeroGPU decorator (must be first) ───────────────────────────
+import spaces  # <-- leave this as the very first import
 
 # ── 1. Standard libs ───────────────────────────────────────────────
 import os, sys, gc, time, warnings, tempfile, subprocess, random
 import numpy as np, psutil
 
-# ── 2. Torch stack (torch 2.1.0 & torchaudio 2.1.0 from requirements) ─
+# ── 2. Torch stack (torch 2.1.0 / torchaudio 2.1.0 from requirements) ─
 import torch, torchaudio
 
-# ── 3. UI / misc deps ─────────────────────────────────────────────
+# ── 3. Misc deps ──────────────────────────────────────────────────
 import gradio as gr
 from pydub import AudioSegment
 from huggingface_hub import login
 from torch.cuda.amp import autocast
 
-# ── 4. Torch <2.3 shim (transformers 4.38 calls get_default_device) ──
+# ── 4. Torch <2.3 shim (transformers 4.38 expects it) ─────────────
 if not hasattr(torch, "get_default_device"):
     torch.get_default_device = lambda: torch.device(
         "cuda" if torch.cuda.is_available() else "cpu"
@@ -22,7 +31,7 @@ if not hasattr(torch, "get_default_device"):
 warnings.filterwarnings("ignore")
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128"
 
-# ── 5. Install audiocraft at runtime (no deps) ─────────────────────
+# ── 5. Install audiocraft 1.3.0 (no deps) at runtime ──────────────
 try:
     from audiocraft.models import MusicGen
 except ModuleNotFoundError:
@@ -35,7 +44,7 @@ except ModuleNotFoundError:
 # ── 6. Hugging Face authentication ────────────────────────────────
 HF_TOKEN = os.getenv("HF_TOKEN")
 if not HF_TOKEN:
-    sys.exit("ERROR: Please add HF_TOKEN as a secret in your Space.")
+    sys.exit("ERROR: Add HF_TOKEN as a secret in your Space settings.")
 login(HF_TOKEN)
 
 # ── 7. Load MusicGen model ────────────────────────────────────────
@@ -45,15 +54,15 @@ musicgen.to(torch.get_default_device())
 musicgen.set_generation_params(duration=10, two_step_cfg=False)
 SR = musicgen.sample_rate
 
-# ── 8. Prompt builders (add more in same style) ───────────────────
+# ── 8. Prompt builders (add more as needed) ───────────────────────
 def _build(base,bpm,dr,syn,st,bass,gtr,def_bass,def_gtr,flow):
-    stxt = f" with {st}" if st!="none" else flow.format(bpm=bpm)
+    step = f" with {st}" if st!="none" else flow.format(bpm=bpm)
     return (f"{base}"
             f"{def_bass if bass=='none' else ', '+bass}"
             f"{def_gtr if gtr=='none' else ', '+gtr+' guitar riffs'}"
             f"{'' if dr=='none' else ', '+dr+' drums'}"
             f"{'' if syn=='none' else ', '+syn+' accents'}"
-            f"{stxt} at {bpm} BPM.")
+            f"{step} at {bpm} BPM.")
 
 def set_rhcp(bpm,dr,syn,st,bass,gtr):
     return _build("Instrumental funk rock",bpm,dr,syn,st,bass,gtr,
@@ -65,7 +74,7 @@ def set_nirvana(bpm,dr,syn,st,bass,gtr):
                   ", melodic basslines",", raw distorted guitar riffs",
                   "{bpm} BPM grungy pulse" if bpm>120 else "grungy rhythmic pulse")
 
-# (⚠️ Add your remaining genre functions here)
+# (➕ Add remaining genre prompt functions here the same way.)
 
 # ── 9. Audio post-FX helpers ──────────────────────────────────────
 def apply_eq(seg):   return seg.low_pass_filter(8000).high_pass_filter(80)
@@ -77,9 +86,9 @@ def log(stage=""):
         a = torch.cuda.memory_allocated()/1024**3
         r = torch.cuda.memory_reserved()/1024**3
         print(f"GPU alloc {a:.2f} GB  reserved {r:.2f} GB")
-    print(f"CPU {psutil.virtual_memory().percent}% used")
+    print(f"CPU {psutil.virtual_memory().percent}%")
 
-# ── 10. Core generation (ZeroGPU decorator) ───────────────────────
+# ── 10. Core generator (ZeroGPU wrapper) ──────────────────────────
 @spaces.GPU
 def generate(prompt,cfg,k,p,temp,
              total_len,chunk_len,xfade,
@@ -121,34 +130,33 @@ def generate(prompt,cfg,k,p,temp,
     track = track[: total_len*1000]
     track = apply_fade(apply_eq(track).normalize(headroom=-9.0))
 
-    out_path = "output_cleaned.mp3"
-    track.export(out_path, format="mp3", bitrate="128k",
+    out_f = "output_cleaned.mp3"
+    track.export(out_f, format="mp3", bitrate="128k",
                  tags={"title":"GhostAI Track","artist":"GhostAI"})
-    return out_path, "✅ Done!"
+    return out_f, "✅ Done!"
 
 def clear_inputs():
     return ("",3.0,250,0.9,1.0,30,10,1000,
             120,"none","none","none","none","none")
 
-# ── 11. Gradio Blocks UI ──────────────────────────────────────────
+# ── 11. Gradio UI ─────────────────────────────────────────────────
 with gr.Blocks(css="body{background:#0A0A0A;color:#E0E0E0;font-family:'Orbitron',sans-serif}") as demo:
     gr.Markdown("<h1 style='text-align:center'>👻 GhostAI MusicGen</h1>")
     prompt = gr.Textbox(lines=4, label="Instrumental Prompt")
 
     with gr.Row():
         gr.Button("RHCP 🌶️").click(set_rhcp,
-                                   inputs=[gr.State(120),"none","none","none","none","none"],
-                                   outputs=prompt)
+                                    inputs=[gr.State(120),"none","none","none","none","none"],
+                                    outputs=prompt)
         gr.Button("Nirvana 🎸").click(set_nirvana,
-                                     inputs=[gr.State(120),"none","none","none","none","none"],
-                                     outputs=prompt)
-        # ➕ Add more genre buttons here …
+                                      inputs=[gr.State(120),"none","none","none","none","none"],
+                                      outputs=prompt)
+        # ➕ add more genre buttons …
 
-    # Parameter controls
     cfg   = gr.Slider(1,10,3,label="CFG")
-    top_k = gr.Slider(10,500,250,step=10,label="Top-K")
-    top_p = gr.Slider(0,1,0.9,step=0.05,label="Top-P")
-    temp  = gr.Slider(0.1,2,1,label="Temperature")
+    k     = gr.Slider(10,500,250,step=10,label="Top-K")
+    p     = gr.Slider(0,1,0.9,step=0.05,label="Top-P")
+    temp  = gr.Slider(0.1,2,1,label="Temp")
     length= gr.Radio([30,60,90,120],value=30,label="Length (s)")
     chunk = gr.Slider(5,15,10,label="Chunk (s)")
     xfade = gr.Slider(100,2000,1000,label="Cross-fade (ms)")
@@ -162,15 +170,13 @@ with gr.Blocks(css="body{background:#0A0A0A;color:#E0E0E0;font-family:'Orbitron'
 
     gen = gr.Button("Generate 🚀")
     clr = gr.Button("Clear 🧹")
-    audio_out = gr.Audio(type="filepath")
-    status    = gr.Textbox(interactive=False)
+    audio = gr.Audio(type="filepath")
+    status= gr.Textbox(interactive=False)
 
     gen.click(generate,
-              inputs=[prompt,cfg,top_k,top_p,temp,length,chunk,xfade,
-                      bpm,drum,synth,steps,bass,gtr],
-              outputs=[audio_out,status])
+              [prompt,cfg,k,p,temp,length,chunk,xfade,bpm,drum,synth,steps,bass,gtr],
+              [audio,status])
     clr.click(clear_inputs, None,
-              [prompt,cfg,top_k,top_p,temp,length,chunk,xfade,
-               bpm,drum,synth,steps,bass,gtr])
+              [prompt,cfg,k,p,temp,length,chunk,xfade,bpm,drum,synth,steps,bass,gtr])
 
 demo.launch(share=False)