| import gradio as gr |
| import numpy as np, soundfile as sf, torch, math, time |
|
|
| GENRES = { |
| "Хаус":"groovy house beat, warm analog bassline, punchy 4/4 kick, 122 bpm", |
| "Техно":"driving hypnotic techno, dark warehouse, rolling kick, acid blips", |
| "Драм-н-бэйс":"fast breakbeat, rolling amen drums, deep reese bass, 174 bpm", |
| "Фонк":"aggressive phonk, cowbell melody, chopped soul samples, heavy 808", |
| "Лоу-фай":"lofi hip hop, dusty vinyl crackle, mellow keys, tape warmth", |
| "Синтвейв":"retro synthwave, neon 80s, lush pads, gated reverb drums", |
| "Трэп":"dark trap, booming 808 slides, crisp hi-hat rolls, half-time", |
| "Эмбиент":"ambient drone, vast space, slowly evolving textures", |
| "Гиперпоп":"hyperpop, pitched vocals, distorted supersaws, chaotic energy", |
| "Транс":"euphoric trance, rolling bassline, uplifting supersaw leads, 138 bpm", |
| "Чиллвейв":"chillwave, hazy tape chorus, dreamy detuned synths", |
| "Вэйпорвейв":"vaporwave, slowed funk sample, lush reverb, nostalgic", |
| "Хардбасс":"hardbass, donk bass, energetic kicks, cheeky rave energy", |
| } |
|
|
| _demucs_model=None; _musicgen=None |
| def get_demucs(): |
| global _demucs_model |
| if _demucs_model is None: |
| from demucs.pretrained import get_model |
| _demucs_model=get_model("htdemucs"); _demucs_model.eval() |
| return _demucs_model |
| def get_musicgen(): |
| global _musicgen |
| if _musicgen is None: |
| from transformers import pipeline |
| _musicgen=pipeline("text-to-audio","facebook/musicgen-small", |
| device=0 if torch.cuda.is_available() else "cpu") |
| return _musicgen |
|
|
| def analyze(path): |
| data,sr=sf.read(path,dtype="float32",always_2d=True) |
| mono=data.mean(axis=1); dur=len(mono)/sr |
| rms=float(np.sqrt(np.mean(mono[::max(1,len(mono)//4000)]**2))) |
| energy=min(99,int(rms*420)); hop,n=1024,len(mono)//1024 |
| e=np.array([np.sum(mono[i*hop:(i+1)*hop:4]**2) for i in range(n)]) |
| peaks=[i for i in range(2,n-2) if e[i]>e[i-1] and e[i]>=e[i+1] and e[i]>1.3*np.mean(e[max(0,i-22):i+22])] |
| bpm=None |
| if len(peaks)>=8: |
| cnt={} |
| for a,b in zip(peaks,peaks[1:]): |
| dt=(b-a)*hop/sr |
| if .25<dt<2: |
| v=60/dt |
| while v<80:v*=2 |
| while v>175:v/=2 |
| cnt[round(v)]=cnt.get(round(v),0)+1 |
| if cnt:bpm=max(cnt,key=cnt.get) |
| return dur,bpm,energy |
|
|
| def separate_vocals(path): |
| from demucs.apply import apply_model |
| from demucs.audio import AudioFile |
| model=get_demucs() |
| wav=AudioFile(path).read(streams=0,samplerate=model.samplerate,channels=model.audio_channels) |
| ref=wav.mean(0); wav=(wav-ref.mean())/ref.std() |
| with torch.no_grad(): |
| src=apply_model(model,wav[None],device="cpu",split=True,overlap=.25)[0] |
| src=src*ref.std()+ref.mean() |
| return src[model.sources.index("vocals")].numpy().T, model.samplerate |
|
|
| def gen_groove(genre): |
| prompt=GENRES.get(genre,f'experimental "{genre}" style, bold unusual sound design') |
| out=get_musicgen()(prompt,forward_params={"max_new_tokens":1024}) |
| return out["audio"][0],out["sampling_rate"] |
|
|
| def mix(voc,vsr,groove,gsr,aggr,out): |
| from scipy.signal import resample_poly |
| if gsr!=vsr: |
| g=math.gcd(vsr,gsr); groove=resample_poly(groove,vsr//g,gsr//g) |
| if groove.ndim==1:groove=np.stack([groove,groove],1) |
| groove=np.tile(groove,(len(voc)//len(groove)+1,1))[:len(voc)] |
| mx=voc*0.95+groove*min(1.1,0.35+aggr/200) |
| mx/=max(1e-6,np.max(np.abs(mx)))/0.95 |
| sf.write(out,mx,vsr) |
|
|
| def remix(track,genre,custom,prompt,aggr): |
| if track is None: |
| yield "⚠ СИСТЕМА: сначала вставь трек в слот",None; return |
| log=[]; push=lambda m:(log.append(m),"\n".join(log))[1]; t0=time.time() |
| yield push("СИСТЕМА: трек в слоте ✔ сворм из 8 каналов проснулся"),None |
| dur,bpm,energy=analyze(track) |
| yield push(f"УШАН-3000: длина {int(dur//60)}:{int(dur%60):02d} · темп ≈ {bpm or '?'} BPM · энергия {energy}%"),None |
| g=(custom or "").strip() or genre |
| yield push(f"ОРКЕСТРАТОР: жанр «{g}». раздаю задачи моделям…"),None |
| if prompt.strip(): yield push(f"ХАЙП-КРИТИК: пожелание «{prompt.strip()[:80]}» — учтено 👌"),None |
| yield push("ВОКАЛ-ХИРУРГ: demucs режет трек на вокал/музыку… (2–6 мин на бесплатном CPU, это нормально)"),None |
| try: voc,vsr=separate_vocals(track) |
| except Exception as ex: |
| yield push(f"⚠ не смог прочитать файл: {ex}\nпопробуй загрузить WAV"),None; return |
| yield push("ВОКАЛ-ХИРУРГ: вокал отделён чисто ✔"),None |
| yield push("РИТМ-АРХИТЕКТОР: musicgen генерит грув… (1–3 мин)"),None |
| groove,gsr=gen_groove(g) |
| yield push(f"РИТМ-АРХИТЕКТОР: грув готов ✔ {len(groove)/gsr:.0f} сек"),None |
| yield push("МИКС-МАСТЕР: свожу вокал+грув, мастерю до красного 🔴"),None |
| out="/tmp/svorm8_remix.wav"; mix(voc,vsr,groove,gsr,aggr,out) |
| yield push(f"✔ ГОТОВО за {(time.time()-t0)/60:.1f} мин. максимальный REMIX подан ⚡"),out |
|
|
| CSS=""" |
| @import url('https://fonts.googleapis.com/css2?family=Anton&family=JetBrains+Mono&display=swap'); |
| .gradio-container{background:#101114!important;font-family:'JetBrains Mono',monospace} |
| h1,h2,.prose h1{font-family:'Anton',sans-serif;color:#ffb43a!important} |
| button.primary{background:#ff5a1f!important;border:3px solid #000!important} |
| """ |
|
|
| with gr.Blocks() as demo: |
| gr.HTML("<h1 style='font-size:42px'>СВОРМ-8 · NEURO REMIX</h1>" |
| "<p style='color:#8a8f9c'>20 моделей в реестре · в работе: demucs + musicgen · " |
| "всё считается прямо здесь, на сервере HF · бесплатно = медленно, но честно</p>") |
| with gr.Row(): |
| with gr.Column(): |
| track=gr.Audio(type="filepath",label="🎧 СЛОТ ДЛЯ ТРЕКА") |
| with gr.Column(): |
| genre=gr.Radio(list(GENRES),value="Хаус",label="ЖАНР") |
| custom=gr.Textbox(label="СВОЙ ЖАНР (перекрывает пэд)") |
| prompt=gr.Textbox(label="ПОЖЕЛАНИЯ НЕЙРОНАМ") |
| aggr=gr.Slider(40,160,value=100,label="АГРЕССИЯ") |
| btn=gr.Button("🔥 ЗАПУСТИТЬ REMIX",variant="primary") |
| log=gr.Textbox(label="📟 ЭФИР НЕЙРОСЕТЕЙ",lines=12,interactive=False) |
| out=gr.Audio(label="🔊 ГОТОВЫЙ REMIX (тут же можно скачать)",type="filepath") |
| btn.click(remix,inputs=[track,genre,custom,prompt,aggr],outputs=[log,out]) |
| gr.HTML("<p style='color:#5b606b'>ни один музыкант не пострадал (почти) · remix максимальный гарантирован*</p>") |
|
|
| demo.queue(max_size=5).launch(css=CSS,theme=gr.themes.Base()) |
