playbeats 0.1.1__py3-none-any.whl

playbeats-0.1.1.dist-info/METADATA

@@ -0,0 +1,29 @@
+Metadata-Version: 2.4
+Name: playbeats
+Version: 0.1.1
+Summary: Generate neural-network-powered beats from your terminal
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: pygame
+Requires-Dist: torch
+Requires-Dist: numpy
+
+### Installation
+
+```bash
+pip install playbeats
+```
+
+### Usage
+
+```bash
+playbeats --bpm=80 --energy=20
+```
+
+- `--bpm` — Set the beat tempo.
+- `--energy` — Set the beat energy.
+
+### Stop Playback
+
+Press `Ctrl+C` to stop playback.
+

playbeats-0.1.1.dist-info/RECORD

@@ -0,0 +1,23 @@
+train.py,sha256=HDgHdY1VH5eHtLpMeWlQ4zWtZEqZ9lVmIQ4STmMsu0Y,2243
+src/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/main.py,sha256=QUmsO_vqCBMGo1avAL-9IX8wgonMOISh9hXGzP8zg9Q,2918
+src/assets/beat_generator.pth,sha256=0R6m8B_IIuxN66LMWSFzgtWbu5Ql0mFMo2H6K14zyH8,993193
+src/assets/sounds/HIHAT.wav,sha256=n_idhvo6EvUzmPe8L9ijhdVkjWkLQttT_okuXd8Euko,325742
+src/assets/sounds/KICK.wav,sha256=N2JZkz1RK7DYQZBE6wfl-fvzvjHjrJjns3mgJNgw2GI,325742
+src/assets/sounds/SAMPLE.wav,sha256=7D4owJEn_vnsrAo1bSwQ-LUJv-0GwMyKdLzPmXnPUt0,325742
+src/assets/sounds/SNARE.wav,sha256=kb_asp_Oy2uOqeRFq9CUPVJKVF4LQ2yiIDYfg_K_yhc,325742
+src/assets/sounds/bass.wav,sha256=Ld89l-Aod3Hazr5w-rCrMK4l6RXrJVf0Mpky0R3Vhv0,373326
+src/assets/sounds/bongo.wav,sha256=9fg_Y-q4YA4EokL8CJ5ki1AERjSWRhmWd6xjo1o7IWQ,221262
+src/assets/sounds/clap.wav,sha256=jqSIuaRcxJHOKk0ic1B5i68nqLR8JDJycmsYLx_nxwM,110670
+src/assets/sounds/clap_fixed.wav,sha256=jqSIuaRcxJHOKk0ic1B5i68nqLR8JDJycmsYLx_nxwM,110670
+src/assets/sounds/closed-hihat.wav,sha256=1fMSbGBLU3sFYpZlDJ3x2yiu_clRV81bIhdoQ0C1luY,18510
+src/assets/sounds/openhihat.wav,sha256=mzuc5kIq6UTyUTJrWOuQkf9vuoZV7Z2e117fs8ckyWw,304898
+src/assets/sounds/tambourine.wav,sha256=tjXIZof3UwZyhkwuNWfrLEywiRfPsnWoytcM9lypXtY,78684
+src/assets/sounds/tom-high.wav,sha256=iyMHm1RiMKuKB-lFMPLUKaGcrJ1NiJZalQvg3cNFdVQ,653986
+src/assets/sounds/tom-low.wav,sha256=QsVsA_INlu7OX_DwJhfZtDaaCNwohDys620Z-Is3gw4,66406
+src/assets/sounds/tom-mid.wav,sha256=IqNEC3uJrvggOEeq2v0CJNOu-50gTIfwdBrQT4_Kddk,99718
+playbeats-0.1.1.dist-info/METADATA,sha256=oc6AuQBH_Gflr4tc4hQXWFpQSD2hpGPQ68y4untF0ao,482
+playbeats-0.1.1.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+playbeats-0.1.1.dist-info/entry_points.txt,sha256=e5rsUXvpD9kABydfwafJILptTKE7odVGVC0I5ezR4HA,44
+playbeats-0.1.1.dist-info/top_level.txt,sha256=C0KbUb1jYk_Q_-z5ac6GHhWdpF2XJjZhx6Q1ccHwbQY,10
+playbeats-0.1.1.dist-info/RECORD,,

playbeats-0.1.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (84.2.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

playbeats-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+playbeats = src.main:main

playbeats-0.1.1.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+src
+train

src/main.py

@@ -0,0 +1,112 @@
+import pygame
+import time
+from train import BeatGenerator
+import torch
+import random
+import sys
+import re
+
+from pathlib import Path
+
+ASSETS = Path(__file__).resolve().parent / "assets"
+
+def shuffel(sounds:list):
+    for i in range(len(sounds) -1,0,-1):
+        j = random.randint(0,i)
+        sounds[i], sounds[j] = sounds[j], sounds[i]
+    return sounds
+
+
+def main(): 
+
+    if len(sys.argv) < 3:
+        print("Error: Missing arguments.", file=sys.stderr)
+        print(f"Usage: playbeats  {sys.argv[0]} bpm energy")
+        sys.exit(1)
+    
+    
+    match1 = re.search(r"\d+", sys.argv[1]) 
+    
+    match2 = re.search(r"\d+", sys.argv[2])
+
+    if match1:
+        bpm = float(match1.group())
+    if match2:
+        energy = float(match2.group())
+    
+    duration = float(random.randint(100,1000))
+
+    model = BeatGenerator()
+
+    state_dict = torch.load(
+        ASSETS / "beat_generator.pth",
+        map_location="cpu",
+        weights_only=True
+    )
+
+
+    model.load_state_dict(state_dict)
+    
+    model.eval()
+
+    x = torch.tensor([[bpm, duration, 44100, energy]])
+
+    with torch.no_grad():
+            output = model(x)
+            beat = (torch.sigmoid(output) >= 0.5).int()
+
+    pygame.mixer.init()
+
+
+    sounds_dir = ASSETS / "sounds"
+
+    kick = pygame.mixer.Sound(str(sounds_dir / "KICK.wav"))
+    snare = pygame.mixer.Sound(str(sounds_dir / "SNARE.wav"))
+    sample = pygame.mixer.Sound(str(sounds_dir / "SAMPLE.wav"))
+    hihat = pygame.mixer.Sound(str(sounds_dir / "HIHAT.wav"))
+    bass = pygame.mixer.Sound(str(sounds_dir / "bass.wav"))
+    tom_mid = pygame.mixer.Sound(str(sounds_dir / "tom-mid.wav"))
+    tom_high = pygame.mixer.Sound(str(sounds_dir / "tom-high.wav"))
+    tom_low = pygame.mixer.Sound(str(sounds_dir / "tom-low.wav"))
+    closed_hithat = pygame.mixer.Sound(str(sounds_dir / "closed-hihat.wav"))
+    open_hihat = pygame.mixer.Sound(str(sounds_dir / "openhihat.wav"))
+    tambourine = pygame.mixer.Sound(str(sounds_dir / "tambourine.wav"))
+    bongo = pygame.mixer.Sound(str(sounds_dir / "bongo.wav"))
+    clap = pygame.mixer.Sound(str(sounds_dir / "clap.wav"))
+
+    instruments = [
+        bass,          
+        snare,        
+        tom_mid,      
+        clap,          
+        kick,          
+        open_hihat,    
+        closed_hithat, 
+        tom_high,      
+        hihat,         
+        tambourine,    
+        tom_low,        
+        sample,    
+        bongo,        
+    ]
+
+    bpm = 120
+    beat_time = 60 / bpm
+
+    try:
+        while True:
+            for i in range(64):
+                for j in range(13):
+                    if beat[0][i][j].item() == 1:
+                        instruments = shuffel(instruments)
+                        instruments[j].play()
+                time.sleep(beat_time)
+
+    except KeyboardInterrupt:
+        print("Beat playback stopped.")
+
+    finally:
+        pygame.mixer.quit()
+
+if __name__ == "__main__":
+    main()

train.py

@@ -0,0 +1,109 @@
+# -*- coding: utf-8 -*-
+"""Untitled1.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1EBzJZoQPsyh-XQYCCsFwpnd93PScSpSq
+"""
+
+# from pathlib import Path
+# import glob
+# import json
+import torch.nn as nn
+
+# path = Path("/content/drive/MyDrive/training-data")
+
+# training_data = glob.glob(str(path / "*.json"))
+
+# training_tensors = []
+
+# target_tensors = []
+
+# for file in training_data:
+#     with open(file, "r") as f:
+#       data = json.load(f)
+
+#       metadata = data["metadata"]
+
+#       meta_data = [
+#             metadata["bpm"],
+#             metadata["duration"],
+#             metadata["sample_rate"],
+#             metadata["energy"]
+#         ]
+
+#     training_tensors.append(
+#         torch.tensor(meta_data, dtype=torch.float32)
+#     )
+
+#     target_tensors.append(
+#         torch.tensor(data["sequence"], dtype=torch.float32)
+#     )
+
+# X = torch.stack(training_tensors)
+# Y = torch.stack(target_tensors)
+
+class BeatGenerator(nn.Module):
+    def __init__(self):
+        super().__init__()
+
+        self.network = nn.Sequential(
+            nn.Linear(4, 128),
+            nn.ReLU(),
+
+            nn.Linear(128, 256),
+            nn.ReLU(),
+
+            nn.Linear(256, 832)
+        )
+
+    def forward(self, x):
+        x = self.network(x)
+        x = x.reshape(-1, 64, 13)
+        return x
+
+# model = BeatGenerator()
+
+# positive = Y.sum()
+
+# negative = Y.numel() - positive
+
+# loss_fc = nn.BCEWithLogitsLoss(pos_weight=negative/positive)
+
+# optimizer = torch.optim.Adam(model.parameters(), lr=0.001)
+
+# for epoch in range(20000):
+
+#   output = model(X)
+
+#   loss = loss_fc(output, Y)
+
+#   optimizer.zero_grad()
+
+#   loss.backward()
+
+#   optimizer.step()
+
+#   if (epoch + 1) % 100 == 0:
+#     print(f"Epoch {epoch + 1}, Loss: {loss.item():.4f}")
+
+
+# torch.save(model.state_dict(), "beat_generator.pth")
+
+# model = BeatGenerator()
+
+# model.load_state_dict(
+#     torch.load("beat_generator.pth", map_location="cpu", weights_only=True)
+# )
+
+# model.eval()
+
+# x = torch.tensor([[13230.0, 13235.89, 250.0, 0.12332411]])
+
+# with torch.no_grad():
+#     output = model(x)
+#     beat = (torch.sigmoid(output) >= 0.5).int()
+
+# print("Shape:", beat.shape)
+# print(beat[0])