SetupDatabase.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "f3302fad",
   "metadata": {},
   "source": [
    "# Setup Database\n",
    "\n",
    "Pulling some random music off of my computer and computing fingerprints for it"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "06646215",
   "metadata": {},
   "outputs": [],
   "source": [
    "%load_ext autoreload\n",
    "%autoreload 2\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "from scipy.io import wavfile\n",
    "from dsp import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "0cf4dc70",
   "metadata": {},
   "outputs": [],
   "source": [
    "# STFT Parameters\n",
    "win = 1024\n",
    "hop = 512\n",
    "\n",
    "# Constellation Parameters\n",
    "max_freq = 128\n",
    "freq_win = 8\n",
    "time_win = 6\n",
    "thresh = 1e3\n",
    "\n",
    "# Fingerprint Parameters\n",
    "d_center = 86\n",
    "width = 50\n",
    "height = 21"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "07d49a65",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "ABC Poison Arrow (4325 points, 61592 hashes)\n",
      "Alphaville Forever Young (5043 points, 78166 hashes)\n",
      "Animotion Obsession (5222 points, 72664 hashes)\n",
      "Bananarama Cruel Summer (4281 points, 57912 hashes)\n",
      "Chaz Jankel Number One (5382 points, 84254 hashes)\n",
      "Corey Hart Sunglasses At Night (5181 points, 77275 hashes)\n",
      "Duran Duran Girls On Film (4577 points, 64764 hashes)\n",
      "Duran Duran Notorious (4507 points, 58287 hashes)\n",
      "Duran Duran Rio (7094 points, 99696 hashes)\n",
      "Go West Call Me (5038 points, 68258 hashes)\n",
      "Human League Don't You Want Me (4689 points, 62946 hashes)\n",
      "Kenny Loggins Danger Zone (4462 points, 61935 hashes)\n",
      "Mr Mister Broken Wings (5870 points, 83022 hashes)\n",
      "Mr Mister Kyrie (5421 points, 75749 hashes)\n",
      "Oingo Boingo Just Another Day (7021 points, 102307 hashes)\n",
      "Peter Gabriel Big Time (5259 points, 71176 hashes)\n",
      "Reflex Politics of Dancing (4973 points, 71346 hashes)\n",
      "Squeeze Tempted (4609 points, 58938 hashes)\n",
      "Talk Talk Life's What You Make It (4679 points, 67345 hashes)\n",
      "Tears for Fears Pale Shelter (6188 points, 95469 hashes)\n",
      "Toto Africa (5237 points, 69999 hashes)\n",
      "Wang Chung Dance Hall Days (4862 points, 66252 hashes)\n",
      "Wang Chung Everybody Have Fun Tonight (5928 points, 80725 hashes)\n",
      "Body Language FallingOut (6023 points, 72581 hashes)\n",
      "Vaughan Williams The Wasps (10994 points, 152497 hashes)\n",
      "Alanis Morissette Ironic (4133 points, 51803 hashes)\n",
      "Jennifer Lopez Im Real (5673 points, 71942 hashes)\n",
      "Katrina and the waves Walking on Sunshine (4711 points, 63985 hashes)\n",
      "Kevin Lyttle I Got It (4431 points, 59982 hashes)\n",
      "Lipps Inc Funkytown (10051 points, 142187 hashes)\n",
      "LollieVox IllBeYourMadonna (3006 points, 39174 hashes)\n",
      "Luis Fonsi ft Daddy Yankee Despacito (5652 points, 78511 hashes)\n",
      "Marky Mark Good Vibrations (5322 points, 68152 hashes)\n",
      "OZone Dragostea Din Tei (4509 points, 63777 hashes)\n",
      "Pilot Oh Ho Ho Its Magic (3827 points, 53942 hashes)\n",
      "Power Station Some Like It Hot (6037 points, 82389 hashes)\n",
      "Bel Biv DeVoe Poison (5084 points, 63972 hashes)\n",
      "C and C Music Factory Things That Make You Go (6357 points, 84697 hashes)\n",
      "Cher Believe (4654 points, 63509 hashes)\n",
      "Coldplay Strawberry Swing (5075 points, 71968 hashes)\n",
      "Cure Just Like Heaven (4675 points, 66437 hashes)\n",
      "Daft Punk Short Circuit (3205 points, 36170 hashes)\n",
      "Deep Purple Hush (5292 points, 70025 hashes)\n",
      "Duck Sauce Barbara Streisand (3702 points, 46584 hashes)\n",
      "Ed Sheeran Beautiful People (3785 points, 46310 hashes)\n",
      "ELO Evil Woman (5775 points, 87355 hashes)\n",
      "ELO Strange Magic (4914 points, 66276 hashes)\n",
      "Enrique Iglesias Bailamos (4264 points, 56427 hashes)\n",
      "Fat Boys The Fat Boys Are Back (6880 points, 85771 hashes)\n",
      "Seal Fly Like an Eagle (5315 points, 73698 hashes)\n",
      "Sting Brand New Day (7808 points, 109486 hashes)\n",
      "Sting Desert Rose (5764 points, 77681 hashes)\n",
      "Sting If I Ever Lose My Faith (5929 points, 89728 hashes)\n",
      "The Proclaimers Im Gonna Be (4101 points, 50900 hashes)\n",
      "The Turtles Happy Together (3481 points, 48086 hashes)\n",
      "The Waterboys Whole of the Moon (5908 points, 79308 hashes)\n",
      "Tom Jones Its Not Unusual (2373 points, 31672 hashes)\n",
      "UB40 Rat in The Kitchen (8499 points, 116076 hashes)\n",
      "Uffie The Party (3035 points, 43498 hashes)\n",
      "Weird Al White and nerdy (3226 points, 40221 hashes)\n",
      "Pure Pwnage Im a gamer (1369 points, 15444 hashes)\n",
      "Bee Gees Stayin Alive (4445 points, 54728 hashes)\n",
      "Bobby Caldwell What You Wont Do For Love (5960 points, 85163 hashes)\n",
      "Cameo Candy (6809 points, 89608 hashes)\n",
      "Cameo Shake Your Pants (7507 points, 98436 hashes)\n",
      "Cameo Strange (8076 points, 102219 hashes)\n",
      "DeBarge I Like It (5281 points, 66803 hashes)\n",
      "Earth Wind and Fire After the Love Has Gone (5312 points, 73396 hashes)\n",
      "Earth Wind and Fire Cant Hide Love (4917 points, 64390 hashes)\n",
      "Earth Wind And Fire Evil (6189 points, 83698 hashes)\n",
      "Earth Wind and Fire Fantasy (5860 points, 81583 hashes)\n",
      "Earth Wind and Fire Gotta Get You Into My Life (4709 points, 61328 hashes)\n",
      "Earth Wind and Fire Saturday Night (5001 points, 69419 hashes)\n",
      "Earth Wind and Fire Thats the way of the world (6626 points, 84590 hashes)\n",
      "Eath Wind and Fire Shining Star (3367 points, 44341 hashes)\n",
      "Prince Pop Life (4199 points, 53989 hashes)\n",
      "Prince Why You Wanna Treat Me So Bad (4480 points, 60648 hashes)\n",
      "Rick James Cold Blooded (7225 points, 94748 hashes)\n",
      "Rick James Ghetto Life (5322 points, 72566 hashes)\n",
      "Rick James Give it to me baby (4646 points, 59356 hashes)\n",
      "Rick James Maryjane (4381 points, 52777 hashes)\n",
      "Rick James You and I (9219 points, 118301 hashes)\n",
      "Rick James You Turn Me On (5436 points, 69386 hashes)\n",
      "Sister Sledge Good Times (4760 points, 59753 hashes)\n",
      "Stevie Wonder Do I Do (6060 points, 79838 hashes)\n",
      "Surface Happy (4407 points, 56824 hashes)\n",
      "Tavares It Only Takes a Minute (4938 points, 68900 hashes)\n",
      "The Trammps Disco Inferno (4511 points, 61306 hashes)\n",
      "The Tubes Tip of My Tongue (4726 points, 62130 hashes)\n",
      "TLC Creep (4616 points, 53085 hashes)\n",
      "Yarbrough and Peoples Don't Stop (4727 points, 59994 hashes)\n",
      "Zapp and Roger Do Wa Diddy (5723 points, 72759 hashes)\n",
      "Evelyn King Love Come Down (6902 points, 84214 hashes)\n",
      "Prince My Love is Forever (5146 points, 70488 hashes)\n",
      "Frankie Smith Double Dutch Bus (6154 points, 76828 hashes)\n",
      "George Benson Turn Your Love Around (4846 points, 66703 hashes)\n",
      "Gino Vannelli I Just Wanna Stop (4973 points, 67279 hashes)\n",
      "Janet Jackson Escapade (6235 points, 91268 hashes)\n",
      "Janet Jackson Love Would Never Do Without You (6759 points, 85090 hashes)\n",
      "Janet Jackson Miss You Much (4647 points, 56208 hashes)\n",
      "Junior Mama Used to Say (4453 points, 64258 hashes)\n",
      "Ohio Players Love Rollercoaster (3390 points, 42780 hashes)\n",
      "Oliver Chetham Get Down Saturday Night (7634 points, 93673 hashes)\n",
      "Prince 1999 (6755 points, 82051 hashes)\n",
      "Prince Controversy (4192 points, 54228 hashes)\n",
      "Prince Head (5339 points, 65583 hashes)\n",
      "Prince Let's Work (4563 points, 56916 hashes)\n",
      "Grateful Dead Friend of The Devil (4177 points, 56758 hashes)\n",
      "Grateful Dead Scarlet Begonias (7009 points, 93687 hashes)\n",
      "Grateful Dead Sugaree 1972 (Studio Version) (6824 points, 85385 hashes)\n",
      "Grateful Dead Terrapin Station (18238 points, 245646 hashes)\n",
      "Grateful Dead Touch of Grey (7524 points, 105166 hashes)\n",
      "Grateful Deadl Althea (8339 points, 108930 hashes)\n",
      "Hudson Mohawke DeDeMouse (5125 points, 70094 hashes)\n",
      "2pac Do For Love (5600 points, 75230 hashes)\n",
      "50 Cent 21 Questions (4019 points, 47424 hashes)\n",
      "Beyonce Be Alive (3961 points, 46614 hashes)\n",
      "Beyonce Single Ladies (3685 points, 46795 hashes)\n",
      "Beyonce Sorry (3976 points, 52439 hashes)\n",
      "Beyonce Upgrade U (5915 points, 84524 hashes)\n",
      "Future Life Is Good (3649 points, 37762 hashes)\n",
      "Kanye West ft T Pain The Good Life (4147 points, 54915 hashes)\n",
      "Kanye West Touch the Sky (4752 points, 62850 hashes)\n",
      "Nicki Minaj Superbass (4333 points, 63441 hashes)\n",
      "Outcast Ms Jackson (5311 points, 70557 hashes)\n",
      "Outcast So fresh so clean (4642 points, 56436 hashes)\n",
      "Outkast Roses (7060 points, 84674 hashes)\n",
      "Outkast The Way You Move (4497 points, 58058 hashes)\n",
      "Sean Paul Temperature (4785 points, 74884 hashes)\n",
      "Snoop Dogg Signs (4818 points, 64043 hashes)\n",
      "TPain Bartender (4601 points, 59467 hashes)\n",
      "TPain Buy You a Drank (4269 points, 53702 hashes)\n",
      "Yung Joc Coffee Shop (4669 points, 59355 hashes)\n",
      "Michael Jackson Dirty Diana (5783 points, 78120 hashes)\n",
      "Michael Jackson Ghosts (6284 points, 84591 hashes)\n",
      "Michael Jackson Human Nature (5090 points, 69935 hashes)\n",
      "Michael Jackson Lady In My Life (6050 points, 81252 hashes)\n",
      "Michael Jackson Man In The Mirror (6114 points, 79669 hashes)\n",
      "Michael Jackson Money (5194 points, 64154 hashes)\n",
      "Michael Jackson Off The Wall (4929 points, 64927 hashes)\n",
      "Michael Jackson PYT (5033 points, 71354 hashes)\n",
      "Michael Jackson Remember The Time (4736 points, 63010 hashes)\n",
      "Michael Jackson Rock With You (4145 points, 51280 hashes)\n",
      "Michael Jackson Smooth Criminal (5289 points, 70586 hashes)\n",
      "Michael Jackson The Way You Make Me Feel (6267 points, 88265 hashes)\n",
      "Michael Jackson Thriller (7059 points, 93938 hashes)\n",
      "Michael Jackson You Rock My World (4848 points, 58556 hashes)\n",
      "Michael Jackson I Just Can't Stop Loving You (5283 points, 72824 hashes)\n",
      "Michael Jackson Don't Stop Till You Get Enough (7318 points, 98066 hashes)\n",
      "The Jacksons I Wanna Be Where You Are (3016 points, 35847 hashes)\n",
      "The Jacksons Walk Right Now (8190 points, 118647 hashes)\n",
      "Michael Jackson & Paul Mcartney Say Say Say (4806 points, 64325 hashes)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Michael Jackson & The Jackson Five My Girl (3284 points, 38728 hashes)\n",
      "Michael Jackson Wanna Be Starting Something (7217 points, 94558 hashes)\n",
      "Michael Jackson Another Part Of Me (4493 points, 57889 hashes)\n",
      "Michael Jackson Bad (4826 points, 61104 hashes)\n",
      "Michael Jackson Beat It (5099 points, 69381 hashes)\n",
      "Michael Jackson Billy Jean (6073 points, 84533 hashes)\n",
      "Michael Jackson Black or White (5318 points, 75669 hashes)\n",
      "Michael Jackson Dangerous (7789 points, 91450 hashes)\n",
      "ACDC Back In Black (4672 points, 56604 hashes)\n",
      "Aerosmith Duke Looks Like A lady (5397 points, 74920 hashes)\n",
      "Goldfinger Superman (4078 points, 57945 hashes)\n",
      "Guns N Roses Welcome To the Jungle (5643 points, 78378 hashes)\n",
      "Jimi Hendrix Purple Haze (3815 points, 45710 hashes)\n",
      "Led Zeppelin Nobody's Fault But Mine (7445 points, 96156 hashes)\n",
      "Led Zeppelin Stairway To Heaven (9708 points, 134056 hashes)\n",
      "Led Zepplin All of My Love (6689 points, 87169 hashes)\n",
      "LIT My Own Worst Enemy (3655 points, 52537 hashes)\n",
      "Lynyrd Skynyrd Tuesday's Gone (9479 points, 132080 hashes)\n",
      "Metallica Enter Sandman (7076 points, 101021 hashes)\n",
      "Pink Floyd Learning To Fly (5780 points, 76668 hashes)\n",
      "Rolling Stones You Cant Always Get What You Want (5684 points, 76273 hashes)\n",
      "Stray Cats Rock This Town (4101 points, 54981 hashes)\n",
      "Sugar Ray Every Morning (4475 points, 59428 hashes)\n",
      "Sugar Ray I Just Wanna Fly (5942 points, 80085 hashes)\n",
      "Supertramp Breakfast in America (3100 points, 41385 hashes)\n",
      "Supertramp Give a Little Bit (4996 points, 66650 hashes)\n",
      "Taking Back Sunday A Decade Under the Influence (5467 points, 79627 hashes)\n",
      "The All American Rejects Dirty Little Secret (4303 points, 63504 hashes)\n",
      "The Black Keys Strange Times (3688 points, 46804 hashes)\n",
      "The Fray How To Save A Life (5164 points, 68474 hashes)\n",
      "C&C Music Factory Make You Sweat (Everybody Dance Now) (7188 points, 82270 hashes)\n",
      "Clapton Change the World (4475 points, 57866 hashes)\n",
      "Counting Crows Mr (5864 points, 83950 hashes)\n",
      "Counting Crows Omaha (4537 points, 63542 hashes)\n",
      "Counting Crows Rain King (5591 points, 81728 hashes)\n",
      "Daniel Powter Bad Day (4548 points, 60875 hashes)\n",
      "Dave Matthews Band Ants Marching (5458 points, 72645 hashes)\n",
      "Dave Matthews Band Satellite (5691 points, 77271 hashes)\n",
      "Dave Matthews Band What Would You Say (4521 points, 59684 hashes)\n",
      "Foo Fighters Best of You (5833 points, 87626 hashes)\n",
      "Foo Fighters Learn to Fly (5266 points, 78154 hashes)\n",
      "Foo Figthers Times Like These (5138 points, 75154 hashes)\n",
      "Fu Manchu Thinkin Out Loud (4489 points, 63244 hashes)\n",
      "The Police Driven To Tears (4212 points, 57594 hashes)\n",
      "The Police Every Little Thing She Does is Magic (5943 points, 87318 hashes)\n",
      "The Police Invisible Sun (4393 points, 57508 hashes)\n",
      "The Police King of Pain (6488 points, 95055 hashes)\n",
      "The Police When the World is Running Down (4390 points, 61158 hashes)\n",
      "The Who Teenage Wasteland (6068 points, 79182 hashes)\n",
      "The Who Won't Get Fooled Again (7496 points, 102876 hashes)\n",
      "Third Eye Blind Never Let You Go (4939 points, 68192 hashes)\n",
      "Third Eye Blind Semicharmed life (5293 points, 77448 hashes)\n",
      "Todd Rundgren Bang The Drum All Day (4186 points, 52817 hashes)\n",
      "Tom Petty Free Fallin' (4933 points, 66569 hashes)\n",
      "Tom Petty I Won't Back Down (3790 points, 53466 hashes)\n",
      "Tom Petty Yer So Bad (3717 points, 51555 hashes)\n",
      "Transplants Tall Cans in the Air (4835 points, 70829 hashes)\n",
      "U2 Beautiful Day (5409 points, 78777 hashes)\n",
      "U2 Mysterious Ways (4621 points, 58720 hashes)\n",
      "Van Halen Jump (5286 points, 78236 hashes)\n",
      "Wang Chung Everybody Have Fun Tonight (5857 points, 80661 hashes)\n",
      "Yes Owner of a Lonely Heart (5577 points, 79099 hashes)\n",
      "Glenn Frey The Heat Is On (4837 points, 67432 hashes)\n",
      "The Killers All These Things That I've Done (5932 points, 77706 hashes)\n",
      "Alien Ant Farm Smooth Criminal (4184 points, 55913 hashes)\n",
      "Alien Ant Farm These Days (3937 points, 55416 hashes)\n",
      "Bachman Turner Overdrive You Ain't Seen Nothing Yet (4842 points, 63975 hashes)\n",
      "Bad Religion You (2899 points, 44390 hashes)\n",
      "Barenaked Ladies Pinch Me (5912 points, 80549 hashes)\n",
      "Blind Melon No Rain (4269 points, 55876 hashes)\n",
      "Don Ellis Gobbledigook (3913 points, 55108 hashes)\n",
      "Don Ellis Pussy Wiggle Stomp (Album Version) (7833 points, 105045 hashes)\n",
      "Luther Vandross Can I Take You Out Tonight (4039 points, 47276 hashes)\n",
      "Acoustic Alchemy Same road same reason (6526 points, 96161 hashes)\n",
      "Boney James 201 AM (6343 points, 89158 hashes)\n",
      "Chris Botti Indian Summer (4180 points, 53447 hashes)\n",
      "Hall and Oates I'll be around (5004 points, 68151 hashes)\n",
      "Jeff Kashiwa Ooh Ah (5200 points, 70447 hashes)\n",
      "Juewett Bostick Elsi N Me (4406 points, 56369 hashes)\n",
      "Lionel Ritchie You Are The Sun (5708 points, 70755 hashes)\n",
      "Luther Vandross Shine (4574 points, 59597 hashes)\n",
      "Marvin Gaye Me and Mrs (5261 points, 75622 hashes)\n",
      "Marvin Gaye Mercy Mercy Me (4031 points, 56600 hashes)\n",
      "Marvin Gaye Sexual Healing (5029 points, 66008 hashes)\n",
      "Marvine Gaye Lets get it on (4448 points, 55205 hashes)\n",
      "Nils Catnap (5363 points, 63708 hashes)\n",
      "Nils Pacific Coast Highway (5709 points, 70675 hashes)\n",
      "Norman Brown After The Storm (6150 points, 88373 hashes)\n",
      "Norman Brown Stay With Me (4075 points, 51501 hashes)\n",
      "Norman Brown Up N At Em (4768 points, 63898 hashes)\n",
      "Phil Collins Come with me (5732 points, 79656 hashes)\n",
      "Ray Charles Oh What a Beautiful Morning (3532 points, 42086 hashes)\n",
      "Simply Red Sunrise (3984 points, 53690 hashes)\n",
      "Steve Cole Got It Goin On (4713 points, 62242 hashes)\n",
      "The Four Tops Ain't No Woman (3436 points, 43239 hashes)\n",
      "The rippingtons aspen (6623 points, 89407 hashes)\n",
      "Blind Mellon No Cigar (3709 points, 55208 hashes)\n",
      "Speedealer Screamer (2561 points, 32926 hashes)\n",
      "The Ernies Here and Now (4587 points, 62428 hashes)\n",
      "Alkaline Trio Armageddon (3539 points, 48119 hashes)\n",
      "Anacron A Prototype (3071 points, 38521 hashes)\n",
      "Angry Amputees She Said (1997 points, 28233 hashes)\n",
      "Authority Zero Everyday (3241 points, 46287 hashes)\n",
      "Clutch Impetus (4232 points, 57649 hashes)\n",
      "Disturbed Liberate (4102 points, 53016 hashes)\n",
      "Dont Wait Blue Collar special (3761 points, 54266 hashes)\n",
      "Frog One Blah Blah (3491 points, 42984 hashes)\n",
      "Fu Manchu California Crossing (4510 points, 60228 hashes)\n",
      "GBH Crush Em (3642 points, 47938 hashes)\n",
      "Hot Water Music Remedy (3611 points, 56412 hashes)\n",
      "Jane's Addiction Suffer Some (5064 points, 66270 hashes)\n",
      "Kiss Rock and Roll All Nite (3560 points, 48458 hashes)\n",
      "LA Symphony King Kong (4065 points, 50549 hashes)\n",
      "Mike V and the Rats The Days (3902 points, 53818 hashes)\n",
      "Mr Lif Phantom (5353 points, 75204 hashes)\n",
      "Nas The World is Yours (4927 points, 68189 hashes)\n",
      "NoFX Separation of Church and Skate (3754 points, 51502 hashes)\n",
      "Quasimoto Low Class Conspiracy (2765 points, 35979 hashes)\n",
      "Rise Against Like the Angel (3263 points, 43352 hashes)\n",
      "Rubber City Rebels I Wanna Pierce My Brain (2503 points, 34500 hashes)\n",
      "Stiff Little Fingers Suspect Device (3642 points, 52385 hashes)\n",
      "Supernatural Internationally Known (4221 points, 59305 hashes)\n",
      "The Distillers Beat Your Heart Out (3616 points, 50411 hashes)\n",
      "Transplants California Babylon (5127 points, 70952 hashes)\n",
      "Wild Child Secondary Protocol (4388 points, 58879 hashes)\n"
     ]
    }
   ],
   "source": [
    "import subprocess\n",
    "import glob\n",
    "import os\n",
    "import numpy as np\n",
    "import pickle\n",
    "\n",
    "data = [] # List of {\"artist\":string, \"title\":string, \"fingerprints\":(hashes, offsets)}\n",
    "\n",
    "for idx, f in enumerate(glob.glob(\"Music/*/*.mp3\")):\n",
    "    name = f.split(\"/\")[-1].split(\".\")[0]\n",
    "    parts = name.split(\"-\")\n",
    "    if len(parts) != 2:\n",
    "        continue\n",
    "    [artist, title] = parts\n",
    "    artist = artist.lstrip().rstrip()\n",
    "    title = title.lstrip().rstrip()\n",
    "    \n",
    "    ## Step 0: Convert to 22050hz sampling and mono, then load audio\n",
    "    if os.path.exists(\"temp.wav\"):\n",
    "        os.remove(\"temp.wav\")\n",
    "    subprocess.call([\"ffmpeg\", \"-i\", f, \"-ar\", \"22050\", \"-ac\", \"1\", \"temp.wav\"], \n",
    "                    stdout=subprocess.DEVNULL,\n",
    "                    stderr=subprocess.STDOUT)\n",
    "    sr, x = wavfile.read(\"temp.wav\")\n",
    "    ## Step 1: Compute STFT\n",
    "    S = np.abs(stft(x, win, hop))\n",
    "    ## Step 2: Get constellations\n",
    "    I, J = get_constellation(S, freq_win, time_win, max_freq, thresh)\n",
    "    ## Step 3: Get fingerprints\n",
    "    hashes, offsets = get_fingerprints(I, J, width, height, d_center)\n",
    "    print(artist, title, \"({} points, {} hashes)\".format(I.size, hashes.size))\n",
    "    ## Step 4: Save data\n",
    "    data.append({\"artist\":artist, \"title\":title, \"fingerprints\":(hashes, offsets)})\n",
    "    if idx%20 == 0:\n",
    "        pickle.dump(data, open(\"database.pkl\", \"wb\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "11ce1db7",
   "metadata": {},
   "outputs": [],
   "source": [
    "pickle.dump(data, open(\"database.pkl\", \"wb\"))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "f19f18a2",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "277\n"
     ]
    }
   ],
   "source": [
    "print(len(data))"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.13"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}