latest

.gitignore

@@ -2,4 +2,5 @@
 .env
 .cache
 __pycache__
-data/audio/*.wav
+data/audio/*.wav
+data_2

app.py

@@ -9,11 +9,10 @@ import pandas as pd
 import json
 import os
 import smtplib, ssl
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
-from email.utils import COMMASPACE, formatdate
+from dotenv import load_dotenv
 st.set_page_config(page_title="Curate me a playlist", layout="wide")
 
+load_dotenv()
 
 @st.cache_data
 def load_data():
@@ -43,10 +42,17 @@ def send_curator(text):
     sender_email = os.getenv("EMAIL_ADDRESS")
     receiver_email = os.getenv("EMAIL_RECEIVER")
     password = os.getenv("EMAIL_PASSWORD")
+    from email.mime.multipart import MIMEMultipart
+    from email.mime.text import MIMEText
+
+    msg = MIMEMultipart("alternative")
+    msg["Subject"] = "Curate me a playlist submission"
+    part1 = MIMEText(body, "plain")
+    msg.attach(part1)
     context = ssl.create_default_context()
     with smtplib.SMTP_SSL(smtp_server, port, context=context) as server:
         server.login(sender_email, password)
-        server.sendmail(sender_email, receiver_email, text)
+        server.sendmail(sender_email, receiver_email, msg)
 
     print("Email sent.")
 
@@ -76,10 +82,10 @@ if is_clicked:
                 # help="The top trending Streamlit apps",
                 # validate="^https://[a-z]+\.streamlit\.app$",
                 # max_chars=100,
-            )
-        },
-    hide_index=False,
-    use_container_width=True
+                    )
+                        },
+        hide_index=False,
+        use_container_width=True
     )
 
     form = st.form("form")
@@ -89,7 +95,7 @@ if is_clicked:
     playlist = [f"{k}\n" for k in dataframe.index]
     playlist_string = "\n".join(dataframe.index.tolist())
     body = f"""\
-    Subject: Hi there
+    Subject: Curate me a playlist submission
 
     Curator --> {sender}
     Query --> {session.text_input}
@@ -98,4 +104,6 @@ if is_clicked:
     {playlist_string}
     """
 
+
+    print(body)
     is_submit = form.form_submit_button("Submit", on_click=send_curator, args=([body]))

data/json/saved_tracks.json

@@ -0,0 +1,2562 @@
+[
+    {
+        "artist": "Coldplay",
+        "track": "Sparks"
+    },
+    {
+        "artist": "America",
+        "track": "A Horse with No Name"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Have A Cigar - 2011 Remaster"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "Bazen"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "The Great Gig In The Sky - 2011 Remastered Version"
+    },
+    {
+        "artist": "Dire Straits",
+        "track": "Once Upon A Time In The West"
+    },
+    {
+        "artist": "Hakk\u0131 Bulut",
+        "track": "Ben K\u00f6yl\u00fcy\u00fcm"
+    },
+    {
+        "artist": "Sting",
+        "track": "Brand New Day"
+    },
+    {
+        "artist": "Pinhani",
+        "track": "D\u00fcnyadan Uzak"
+    },
+    {
+        "artist": "The Corrs",
+        "track": "Little Wing - MTV Unplugged Version"
+    },
+    {
+        "artist": "Chris Isaak",
+        "track": "Wicked Game"
+    },
+    {
+        "artist": "The Police",
+        "track": "Roxanne"
+    },
+    {
+        "artist": "Frank Sark",
+        "track": "Amor Mio"
+    },
+    {
+        "artist": "David Dallas",
+        "track": "Runnin'"
+    },
+    {
+        "artist": "mor ve \u00f6tesi",
+        "track": "Re"
+    },
+    {
+        "artist": "Fl\u00f6rt",
+        "track": "Biz"
+    },
+    {
+        "artist": "Telenova",
+        "track": "Bones"
+    },
+    {
+        "artist": "emir taha",
+        "track": "Huyu Suyu"
+    },
+    {
+        "artist": "mor ve \u00f6tesi",
+        "track": "G\u00fcneye Giderken"
+    },
+    {
+        "artist": "Replikas",
+        "track": "Ya\u015f Elli"
+    },
+    {
+        "artist": "Grover Washington, Jr.",
+        "track": "Just the Two of Us"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Stir It Up"
+    },
+    {
+        "artist": "KISS",
+        "track": "I Was Made For Lovin' You"
+    },
+    {
+        "artist": "Steely Dan",
+        "track": "Do It Again"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "Sak\u0131n Gelme"
+    },
+    {
+        "artist": "MIKA",
+        "track": "Relax, Take It Easy"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "\u00c7ok Yorgunum"
+    },
+    {
+        "artist": "John Mayer",
+        "track": "New Light"
+    },
+    {
+        "artist": "The Weeknd",
+        "track": "Blinding Lights"
+    },
+    {
+        "artist": "Jamelia",
+        "track": "Stop"
+    },
+    {
+        "artist": "Erkut Ta\u00e7k\u0131n",
+        "track": "Sevmek I\u0307stiyorum"
+    },
+    {
+        "artist": "Estelle",
+        "track": "American Boy"
+    },
+    {
+        "artist": "Sting",
+        "track": "Fields Of Gold"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Mavi Ekran"
+    },
+    {
+        "artist": "H\u00fcseyin Bitmez",
+        "track": "Nazende Sevgilim"
+    },
+    {
+        "artist": "Pinhani",
+        "track": "\u0130stanbul'da"
+    },
+    {
+        "artist": "M\u00fcsl\u00fcm G\u00fcrses",
+        "track": "Hayat Berbat"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Brain Damage - 2011 Remastered Version"
+    },
+    {
+        "artist": "Z\u00fclf\u00fc Livaneli",
+        "track": "G\u00f6zlerin"
+    },
+    {
+        "artist": "David Bowie",
+        "track": "Fame - 2016 Remaster"
+    },
+    {
+        "artist": "Fl\u00f6rt",
+        "track": "Lan O\u011flum B\u00f6yle Olmaz"
+    },
+    {
+        "artist": "M\u00fcsl\u00fcm G\u00fcrses",
+        "track": "Affet"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Stairway to Heaven - Remaster"
+    },
+    {
+        "artist": "Y\u00fczy\u00fczeyken Konu\u015furuz",
+        "track": "Dinle Beni Bi'"
+    },
+    {
+        "artist": "Sting",
+        "track": "It's Probably Me"
+    },
+    {
+        "artist": "Stromae",
+        "track": "Tous les m\u00eames"
+    },
+    {
+        "artist": "Steely Dan",
+        "track": "Kid Charlemagne"
+    },
+    {
+        "artist": "Ali K\u0131z\u0131ltu\u011f",
+        "track": "Sen Gel Diyorsun - \u00d6f \u00d6f"
+    },
+    {
+        "artist": "The Doors",
+        "track": "People Are Strange"
+    },
+    {
+        "artist": "Tame Impala",
+        "track": "The Less I Know The Better"
+    },
+    {
+        "artist": "Nur Yoldas",
+        "track": "Sultan-\u0131 Yegah"
+    },
+    {
+        "artist": "Okay Temiz",
+        "track": "East Breeze"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Saat 03:00"
+    },
+    {
+        "artist": "No Land",
+        "track": "Niy\u0259 Bel\u0259 Uzundur Bu Yollar"
+    },
+    {
+        "artist": "Okan Kaya",
+        "track": "Katip Arzuhalim"
+    },
+    {
+        "artist": "Queen",
+        "track": "Who Wants To Live Forever - Remastered 2011"
+    },
+    {
+        "artist": "Erkin Koray",
+        "track": "Hayat Katar\u0131"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Dadalo\u011flu"
+    },
+    {
+        "artist": "The Derek Trucks Band",
+        "track": "Crow Jane"
+    },
+    {
+        "artist": "Y\u00fczy\u00fczeyken Konu\u015furuz",
+        "track": "Ne Farkeder"
+    },
+    {
+        "artist": "Loredana Bert\u00e8",
+        "track": "J'adore Venise"
+    },
+    {
+        "artist": "B\u00fcy\u00fck Ev Ablukada",
+        "track": "HO\u015e\u00c7AKAL KADAR"
+    },
+    {
+        "artist": "Dario Moreno",
+        "track": "Her Ak\u015fam \"Sarho\u015f\""
+    },
+    {
+        "artist": "B\u00fcy\u00fck Ev Ablukada",
+        "track": "En G\u00fczel Yerinde Evin"
+    },
+    {
+        "artist": "Talking Heads",
+        "track": "Psycho Killer - 2005 Remaster"
+    },
+    {
+        "artist": "Black Sabbath",
+        "track": "Planet Caravan"
+    },
+    {
+        "artist": "\u00d6zdemir Erdo\u011fan",
+        "track": "Gurbet"
+    },
+    {
+        "artist": "Edwin Starr",
+        "track": "War"
+    },
+    {
+        "artist": "Boney M.",
+        "track": "Rasputin - Live"
+    },
+    {
+        "artist": "Pinhani",
+        "track": "Ben Nas\u0131 B\u00fcy\u00fck Adam Olucam"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "River of Tears - Live at Staples Center, Los Angeles, CA, 8/18 - 19/2001"
+    },
+    {
+        "artist": "\u00dcmit Say\u0131n",
+        "track": "Hicran"
+    },
+    {
+        "artist": "Sufle",
+        "track": "K\u00f6pr\u00fcalt\u0131"
+    },
+    {
+        "artist": "Duman",
+        "track": "Seni Kendime Saklad\u0131m"
+    },
+    {
+        "artist": "Ray LaMontagne",
+        "track": "You Are the Best Thing"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Sar\u0131l\u0131r\u0131m Birine"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Zombi"
+    },
+    {
+        "artist": "Jimi Hendrix",
+        "track": "Who Knows - Live At Fillmore East, 1970 / 50th Anniversary"
+    },
+    {
+        "artist": "Queen",
+        "track": "These Are The Days Of Our Lives"
+    },
+    {
+        "artist": "Argent",
+        "track": "Lothlorien"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Hayalperest"
+    },
+    {
+        "artist": "Herbie Hancock",
+        "track": "Cantaloupe Island - Remastered 1999/Rudy Van Gelder Edition"
+    },
+    {
+        "artist": "Jim Croce",
+        "track": "Time In A Bottle"
+    },
+    {
+        "artist": "Paul Desmond",
+        "track": "Autumn Leaves"
+    },
+    {
+        "artist": "Queen",
+        "track": "Radio Ga Ga - Remastered 2011"
+    },
+    {
+        "artist": "Commodores",
+        "track": "Easy"
+    },
+    {
+        "artist": "Buena Vista Social Club",
+        "track": "Tiene Sabor"
+    },
+    {
+        "artist": "Red Hot Chili Peppers",
+        "track": "By the Way"
+    },
+    {
+        "artist": "Andrea Motis",
+        "track": "Dream a Little Dream of Me"
+    },
+    {
+        "artist": "Brad Mehldau",
+        "track": "Hey Joe"
+    },
+    {
+        "artist": "Santana",
+        "track": "Maria Maria (feat. The Product G&B)"
+    },
+    {
+        "artist": "M\u00fcsl\u00fcm G\u00fcrses",
+        "track": "Bir \u00d6m\u00fcr Yetmez"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Immigrant Song - 1990 Remaster"
+    },
+    {
+        "artist": "Mert F\u0131rat",
+        "track": "Bana G\u00f6re De\u011fil"
+    },
+    {
+        "artist": "Bar\u0131\u015f Man\u00e7o",
+        "track": "D\u00f6nence"
+    },
+    {
+        "artist": "Nirvana",
+        "track": "The Man Who Sold The World"
+    },
+    {
+        "artist": "Queen",
+        "track": "We Will Rock You - Remastered 2011"
+    },
+    {
+        "artist": "Muse",
+        "track": "Time is Running Out"
+    },
+    {
+        "artist": "Rage Against The Machine",
+        "track": "Killing In The Name"
+    },
+    {
+        "artist": "Buddy Guy",
+        "track": "Mustang Sally (feat. Jeff Beck)"
+    },
+    {
+        "artist": "Sedef Seb\u00fcktekin",
+        "track": "Kayboluyorum (S\u00fct)"
+    },
+    {
+        "artist": "Alice Merton",
+        "track": "No Roots"
+    },
+    {
+        "artist": "Kovacs",
+        "track": "It's the Weekend"
+    },
+    {
+        "artist": "Duman",
+        "track": "K\u00f6pekler"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Nobody Knows You When You're Down and Out - Acoustic; Live at MTV Unplugged, Bray Film Studios, Windsor, England, UK, 1/16/1992; 2013 Remaster"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "Fever"
+    },
+    {
+        "artist": "B\u00fclent Orta\u00e7gil",
+        "track": "Bu Su Hi\u00e7 Durmaz"
+    },
+    {
+        "artist": "Athena",
+        "track": "Ka\u00e7ak"
+    },
+    {
+        "artist": "Dawn Penn",
+        "track": "You Don't Love Me - No No No"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "High Hopes - 2011 Remaster"
+    },
+    {
+        "artist": "Wild Cherry",
+        "track": "Play That Funky Music"
+    },
+    {
+        "artist": "Athena",
+        "track": "Ad\u0131m\u0131z Miskindir Bizim"
+    },
+    {
+        "artist": "B\u00fclent Orta\u00e7gil",
+        "track": "Bozburun"
+    },
+    {
+        "artist": "TOTO",
+        "track": "Hold the Line"
+    },
+    {
+        "artist": "Feyyaz Yigit",
+        "track": "Samsun'lu De\u011filim"
+    },
+    {
+        "artist": "Muddy Waters",
+        "track": "Mannish Boy"
+    },
+    {
+        "artist": "Wes Montgomery",
+        "track": "In Your Own Sweet Way"
+    },
+    {
+        "artist": "Blonde Redhead",
+        "track": "For the Damaged Coda"
+    },
+    {
+        "artist": "Eagles",
+        "track": "Hotel California - 2013 Remaster"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "Bu A\u015fk Olur Mu"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "A\u015fk\u0131n Kenar\u0131ndan"
+    },
+    {
+        "artist": "Gipsy Kings",
+        "track": "Un Amor"
+    },
+    {
+        "artist": "Black Sabbath",
+        "track": "Paranoid"
+    },
+    {
+        "artist": "Santana",
+        "track": "She's Not There"
+    },
+    {
+        "artist": "Janis Joplin",
+        "track": "Kozmic Blues"
+    },
+    {
+        "artist": "Cannonball Adderley",
+        "track": "Autumn Leaves"
+    },
+    {
+        "artist": "Julian Lage",
+        "track": "Nocturne"
+    },
+    {
+        "artist": "Son Feci Bisiklet",
+        "track": "Bu K\u0131z"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Hold on I'm Coming"
+    },
+    {
+        "artist": "Amason",
+        "track": "California Dreamin'"
+    },
+    {
+        "artist": "Ajda Pekkan",
+        "track": "D\u00fc\u015f\u00fcnme Hi\u00e7"
+    },
+    {
+        "artist": "Selda Ba\u011fcan",
+        "track": "Dostum Dostum"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Old Love - Acoustic; Live at MTV Unplugged, Bray Film Studios, Windsor, England, UK, 1/16/1992; 2013 Remaster"
+    },
+    {
+        "artist": "\u00d6zdemir Erdo\u011fan",
+        "track": "A\u00e7 Kap\u0131y\u0131 Gir \u0130\u00e7eri"
+    },
+    {
+        "artist": "Frazey Ford",
+        "track": "One More Cup Of Coffee"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Sevda Ku\u015fun Kanad\u0131nda"
+    },
+    {
+        "artist": "Joe Walsh",
+        "track": "Life's Been Good"
+    },
+    {
+        "artist": "Foo Fighters",
+        "track": "The Sky Is A Neighborhood"
+    },
+    {
+        "artist": "Gary Clark Jr.",
+        "track": "When My Train Pulls In - Live"
+    },
+    {
+        "artist": "D\u00fc\u015f Soka\u011f\u0131 Sakinleri",
+        "track": "Huzun Kovan Kusu"
+    },
+    {
+        "artist": "Athena",
+        "track": "Yalan"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "Don't Let Me Down - Remastered"
+    },
+    {
+        "artist": "Kovacs",
+        "track": "My Love"
+    },
+    {
+        "artist": "Dire Straits",
+        "track": "Money For Nothing"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Shine On You Crazy Diamond - Parts 1 - 7;Edit; 2001 Remastered Version"
+    },
+    {
+        "artist": "Blue \u00d6yster Cult",
+        "track": "(Don't Fear) The Reaper"
+    },
+    {
+        "artist": "Santana",
+        "track": "Day of Celebration"
+    },
+    {
+        "artist": "Stevie Ray Vaughan",
+        "track": "Voodoo Child (Slight Return)"
+    },
+    {
+        "artist": "Santana",
+        "track": "Whole Lotta Love (feat. Chris Cornell)"
+    },
+    {
+        "artist": "Santana",
+        "track": "Everybody's Everything"
+    },
+    {
+        "artist": "Santana",
+        "track": "Evil Ways"
+    },
+    {
+        "artist": "Santana",
+        "track": "I Am Somebody (feat. will.i.am)"
+    },
+    {
+        "artist": "Santana",
+        "track": "The Calling (feat. Eric Clapton)"
+    },
+    {
+        "artist": "Jimi Hendrix",
+        "track": "Hey Joe"
+    },
+    {
+        "artist": "A New Music Experience",
+        "track": "Shine On You Crazy Diamond"
+    },
+    {
+        "artist": "Bon Jovi",
+        "track": "Dry County"
+    },
+    {
+        "artist": "Soul Keita",
+        "track": "Democracy, I Was Thirsty"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "I Want To Hold Your Hand - Remastered 2015"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "It's Now or Never"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "War"
+    },
+    {
+        "artist": "Steely Dan",
+        "track": "Reelin' In The Years"
+    },
+    {
+        "artist": "Tom Petty and the Heartbreakers",
+        "track": "Breakdown"
+    },
+    {
+        "artist": "Pat Martino",
+        "track": "Sunny"
+    },
+    {
+        "artist": "B.B. King",
+        "track": "Baby I Love You"
+    },
+    {
+        "artist": "Metallica",
+        "track": "For Whom The Bell Tolls - Remastered"
+    },
+    {
+        "artist": "Barcelona Gipsy Klezmer Orchestra",
+        "track": "Hasta Siempre, Comandante"
+    },
+    {
+        "artist": "Elle King",
+        "track": "Playing For Keeps"
+    },
+    {
+        "artist": "B.B. King",
+        "track": "Ghetto Woman"
+    },
+    {
+        "artist": "B.B. King",
+        "track": "Caldonia"
+    },
+    {
+        "artist": "Tom Petty and the Heartbreakers",
+        "track": "Mary Jane's Last Dance"
+    },
+    {
+        "artist": "Athena",
+        "track": "Bu Adam Fezadan"
+    },
+    {
+        "artist": "Foghat",
+        "track": "Louisiana Blues"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Kerk\u00fck Zindan\u0131"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Deniz \u00dcst\u00fc K\u00f6p\u00fcr\u00fcr"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "Warriors"
+    },
+    {
+        "artist": "Santana",
+        "track": "Treat"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Heartbreaker - 1990 Remaster"
+    },
+    {
+        "artist": "Mazhar Alanson",
+        "track": "Bu Ne Bi\u00e7im Hikaye B\u00f6yle"
+    },
+    {
+        "artist": "Santana",
+        "track": "Primavera"
+    },
+    {
+        "artist": "Santana",
+        "track": "Santana Jam"
+    },
+    {
+        "artist": "Santana",
+        "track": "Blues Magic"
+    },
+    {
+        "artist": "Lynyrd Skynyrd",
+        "track": "Poison Whiskey"
+    },
+    {
+        "artist": "Bar\u0131\u015f Man\u00e7o",
+        "track": "Nane Limon Kabu\u011fu"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Beni Siz Delirttiniz"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "Friction"
+    },
+    {
+        "artist": "Bob Dylan",
+        "track": "You Belong To Me"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Don't Panic"
+    },
+    {
+        "artist": "Gevende",
+        "track": "Sanki"
+    },
+    {
+        "artist": "Ray Charles",
+        "track": "Mess Around"
+    },
+    {
+        "artist": "Kylie Minogue",
+        "track": "Chocolate"
+    },
+    {
+        "artist": "Scorpions",
+        "track": "Still Loving You"
+    },
+    {
+        "artist": "Aretha Franklin",
+        "track": "At Last"
+    },
+    {
+        "artist": "Kool & The Gang",
+        "track": "Jungle Boogie"
+    },
+    {
+        "artist": "Gipsy Kings",
+        "track": "Hotel California - Spanish Mix"
+    },
+    {
+        "artist": "Gipsy Kings",
+        "track": "Volare (Nel Blu di Pinto di Blu)"
+    },
+    {
+        "artist": "Stevie Ray Vaughan",
+        "track": "The Things (That) I Used to Do"
+    },
+    {
+        "artist": "Erkin Koray",
+        "track": "Seni Her G\u00f6rd\u00fc\u011f\u00fcmde"
+    },
+    {
+        "artist": "Erkin Koray",
+        "track": "K\u0131zlar\u0131da Al\u0131n Askere"
+    },
+    {
+        "artist": "Erkin Koray",
+        "track": "Akrebin G\u00f6zleri"
+    },
+    {
+        "artist": "Wilson Pickett",
+        "track": "Hey Joe"
+    },
+    {
+        "artist": "Parliament",
+        "track": "Give Up The Funk (Tear The Roof Off The Sucker)"
+    },
+    {
+        "artist": "Al Green",
+        "track": "Love And Happiness"
+    },
+    {
+        "artist": "Santana",
+        "track": "Corazon Espinado (feat. Mana)"
+    },
+    {
+        "artist": "Kiss & Drive",
+        "track": "In Your Eyes"
+    },
+    {
+        "artist": "Amy Winehouse",
+        "track": "Mr Magic (Through The Smoke)"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Good Times Bad Times - 1993 Remaster"
+    },
+    {
+        "artist": "Al Jarreau",
+        "track": "Ain't no sunshine"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "T\u0131n T\u0131n"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Hepinize El Sallad\u0131m"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Orda Ortada"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "\u015eakac\u0131 Birisin Sen"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Ac\u0131n\u0131n \u0130lac\u0131"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "R\u00fcyalarda Buru\u015fmu\u015fum"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Ah Benim Hayat\u0131m"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "E Tabi"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Yanm\u0131\u015f \u0130\u00e7inden"
+    },
+    {
+        "artist": "Arctic Monkeys",
+        "track": "Fluorescent Adolescent"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Wish You Were Here - 2011 Remastered Version"
+    },
+    {
+        "artist": "Hooverphonic",
+        "track": "Mad About You"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Kashmir - 1990 Remaster"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Falling In Love (Is Hard On The Knees)"
+    },
+    {
+        "artist": "mor ve \u00f6tesi",
+        "track": "Serseri"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "Let It Be - Remastered"
+    },
+    {
+        "artist": "Procol Harum",
+        "track": "A Whiter Shade Of Pale"
+    },
+    {
+        "artist": "Beirut",
+        "track": "Nantes"
+    },
+    {
+        "artist": "Rod Stewart",
+        "track": "Da Ya Think I'm Sexy? - 2008 Remaster"
+    },
+    {
+        "artist": "Erik Truffaz",
+        "track": "Let Me Go ! (feat. Sophie Hunger)"
+    },
+    {
+        "artist": "Stevie Wonder",
+        "track": "Signed, Sealed, Delivered (I'm Yours)"
+    },
+    {
+        "artist": "ZZ Top",
+        "track": "La Grange - 1999 Remaster"
+    },
+    {
+        "artist": "TOTO",
+        "track": "Africa"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Sweet Emotion"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Knockin' On Heaven's Door"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Up&Up"
+    },
+    {
+        "artist": "Nev",
+        "track": "Sen Gibi"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "Hep Ya\u015f\u0131n 19"
+    },
+    {
+        "artist": "Guthrie Govan",
+        "track": "Lost in Rio"
+    },
+    {
+        "artist": "Craig David",
+        "track": "Rise & Fall - feat. Sting"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Autumn Leaves"
+    },
+    {
+        "artist": "James Blunt",
+        "track": "Goodbye My Lover"
+    },
+    {
+        "artist": "Harley's Studio Band",
+        "track": "This Life"
+    },
+    {
+        "artist": "Derek & The Dominos",
+        "track": "Layla - 40th Anniversary Version / 2010 Remastered"
+    },
+    {
+        "artist": "Nina Zilli",
+        "track": "50mila"
+    },
+    {
+        "artist": "Michael Jackson",
+        "track": "Love Never Felt So Good"
+    },
+    {
+        "artist": "Guadalupe Pineda",
+        "track": "Historia de un Amor"
+    },
+    {
+        "artist": "Celia Cruz",
+        "track": "La Vida Es Un Carnaval"
+    },
+    {
+        "artist": "Gipsy Kings",
+        "track": "Bamboleo"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Babe I'm Gonna Leave You - 1990 Remaster"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "D'yer Mak'er - 1990 Remaster"
+    },
+    {
+        "artist": "Louis Armstrong",
+        "track": "Summertime"
+    },
+    {
+        "artist": "Yusuf / Cat Stevens",
+        "track": "Wild World"
+    },
+    {
+        "artist": "Premiata Forneria Marconi",
+        "track": "Impressioni di settembre"
+    },
+    {
+        "artist": "Los Naranjos",
+        "track": "Cantarte a Ti"
+    },
+    {
+        "artist": "Stevie Wonder",
+        "track": "Superstition"
+    },
+    {
+        "artist": "Junior",
+        "track": "Mama Used To Say"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Could You Be Loved"
+    },
+    {
+        "artist": "B.B. King",
+        "track": "The Thrill Is Gone"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "\u0130nsan\u0131n D\u00fc\u015ft\u00fc\u011f\u00fc Durumlar"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Kad\u0131n"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Koca Ya\u015fl\u0131 \u015ei\u015fko D\u00fcnya"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Utanmazsan Unutmam"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Ko\u015fmadan"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "\u00c7ek Sifonu"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Bizim Zaman\u0131m\u0131z"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Kendime \u00c7aylar"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "\u00d6yle Normal"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Ate\u015f Ederim"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Bi \u00d6yle Bi B\u00f6yle"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Kap\u0131s\u0131 Kapal\u0131"
+    },
+    {
+        "artist": "Pilli Bebek",
+        "track": "Bak"
+    },
+    {
+        "artist": "Stevie Ray Vaughan",
+        "track": "Little Wing"
+    },
+    {
+        "artist": "Duman",
+        "track": "\u00d6yle Dertli"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Gotta Get Over - 2015 Remaster"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Ramblin' on My Mind"
+    },
+    {
+        "artist": "Howlin' Wolf",
+        "track": "Highway 49"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Black Dog - 1990 Remaster"
+    },
+    {
+        "artist": "Adamlar",
+        "track": "Kap\u0131s\u0131 Kapal\u0131"
+    },
+    {
+        "artist": "Guns N' Roses",
+        "track": "Civil War"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Hoochie Coochie Man"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Worried Life Blues"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Can't Hold Out Much Longer"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "My Very Good Friend the Milkman"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "For No One - Remastered"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "And I Love Her - Remastered"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "I Saw Her Standing There - Remastered"
+    },
+    {
+        "artist": "Wun Two",
+        "track": "Frio"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "While My Guitar Gently Weeps - Remastered"
+    },
+    {
+        "artist": "The Beatles",
+        "track": "Michelle - Remastered"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Adventure of a Lifetime"
+    },
+    {
+        "artist": "The Ting Tings",
+        "track": "Shut Up and Let Me Go"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "True Love"
+    },
+    {
+        "artist": "Queen",
+        "track": "Innuendo - Remastered 2011"
+    },
+    {
+        "artist": "Maroon 5",
+        "track": "She Will Be Loved"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Another Brick In The Wall - 2001 Remastered Version"
+    },
+    {
+        "artist": "Guns N' Roses",
+        "track": "November Rain"
+    },
+    {
+        "artist": "Jimi Hendrix",
+        "track": "Little Wing"
+    },
+    {
+        "artist": "James Brown",
+        "track": "It's A Man's, Man's, Man's World - Mono"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Time - 2011 Remastered Version"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Ya\u011fmur"
+    },
+    {
+        "artist": "Florence + The Machine",
+        "track": "Spectrum (Say My Name) - Calvin Harris Remix"
+    },
+    {
+        "artist": "Sia",
+        "track": "Elastic Heart"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Three Little Birds"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Worried Life Blues - Live; 2015 Remaster"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Wonderful Tonight"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Change the World - 2015 Remaster"
+    },
+    {
+        "artist": "The Rolling Stones",
+        "track": "Beast Of Burden - Remastered"
+    },
+    {
+        "artist": "Hugh Laurie",
+        "track": "The St. Louis Blues"
+    },
+    {
+        "artist": "Al Green",
+        "track": "Tired Of Being Alone"
+    },
+    {
+        "artist": "Afroman",
+        "track": "Because I Got High"
+    },
+    {
+        "artist": "Ray Charles",
+        "track": "Hallelujah, I Love Her So"
+    },
+    {
+        "artist": "The Doors",
+        "track": "Riders on the Storm"
+    },
+    {
+        "artist": "Queen",
+        "track": "I Was Born To Love You - 2011 Remaster"
+    },
+    {
+        "artist": "Van Halen",
+        "track": "Hot for Teacher - 2004 Remaster"
+    },
+    {
+        "artist": "Radiohead",
+        "track": "Creep"
+    },
+    {
+        "artist": "Maxence Cyrin",
+        "track": "Where Is My Mind"
+    },
+    {
+        "artist": "The Rolling Stones",
+        "track": "Miss You"
+    },
+    {
+        "artist": "Beats Antique",
+        "track": "The Porch"
+    },
+    {
+        "artist": "Sublime",
+        "track": "Santeria"
+    },
+    {
+        "artist": "Oasis",
+        "track": "Wonderwall"
+    },
+    {
+        "artist": "Jimi Hendrix",
+        "track": "Purple Haze"
+    },
+    {
+        "artist": "The Police",
+        "track": "Don't Stand So Close To Me"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Cocaine"
+    },
+    {
+        "artist": "Sting",
+        "track": "Fragile"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Ruhun Sar\u0131\u015f\u0131n"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Rapsodi \u0130stanbul"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Haziran"
+    },
+    {
+        "artist": "Teoman",
+        "track": "A\u015fk K\u0131r\u0131nt\u0131lar\u0131"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Senden \u00d6nce Senden Sonra"
+    },
+    {
+        "artist": "Teoman",
+        "track": "\u00c7oban Y\u0131ld\u0131z\u0131"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Renkli R\u00fcyalar Oteli"
+    },
+    {
+        "artist": "Guns N' Roses",
+        "track": "This I Love"
+    },
+    {
+        "artist": "Sam Smith",
+        "track": "I'm Not The Only One"
+    },
+    {
+        "artist": "James Blunt",
+        "track": "1973"
+    },
+    {
+        "artist": "Dire Straits",
+        "track": "Sultans Of Swing"
+    },
+    {
+        "artist": "AC/DC",
+        "track": "Back In Black"
+    },
+    {
+        "artist": "Marcus Miller",
+        "track": "Rush Over"
+    },
+    {
+        "artist": "Marcus Miller",
+        "track": "Detroit"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Princess of China"
+    },
+    {
+        "artist": "Louis Armstrong",
+        "track": "Kiss Of Fire"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Money - 2011 Remastered Version"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "African Herbsman"
+    },
+    {
+        "artist": "Nelly Furtado",
+        "track": "Say It Right - Main"
+    },
+    {
+        "artist": "Zaz",
+        "track": "Je veux"
+    },
+    {
+        "artist": "Bee Gees",
+        "track": "Stayin' Alive - Remastered Version"
+    },
+    {
+        "artist": "Yann Tiersen",
+        "track": "L'autre valse d'Am\u00e9lie"
+    },
+    {
+        "artist": "Jack Johnson",
+        "track": "Better Together"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Paradise"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Brain Damage - 2011 Remastered Version"
+    },
+    {
+        "artist": "Bastille",
+        "track": "Pompeii"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Get Up, Stand Up"
+    },
+    {
+        "artist": "The Black Keys",
+        "track": "Lonely Boy"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Sen De Ba\u015f\u0131n\u0131 Al\u0131p Gitme"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "O\u011fluma"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "I Shot the Sheriff - 2005 Remaster"
+    },
+    {
+        "artist": "AC/DC",
+        "track": "Highway to Hell"
+    },
+    {
+        "artist": "Jason Mraz",
+        "track": "I'm Yours"
+    },
+    {
+        "artist": "Neil Young",
+        "track": "Harvest Moon"
+    },
+    {
+        "artist": "Muse",
+        "track": "Starlight"
+    },
+    {
+        "artist": "Tina Turner",
+        "track": "The Best - Edit"
+    },
+    {
+        "artist": "Donna Summer",
+        "track": "Bad Girls"
+    },
+    {
+        "artist": "Florence + The Machine",
+        "track": "Shake It Out"
+    },
+    {
+        "artist": "Scorpions",
+        "track": "Where the River Flows - MTV Unplugged"
+    },
+    {
+        "artist": "Mick Jagger",
+        "track": "God Gave Me Everything"
+    },
+    {
+        "artist": "Gipsy Kings",
+        "track": "Djobi, Djoba"
+    },
+    {
+        "artist": "Gary Moore",
+        "track": "Still Got The Blues"
+    },
+    {
+        "artist": "80's Pop Band",
+        "track": "I Wanna Dance with Somebody"
+    },
+    {
+        "artist": "The Tolkien Ensemble",
+        "track": "Treebeard's Song"
+    },
+    {
+        "artist": "Hozier",
+        "track": "Take Me To Church"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Brain Damage - 2011 Remastered Version"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Comfortably Numb - 2011 Remastered Version"
+    },
+    {
+        "artist": "Bill Conti",
+        "track": "Gonna Fly Now (Theme From \"Rocky\") - 2006 - Remaster"
+    },
+    {
+        "artist": "Christina Perri",
+        "track": "Jar of Hearts"
+    },
+    {
+        "artist": "Bonnie Tyler",
+        "track": "Total Eclipse of the Heart"
+    },
+    {
+        "artist": "Bon Jovi",
+        "track": "Livin' On A Prayer"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Crazy"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "I Don't Want to Miss a Thing - From \"Armageddon\" Soundtrack"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Walk This Way"
+    },
+    {
+        "artist": "Daryl Hall & John Oates",
+        "track": "You Make My Dreams (Come True)"
+    },
+    {
+        "artist": "David Bowie",
+        "track": "Life On Mars? - 1999 Remastered Version"
+    },
+    {
+        "artist": "\u015eebnem Ferah",
+        "track": "Sigara"
+    },
+    {
+        "artist": "Nil Karaibrahimgil",
+        "track": "Kanatlar\u0131m Var Ruhumda"
+    },
+    {
+        "artist": "Deep Purple",
+        "track": "Smoke On The Water - Remastered 2012"
+    },
+    {
+        "artist": "Deep Purple",
+        "track": "Woman from Tokyo - 2000 Remaster"
+    },
+    {
+        "artist": "Herbie Hancock",
+        "track": "Tempo de Amor"
+    },
+    {
+        "artist": "Jeff Buckley",
+        "track": "Hallelujah"
+    },
+    {
+        "artist": "Israel Kamakawiwo'ole",
+        "track": "Somewhere Over The Rainbow_What A Wonderful World"
+    },
+    {
+        "artist": "Israel Kamakawiwo'ole",
+        "track": "Over The Rainbow"
+    },
+    {
+        "artist": "B\u00fclent Orta\u00e7gil",
+        "track": "De\u011firmenler"
+    },
+    {
+        "artist": "Emil\u00edana Torrini",
+        "track": "Dead Duck"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Three Little Birds"
+    },
+    {
+        "artist": "London Grammar",
+        "track": "Maybe"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "Whole Lotta Love - 1990 Remaster"
+    },
+    {
+        "artist": "Guns N' Roses",
+        "track": "Sweet Child O' Mine"
+    },
+    {
+        "artist": "Peter Hall",
+        "track": "Tom Bombadil's Song (I)"
+    },
+    {
+        "artist": "John Lennon",
+        "track": "Imagine - Remastered 2010"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Mavi Ku\u015f \u0130le K\u00fc\u00e7\u00fck K\u0131z"
+    },
+    {
+        "artist": "Santana",
+        "track": "Oye Como Va"
+    },
+    {
+        "artist": "Scorpions",
+        "track": "Wind Of Change"
+    },
+    {
+        "artist": "Muse",
+        "track": "Supermassive Black Hole"
+    },
+    {
+        "artist": "Bar\u0131\u015f Man\u00e7o",
+        "track": "Nazar Eyle - V1"
+    },
+    {
+        "artist": "James Brown",
+        "track": "I Got You (I Feel Good)"
+    },
+    {
+        "artist": "Y\u00fczy\u00fczeyken Konu\u015furuz",
+        "track": "Ate\u015f Edecek Misin?"
+    },
+    {
+        "artist": "Gevende",
+        "track": "Esinti"
+    },
+    {
+        "artist": "Tae Lane",
+        "track": "Low Down Pleasures"
+    },
+    {
+        "artist": "Robbie Williams",
+        "track": "Dream A Little Dream"
+    },
+    {
+        "artist": "HIM",
+        "track": "Gone With The Sin"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Trouble"
+    },
+    {
+        "artist": "Steely Dan",
+        "track": "Dirty Work"
+    },
+    {
+        "artist": "Marvin Gaye",
+        "track": "Got To Give It Up - Pt. 1"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Cryin'"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "God Put a Smile upon Your Face"
+    },
+    {
+        "artist": "Stan Getz",
+        "track": "Corcovado - Quiet Nights Of Quiet Stars"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "See You Soon"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Layla - Acoustic; Live at MTV Unplugged, Bray Film Studios, Windsor, England, UK, 1/16/1992; 1999 Remaster"
+    },
+    {
+        "artist": "Andrea Motis",
+        "track": "L.O.V.E"
+    },
+    {
+        "artist": "Scott Matthew",
+        "track": "L.o.v.e."
+    },
+    {
+        "artist": "Buena Vista Social Club",
+        "track": "Chan Chan"
+    },
+    {
+        "artist": "Pink Floyd",
+        "track": "Hey You - 2011 Remastered Version"
+    },
+    {
+        "artist": "The Neighbourhood",
+        "track": "Sweater Weather"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "True Love"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "A Rush of Blood to the Head"
+    },
+    {
+        "artist": "\u015eevval Sam",
+        "track": "Bozcaada (Por Uno Cabeza)"
+    },
+    {
+        "artist": "George Ezra",
+        "track": "Spectacular Rival"
+    },
+    {
+        "artist": "George Ezra",
+        "track": "Listen to the Man"
+    },
+    {
+        "artist": "Lucio Quarantotto",
+        "track": "Time To Say Goodbye (Con Te Partir\u00f2)"
+    },
+    {
+        "artist": "Fallulah",
+        "track": "Bridges - International Single Version"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "42"
+    },
+    {
+        "artist": "Marvin Gaye",
+        "track": "Let's Get It On"
+    },
+    {
+        "artist": "Amy Winehouse",
+        "track": "Tears Dry On Their Own"
+    },
+    {
+        "artist": "Lily Allen",
+        "track": "Littlest Things"
+    },
+    {
+        "artist": "Lily Allen",
+        "track": "22"
+    },
+    {
+        "artist": "Ben E. King",
+        "track": "Stand By Me"
+    },
+    {
+        "artist": "Marvin Gaye",
+        "track": "Let's Get It On"
+    },
+    {
+        "artist": "Led Zeppelin",
+        "track": "The Girl I Love She Got Long Black Wavy Hair - \"Tasty Pop Sundae\" Live Version From BBC Sessions"
+    },
+    {
+        "artist": "Sleeping At Last",
+        "track": "Mars"
+    },
+    {
+        "artist": "Keane",
+        "track": "Everybody's Changing"
+    },
+    {
+        "artist": "Lily Allen",
+        "track": "Everybody's Changing"
+    },
+    {
+        "artist": "Joe Cocker",
+        "track": "Unchain My Heart"
+    },
+    {
+        "artist": "Ray Charles",
+        "track": "I've Got a Woman"
+    },
+    {
+        "artist": "Back In Black",
+        "track": "TNT"
+    },
+    {
+        "artist": "Sting",
+        "track": "Shape Of My Heart"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Strawberry Swing"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Viva La Vida"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "Warriors"
+    },
+    {
+        "artist": "Sia",
+        "track": "Chandelier"
+    },
+    {
+        "artist": "Linkin Park",
+        "track": "Numb"
+    },
+    {
+        "artist": "ABBA",
+        "track": "Dancing Queen"
+    },
+    {
+        "artist": "No Doubt",
+        "track": "Don't Speak"
+    },
+    {
+        "artist": "Nina Simone",
+        "track": "Feeling Good"
+    },
+    {
+        "artist": "Nitin Sawhney",
+        "track": "Homelands"
+    },
+    {
+        "artist": "Buena Vista Social Club",
+        "track": "El Carretero"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Christmas Lights"
+    },
+    {
+        "artist": "Foster The People",
+        "track": "Pumped Up Kicks"
+    },
+    {
+        "artist": "Queen",
+        "track": "The Show Must Go On - Remastered 2011"
+    },
+    {
+        "artist": "Passion Pit",
+        "track": "Take a Walk"
+    },
+    {
+        "artist": "Back In Black",
+        "track": "Back in Black"
+    },
+    {
+        "artist": "Pixies",
+        "track": "Where Is My Mind?"
+    },
+    {
+        "artist": "Michael Jackson",
+        "track": "Smooth Criminal - 2012 Remaster"
+    },
+    {
+        "artist": "Andru Donalds",
+        "track": "Mishale"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "A Little Less Conversation"
+    },
+    {
+        "artist": "Ciara",
+        "track": "Can't Leave 'Em Alone (feat. 50 Cent)"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "A Sky Full of Stars"
+    },
+    {
+        "artist": "Rodr\u00edguez",
+        "track": "Sugar Man"
+    },
+    {
+        "artist": "The London Film Score Orchestra",
+        "track": "Far from Any Road - Main Title Theme From \"True Detectives\""
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Charlie Brown"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "It's Time"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "Demons"
+    },
+    {
+        "artist": "The Civil Wars",
+        "track": "Poison & Wine"
+    },
+    {
+        "artist": "Arctic Monkeys",
+        "track": "Why'd You Only Call Me When You're High?"
+    },
+    {
+        "artist": "Milky Chance",
+        "track": "Stolen Dance"
+    },
+    {
+        "artist": "Mirkelam",
+        "track": "Hat\u0131ralar"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Magic"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Dream On"
+    },
+    {
+        "artist": "Cem Karaca",
+        "track": "Resimdeki G\u00f6zyaslari (Apaslar)"
+    },
+    {
+        "artist": "Gorillaz",
+        "track": "Feel Good Inc."
+    },
+    {
+        "artist": "Howard Shore",
+        "track": "Shore: Lord of the Rings: The Fellowship"
+    },
+    {
+        "artist": "B\u00fclent Orta\u00e7gil",
+        "track": "Benimle Oynarm\u0131s\u0131n"
+    },
+    {
+        "artist": "Serena Ryder",
+        "track": "Stompa"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "One Love"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Shiver"
+    },
+    {
+        "artist": "Red Hot Chili Peppers",
+        "track": "Snow (Hey Oh)"
+    },
+    {
+        "artist": "Gnarls Barkley",
+        "track": "Crazy"
+    },
+    {
+        "artist": "Empire of the Sun",
+        "track": "Alive"
+    },
+    {
+        "artist": "Empire of the Sun",
+        "track": "We Are The People"
+    },
+    {
+        "artist": "Sting",
+        "track": "Englishman In New York"
+    },
+    {
+        "artist": "Ben E. King",
+        "track": "Stand By Me"
+    },
+    {
+        "artist": "John Lennon",
+        "track": "Imagine"
+    },
+    {
+        "artist": "Santa Esmeralda",
+        "track": "Don't Let Me Be Misunderstood"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "No Woman, No Cry - Live At The Lyceum, London/1975"
+    },
+    {
+        "artist": "Frank Sinatra",
+        "track": "Blue Moon - 1999 Remastered"
+    },
+    {
+        "artist": "Sertab Erener",
+        "track": "Zor Kad\u0131n"
+    },
+    {
+        "artist": "Sertab Erener",
+        "track": "Kumsalda"
+    },
+    {
+        "artist": "Moby",
+        "track": "Porcelain"
+    },
+    {
+        "artist": "Band Of Skulls",
+        "track": "I Know What I Am"
+    },
+    {
+        "artist": "Urge Overkill",
+        "track": "Girl, You'll Be a Woman Soon"
+    },
+    {
+        "artist": "Ingrid Michaelson",
+        "track": "You and I"
+    },
+    {
+        "artist": "Pharrell Williams",
+        "track": "Happy"
+    },
+    {
+        "artist": "Hurts",
+        "track": "Stay"
+    },
+    {
+        "artist": "Norah Jones",
+        "track": "Say Goodbye"
+    },
+    {
+        "artist": "Fallulah",
+        "track": "I Lay My Head"
+    },
+    {
+        "artist": "Jimi Hendrix",
+        "track": "Izabella"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Only Superstition"
+    },
+    {
+        "artist": "The Wailers",
+        "track": "I Shot The Sheriff"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Is This Love"
+    },
+    {
+        "artist": "Quantic",
+        "track": "Time Is the Enemy"
+    },
+    {
+        "artist": "Corinne Bailey Rae",
+        "track": "Choux Pastry Heart"
+    },
+    {
+        "artist": "Agnes Obel",
+        "track": "Just So"
+    },
+    {
+        "artist": "Agnes Obel",
+        "track": "The Curse"
+    },
+    {
+        "artist": "Fallulah",
+        "track": "Carol of the Bells"
+    },
+    {
+        "artist": "Patrick Watson",
+        "track": "Lighthouse"
+    },
+    {
+        "artist": "Aerosmith",
+        "track": "Come Together"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Tek Ba\u015f\u0131na Dans"
+    },
+    {
+        "artist": "Teoman",
+        "track": "G\u00fczel Bir G\u00fcn"
+    },
+    {
+        "artist": "M\u00fcsl\u00fcm G\u00fcrses",
+        "track": "Sensiz Olmaz"
+    },
+    {
+        "artist": "Imagine Dragons",
+        "track": "Radioactive"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "Jailhouse Rock"
+    },
+    {
+        "artist": "The Blues Brothers",
+        "track": "Everybody Needs Somebody To Love"
+    },
+    {
+        "artist": "Queen",
+        "track": "Sail Away Sweet Sister - Remastered 2011"
+    },
+    {
+        "artist": "Stereo Avenue",
+        "track": "Mad World"
+    },
+    {
+        "artist": "Feist",
+        "track": "Mushaboom"
+    },
+    {
+        "artist": "Carla Bruni",
+        "track": "Quelqu'un M'a Dit"
+    },
+    {
+        "artist": "The Cinematic Orchestra",
+        "track": "That Home"
+    },
+    {
+        "artist": "K Team",
+        "track": "Sorry Seem to Be the Hardest Word - Originally Performed By Elton John"
+    },
+    {
+        "artist": "Sean Paul",
+        "track": "Hold My Hand (feat. Keri Hilson)"
+    },
+    {
+        "artist": "Dido",
+        "track": "White Flag"
+    },
+    {
+        "artist": "Queen",
+        "track": "Save Me - 2011 Remaster"
+    },
+    {
+        "artist": "Hans Zimmer",
+        "track": "Time"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Every Teardrop Is a Waterfall"
+    },
+    {
+        "artist": "S\u00f3ley",
+        "track": "Pretty Face"
+    },
+    {
+        "artist": "S\u00f3ley",
+        "track": "The Sun Is Going Down II"
+    },
+    {
+        "artist": "Bob Dylan",
+        "track": "The Man in Me"
+    },
+    {
+        "artist": "Bob Dylan",
+        "track": "Like a Rolling Stone"
+    },
+    {
+        "artist": "Bob Dylan",
+        "track": "Isis"
+    },
+    {
+        "artist": "Bob Dylan",
+        "track": "Hurricane"
+    },
+    {
+        "artist": "Lana Del Rey",
+        "track": "Yayo"
+    },
+    {
+        "artist": "Red Hot Chili Peppers",
+        "track": "Can't Stop"
+    },
+    {
+        "artist": "Pirates Of New Providence",
+        "track": "Dead Man's Chest"
+    },
+    {
+        "artist": "fun.",
+        "track": "We Are Young (feat. Janelle Mon\u00e1e)"
+    },
+    {
+        "artist": "Amatorski",
+        "track": "Soldier"
+    },
+    {
+        "artist": "Scott McKenzie",
+        "track": "San Francisco (Be Sure to Wear Flowers in Your Hair)"
+    },
+    {
+        "artist": "Red Hot Chili Peppers",
+        "track": "Otherside"
+    },
+    {
+        "artist": "The Yesteryears",
+        "track": "Yesterday"
+    },
+    {
+        "artist": "Robbie Williams",
+        "track": "Angels"
+    },
+    {
+        "artist": "Robbie Williams",
+        "track": "Feel"
+    },
+    {
+        "artist": "The Verve",
+        "track": "Bitter Sweet Symphony"
+    },
+    {
+        "artist": "Queen",
+        "track": "Crazy Little Thing Called Love - 2011 Remaster"
+    },
+    {
+        "artist": "Queen",
+        "track": "Another One Bites The Dust - 2011 Remaster"
+    },
+    {
+        "artist": "Queen",
+        "track": "Don't Stop Me Now - 2011 Remaster"
+    },
+    {
+        "artist": "Johnny Cash",
+        "track": "Hurt"
+    },
+    {
+        "artist": "Christina Aguilera",
+        "track": "Hurt"
+    },
+    {
+        "artist": "Duman",
+        "track": "Elleri Ellerime"
+    },
+    {
+        "artist": "Duman",
+        "track": "Yanibasimdan"
+    },
+    {
+        "artist": "Duman",
+        "track": "Senden Daha G\u00fczel"
+    },
+    {
+        "artist": "Mathieu Boogaerts",
+        "track": "Avant que je m'ennuie"
+    },
+    {
+        "artist": "Queen",
+        "track": "Love Of My Life - Remastered 2011"
+    },
+    {
+        "artist": "Cinescore",
+        "track": "The Hobbit - Misty Mountains (The Dwarves Song) - Cover Version"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "All Shook Up"
+    },
+    {
+        "artist": "Pinhani",
+        "track": "Zaman Beklemez"
+    },
+    {
+        "artist": "Volkan Konak",
+        "track": "Gurbet"
+    },
+    {
+        "artist": "\u00d6rsten",
+        "track": "Fleur blanche"
+    },
+    {
+        "artist": "Stateless",
+        "track": "Bloodstream"
+    },
+    {
+        "artist": "Keane",
+        "track": "Somewhere Only We Know"
+    },
+    {
+        "artist": "Sakin",
+        "track": "Edepsiz Komedya"
+    },
+    {
+        "artist": "Rod Stewart",
+        "track": "Moonglow"
+    },
+    {
+        "artist": "Christina Aguilera",
+        "track": "You Lost Me"
+    },
+    {
+        "artist": "Donae'o",
+        "track": "Move to Da Gyal Dem - Original"
+    },
+    {
+        "artist": "M83",
+        "track": "Outro"
+    },
+    {
+        "artist": "Morcheeba",
+        "track": "Enjoy The Ride"
+    },
+    {
+        "artist": "MF\u00d6",
+        "track": "Bu Sabah Ya\u011fmur Var \u0130stanbul'da"
+    },
+    {
+        "artist": "Aretha Franklin",
+        "track": "The House That Jack Built"
+    },
+    {
+        "artist": "The Kooks",
+        "track": "Mr. Nice Guy"
+    },
+    {
+        "artist": "Muse",
+        "track": "Uprising"
+    },
+    {
+        "artist": "123",
+        "track": "so much to say"
+    },
+    {
+        "artist": "123",
+        "track": "turuncu"
+    },
+    {
+        "artist": "123",
+        "track": "Again"
+    },
+    {
+        "artist": "123",
+        "track": "binalar"
+    },
+    {
+        "artist": "123",
+        "track": "sun in the arms of love"
+    },
+    {
+        "artist": "Leonard Cohen",
+        "track": "Dance Me To The End Of Love"
+    },
+    {
+        "artist": "Roy Hargrove",
+        "track": "I'm Not So Sure"
+    },
+    {
+        "artist": "Johnnie Taylor",
+        "track": "Running Out Of Lies"
+    },
+    {
+        "artist": "Roy Hargrove",
+        "track": "Strasbourg / St. Denis"
+    },
+    {
+        "artist": "Damian Marley",
+        "track": "Road To Zion"
+    },
+    {
+        "artist": "Bob Marley & The Wailers",
+        "track": "Buffalo Soldier"
+    },
+    {
+        "artist": "Depeche Mode",
+        "track": "Personal Jesus - 2006 Digital Remaster"
+    },
+    {
+        "artist": "Daft Punk",
+        "track": "Lose Yourself to Dance (feat. Pharrell Williams)"
+    },
+    {
+        "artist": "Mark Isham",
+        "track": "Our Eyes Aren't What Make Us See"
+    },
+    {
+        "artist": "Elvis Presley",
+        "track": "Blue Suede Shoes"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Atlas - From \u201cThe Hunger Games: Catching Fire\u201d/Soundtrack"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Yellow"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Warning Sign"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "The Scientist"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "In My Place"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Twisted Logic"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "The Hardest Part"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Speed of Sound"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Talk"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Fix You"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "White Shadows"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "What If"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Square One"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Violet Hill"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Yes"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Lost!"
+    },
+    {
+        "artist": "Santana",
+        "track": "Black Magic Woman"
+    },
+    {
+        "artist": "Santana",
+        "track": "Smooth"
+    },
+    {
+        "artist": "Maroon 5",
+        "track": "Harder To Breathe"
+    },
+    {
+        "artist": "MIKA",
+        "track": "Grace Kelly"
+    },
+    {
+        "artist": "Enya",
+        "track": "May It Be"
+    },
+    {
+        "artist": "Enya",
+        "track": "Only Time"
+    },
+    {
+        "artist": "Queen",
+        "track": "It's A Hard Life - Remastered 2011"
+    },
+    {
+        "artist": "Queen",
+        "track": "Good Old-Fashioned Lover Boy - Remastered 2011"
+    },
+    {
+        "artist": "Queen",
+        "track": "I Want To Break Free - Remastered 2011"
+    },
+    {
+        "artist": "Queen",
+        "track": "Bohemian Rhapsody - 2011 Remaster"
+    },
+    {
+        "artist": "Queen",
+        "track": "Fat Bottomed Girls - 2011 Remaster"
+    },
+    {
+        "artist": "Adele",
+        "track": "Don't You Remember"
+    },
+    {
+        "artist": "Adele",
+        "track": "Someone Like You"
+    },
+    {
+        "artist": "Adele",
+        "track": "Hometown Glory"
+    },
+    {
+        "artist": "Iggy Pop",
+        "track": "The Passenger"
+    },
+    {
+        "artist": "Coldplay",
+        "track": "Mylo Xyloto"
+    },
+    {
+        "artist": "ABBA",
+        "track": "Waterloo"
+    },
+    {
+        "artist": "Guns N' Roses",
+        "track": "Knockin' On Heaven's Door"
+    },
+    {
+        "artist": "Train",
+        "track": "Hey, Soul Sister"
+    },
+    {
+        "artist": "The D\u00f8",
+        "track": "Too Insistent"
+    },
+    {
+        "artist": "Eric Clapton",
+        "track": "Riding with the King"
+    },
+    {
+        "artist": "Teoman",
+        "track": "Papatya"
+    },
+    {
+        "artist": "Agnes Obel",
+        "track": "Riverside"
+    },
+    {
+        "artist": "Lily Allen",
+        "track": "Not Fair"
+    },
+    {
+        "artist": "Lily Allen",
+        "track": "Fuck You"
+    },
+    {
+        "artist": "John Newman",
+        "track": "Love Me Again"
+    },
+    {
+        "artist": "Ya\u015far",
+        "track": "Gel Benimle"
+    },
+    {
+        "artist": "Ya\u015far",
+        "track": "Kumral\u0131m"
+    },
+    {
+        "artist": "Queen",
+        "track": "I Want It All - Remastered 2011"
+    },
+    {
+        "artist": "Selah Sue",
+        "track": "This World"
+    },
+    {
+        "artist": "Bon Jovi",
+        "track": "You Give Love A Bad Name"
+    },
+    {
+        "artist": "Bon Jovi",
+        "track": "It's My Life"
+    },
+    {
+        "artist": "Jack Johnson",
+        "track": "Never Know"
+    },
+    {
+        "artist": "The Rolling Stones",
+        "track": "Gimme Shelter"
+    },
+    {
+        "artist": "The Rolling Stones",
+        "track": "Paint It Black"
+    },
+    {
+        "artist": "Edward Sharpe & The Magnetic Zeros",
+        "track": "Home"
+    },
+    {
+        "artist": "Nirvana",
+        "track": "Smells Like Teen Spirit"
+    },
+    {
+        "artist": "Arctic Monkeys",
+        "track": "Do I Wanna Know?"
+    }
+]

notebook.ipynb

@@ -0,0 +1,2256 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%load_ext autoreload\n",
+    "%autoreload 2"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 99,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import numpy as np\n",
+    "import librosa\n",
+    "import torch\n",
+    "from src import laion_clap\n",
+    "from glob import glob\n",
+    "import pandas as pd\n",
+    "import jmespath"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 54,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# quantization\n",
+    "def int16_to_float32(x):\n",
+    "    return (x / 32767.0).astype(np.float32)\n",
+    "\n",
+    "\n",
+    "def float32_to_int16(x):\n",
+    "    x = np.clip(x, a_min=-1., a_max=1.)\n",
+    "    return (x * 32767.).astype(np.int16)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "model = laion_clap.CLAP_Module(enable_fusion=False, amodel= 'HTSAT-base')\n",
+    "model.load_ckpt(ckpt=\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/model_checkpoints/music_audioset_epoch_15_esc_90.14.pt\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 55,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "def load_music_file(file_name):\n",
+    "    audio_data, _ = librosa.load(file_name, sr=48000) # sample rate should be 48000\n",
+    "    audio_data = audio_data.reshape(1, -1) # Make it (1,T) or (N,T)\n",
+    "    # audio_data = torch.from_numpy(int16_to_float32(float32_to_int16(audio_data))).float() # quantize before send it in to the model\n",
+    "    with torch.no_grad():\n",
+    "        audio_embed = model.get_audio_embedding_from_data(x = audio_data, use_tensor=False)\n",
+    "    return audio_embed\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 54,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "with open(\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/data/json/final_track_data.json\", \"r\") as reader:\n",
+    "    track_data = json.load(reader)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "music_files = glob(\"/Users/berkayg/Codes/music-project/AudioCLIP/data/downloaded_tracks/*.wav\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import pickle\n",
+    "with open(\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/data/vectors/song_names.pkl\", \"rb\") as reader:\n",
+    "    ls = pickle.load(reader)\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 69,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/var/folders/sr/r72219hj06x_1xvw7hhd517h0000gn/T/ipykernel_39391/3009710654.py:2: UserWarning: PySoundFile failed. Trying audioread instead.\n",
+      "  audio_data, _ = librosa.load(file_name, sr=48000) # sample rate should be 48000\n",
+      "/Users/berkayg/miniforge3/envs/playlist-curator/lib/python3.10/site-packages/librosa/core/audio.py:183: FutureWarning: librosa.core.audio.__audioread_load\n",
+      "\tDeprecated as of librosa version 0.10.0.\n",
+      "\tIt will be removed in librosa version 1.0.\n",
+      "  y, sr_native = __audioread_load(path, offset, duration, dtype)\n"
+     ]
+    }
+   ],
+   "source": [
+    "music_data = np.zeros((len(track_data), 512), dtype=np.float32)\n",
+    "track_data_new = []\n",
+    "idx = 0\n",
+    "for m in track_data:\n",
+    "    if m[\"file_path\"]:\n",
+    "        music_data[idx] = load_music_file(m[\"file_path\"])\n",
+    "        dc = m.copy()\n",
+    "        dc.update({\"vector_idx\": idx})\n",
+    "        track_data_new.append(dc)\n",
+    "        idx += 1\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 105,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "vector_db_data = track_data_new.copy()\n",
+    "vector_db_data.sort(key=lambda x: x[\"vector_idx\"])\n",
+    "vector_ids = [f\"audio_{k['vector_idx']}\" for k in vector_db_data]\n",
+    "vector_indices = [k[\"vector_idx\"] for k in vector_db_data]\n",
+    "vector_db_data = jmespath.search(\"[*].{artist_name: artist_name, track_name: track_name, title: title, link: link, vector_idx: vector_idx}\", vector_db_data)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 110,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from chromadb import Documents, EmbeddingFunction, Embeddings\n",
+    "\n",
+    "class CuratorTextEmbedding(EmbeddingFunction):\n",
+    "    def __call__(self, text: Documents) -> Embeddings:\n",
+    "        # embed the documents somehow\n",
+    "        return embeddings"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import chromadb\n",
+    "chroma_client = chromadb.Client()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 166,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "chroma_client.delete_collection(name=\"playlist_collection\")\n",
+    "collection = chroma_client.create_collection(name=\"playlist_collection\", metadata={\"hnsw:space\": \"ip\"})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 167,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "collection.add(\n",
+    "    embeddings=music_data[vector_indices].tolist(),\n",
+    "    metadatas=vector_db_data,\n",
+    "    ids=vector_ids\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 172,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\u001b[0;31mSignature:\u001b[0m\n",
+      "\u001b[0mcollection\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mquery\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mquery_embeddings\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mSequence\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mfloat\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mSequence\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mint\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mSequence\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mfloat\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mSequence\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mint\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mNoneType\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mquery_texts\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mNoneType\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mquery_images\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mndarray\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mAny\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdtype\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0muint64\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mint64\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfloat64\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mndarray\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mAny\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mdtype\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0muint64\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mint64\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mnumpy\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mfloat64\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mNoneType\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mquery_uris\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mNoneType\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mn_results\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mint\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;36m10\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mwhere\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$and'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$or'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mint\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfloat\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mbool\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mDict\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$gt'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$gte'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$lt'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$lte'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$ne'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$eq'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$and'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$or'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mint\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfloat\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mbool\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mDict\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$in'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$nin'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mint\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mfloat\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mbool\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mForwardRef\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'Where'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0mwhere_document\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mOptional\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mDict\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$contains'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$and'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'$or'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mstr\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mForwardRef\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'WhereDocument'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;32mNone\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m    \u001b[0minclude\u001b[0m\u001b[0;34m:\u001b[0m \u001b[0mList\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mUnion\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'documents'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'embeddings'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'metadatas'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'distances'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'uris'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mLiteral\u001b[0m\u001b[0;34m[\u001b[0m\u001b[0;34m'data'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m]\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0;34m'metadatas'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'documents'\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;34m'distances'\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\n",
+      "\u001b[0;34m\u001b[0m\u001b[0;34m)\u001b[0m \u001b[0;34m->\u001b[0m \u001b[0mchromadb\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mapi\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mtypes\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mQueryResult\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
+      "\u001b[0;31mDocstring:\u001b[0m\n",
+      "Get the n_results nearest neighbor embeddings for provided query_embeddings or query_texts.\n",
+      "\n",
+      "Args:\n",
+      "    query_embeddings: The embeddings to get the closes neighbors of. Optional.\n",
+      "    query_texts: The document texts to get the closes neighbors of. Optional.\n",
+      "    query_images: The images to get the closes neighbors of. Optional.\n",
+      "    n_results: The number of neighbors to return for each query_embedding or query_texts. Optional.\n",
+      "    where: A Where type dict used to filter results by. E.g. `{\"$and\": [\"color\" : \"red\", \"price\": {\"$gte\": 4.20}]}`. Optional.\n",
+      "    where_document: A WhereDocument type dict used to filter by the documents. E.g. `{$contains: {\"text\": \"hello\"}}`. Optional.\n",
+      "    include: A list of what to include in the results. Can contain `\"embeddings\"`, `\"metadatas\"`, `\"documents\"`, `\"distances\"`. Ids are always included. Defaults to `[\"metadatas\", \"documents\", \"distances\"]`. Optional.\n",
+      "\n",
+      "Returns:\n",
+      "    QueryResult: A QueryResult object containing the results.\n",
+      "\n",
+      "Raises:\n",
+      "    ValueError: If you don't provide either query_embeddings, query_texts, or query_images\n",
+      "    ValueError: If you provide both query_embeddings and query_texts\n",
+      "    ValueError: If you provide both query_embeddings and query_images\n",
+      "    ValueError: If you provide both query_texts and query_images\n",
+      "\u001b[0;31mFile:\u001b[0m      ~/miniforge3/envs/playlist-curator/lib/python3.10/site-packages/chromadb/api/models/Collection.py\n",
+      "\u001b[0;31mType:\u001b[0m      method"
+     ]
+    }
+   ],
+   "source": [
+    "collection.query?"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 173,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "{'ids': [['audio_292',\n",
+       "   'audio_481',\n",
+       "   'audio_298',\n",
+       "   'audio_474',\n",
+       "   'audio_121',\n",
+       "   'audio_476',\n",
+       "   'audio_337',\n",
+       "   'audio_472',\n",
+       "   'audio_225',\n",
+       "   'audio_482']],\n",
+       " 'distances': [[0.6302633285522461,\n",
+       "   0.6571106910705566,\n",
+       "   0.6896730661392212,\n",
+       "   0.7028166055679321,\n",
+       "   0.7428299784660339,\n",
+       "   0.7440136671066284,\n",
+       "   0.7793576717376709,\n",
+       "   0.7837952971458435,\n",
+       "   0.8032999038696289,\n",
+       "   0.8056029081344604]],\n",
+       " 'metadatas': [[{'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=kcASPx3-HuI',\n",
+       "    'title': 'Coldplay - Trouble (Official video)',\n",
+       "    'track_name': 'Trouble',\n",
+       "    'vector_idx': 292},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=k4V3Mo61fJM',\n",
+       "    'title': 'Coldplay - Fix You (Official Video)',\n",
+       "    'track_name': 'Fix You',\n",
+       "    'vector_idx': 481},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=57rEQZiklxQ',\n",
+       "    'title': 'See You Soon',\n",
+       "    'track_name': 'See You Soon',\n",
+       "    'vector_idx': 298},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=xtQirM784oM',\n",
+       "    'title': 'Warning Sign',\n",
+       "    'track_name': 'Warning Sign',\n",
+       "    'vector_idx': 474},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=BPNTC7uZYrI',\n",
+       "    'title': 'Coldplay - Up&Up (Official Video)',\n",
+       "    'track_name': 'Up&Up',\n",
+       "    'vector_idx': 121},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=gnIZ7RMuLpU',\n",
+       "    'title': 'Coldplay - In My Place (Official 4K Video)',\n",
+       "    'track_name': 'In My Place',\n",
+       "    'vector_idx': 476},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=z1rYmzQ8C9Q',\n",
+       "    'title': 'Coldplay - Christmas Lights (Official Video)',\n",
+       "    'track_name': 'Christmas Lights',\n",
+       "    'vector_idx': 337},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=Lh3TokLzzmw',\n",
+       "    'title': 'Coldplay - Atlas (Hunger Games: Catching Fire)(Official Lyric Video)',\n",
+       "    'track_name': 'Atlas - From “The Hunger Games: Catching Fire”/Soundtrack',\n",
+       "    'vector_idx': 472},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=1Uw6ZkbsAH8',\n",
+       "    'title': 'Coldplay - Princess Of China ft. Rihanna (Official Video)',\n",
+       "    'track_name': 'Princess of China',\n",
+       "    'vector_idx': 225},\n",
+       "   {'artist_name': 'Coldplay',\n",
+       "    'link': 'https://www.youtube.com/watch?v=EhC_c7p50so',\n",
+       "    'title': 'White Shadows',\n",
+       "    'track_name': 'White Shadows',\n",
+       "    'vector_idx': 482}]],\n",
+       " 'embeddings': None,\n",
+       " 'documents': [[None, None, None, None, None, None, None, None, None, None]],\n",
+       " 'uris': None,\n",
+       " 'data': None}"
+      ]
+     },
+     "execution_count": 173,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "text_data = [\"romantic, depressing\"] \n",
+    "text_embed = model.get_text_embedding(text_data)\n",
+    "collection.query(query_embeddings=text_embed.tolist(), n_results=10, where={\"artist_name\": \"Coldplay\"})"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 111,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "(1, 512)\n"
+     ]
+    }
+   ],
+   "source": [
+    "text_data = [\"This audio is an energetic, uplifting song\"] \n",
+    "text_embed = model.get_text_embedding(text_data)\n",
+    "print(text_embed.shape)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "song_names = [k.split(\"/\")[-1] for k in music_files]"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 52,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "torch.Size([630, 1])\n"
+     ]
+    }
+   ],
+   "source": [
+    "ranking = torch.tensor(audio_vectors) @ torch.tensor(text_embed).t()\n",
+    "ranking = ranking[:, 0].reshape(-1, 1)\n",
+    "print(ranking.shape)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 53,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>This audio is an energetic, uplifting song</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>Ciara - Can't Leave 'Em Alone (feat. 50 Cent).wav</th>\n",
+       "      <td>0.423258</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>The Weeknd - Blinding Lights.wav</th>\n",
+       "      <td>0.415049</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Stromae - Tous les mêmes.wav</th>\n",
+       "      <td>0.412208</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Empire of the Sun - Alive.wav</th>\n",
+       "      <td>0.373571</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Kylie Minogue - Chocolate.wav</th>\n",
+       "      <td>0.348002</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Sia - Elastic Heart.wav</th>\n",
+       "      <td>0.334062</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Sia - Chandelier.wav</th>\n",
+       "      <td>0.330263</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Stevie Wonder - Signed, Sealed, Delivered (I'm Yours).wav</th>\n",
+       "      <td>0.324698</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Coldplay - Princess of China.wav</th>\n",
+       "      <td>0.319533</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Florence + The Machine - Spectrum (Say My Name) - Calvin Harris Remix.wav</th>\n",
+       "      <td>0.315443</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Coldplay - True Love.wav</th>\n",
+       "      <td>0.308231</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Coldplay - Talk.wav</th>\n",
+       "      <td>0.307894</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Lily Allen - Not Fair.wav</th>\n",
+       "      <td>0.300576</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Andru Donalds - Mishale.wav</th>\n",
+       "      <td>0.296594</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>Linkin Park - Numb.wav</th>\n",
+       "      <td>0.294131</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "                                                    This audio is an energetic, uplifting song\n",
+       "Ciara - Can't Leave 'Em Alone (feat. 50 Cent).wav                                     0.423258\n",
+       "The Weeknd - Blinding Lights.wav                                                      0.415049\n",
+       "Stromae - Tous les mêmes.wav                                                          0.412208\n",
+       "Empire of the Sun - Alive.wav                                                         0.373571\n",
+       "Kylie Minogue - Chocolate.wav                                                         0.348002\n",
+       "Sia - Elastic Heart.wav                                                               0.334062\n",
+       "Sia - Chandelier.wav                                                                  0.330263\n",
+       "Stevie Wonder - Signed, Sealed, Delivered (I'm ...                                    0.324698\n",
+       "Coldplay - Princess of China.wav                                                      0.319533\n",
+       "Florence + The Machine - Spectrum (Say My Name)...                                    0.315443\n",
+       "Coldplay - True Love.wav                                                              0.308231\n",
+       "Coldplay - Talk.wav                                                                   0.307894\n",
+       "Lily Allen - Not Fair.wav                                                             0.300576\n",
+       "Andru Donalds - Mishale.wav                                                           0.296594\n",
+       "Linkin Park - Numb.wav                                                                0.294131"
+      ]
+     },
+     "execution_count": 53,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "pd.DataFrame(ranking, columns=[text_data[0]], index=song_names).nlargest(15, text_data[0])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import json\n",
+    "import pickle\n",
+    "def load_data():\n",
+    "    vectors = np.load(\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/data/vectors/audio_representations.npy\")\n",
+    "    with open(\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/data/vectors/song_names.pkl\", \"rb\") as reader:\n",
+    "        song_names = pickle.load(reader)\n",
+    "\n",
+    "    with open(\"/Users/berkayg/Codes/music-project/laion-clap-project/curate-me-a-playlist/data/json/youtube_data.json\", \"r\") as reader:\n",
+    "        youtube_data = json.load(reader)\n",
+    "\n",
+    "    df_youtube = pd.DataFrame(youtube_data)\n",
+    "    df_youtube[\"id\"] = df_youtube[\"artist_name\"] + \" - \" + df_youtube[\"track_name\"] + \".wav\"\n",
+    "    df_youtube.set_index(\"id\", inplace=True)\n",
+    "    return vectors, song_names, df_youtube\n",
+    "audio_vectors, song_names, df_youtube = load_data()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 44,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from sklearn.manifold import TSNE\n",
+    "decomposer = TSNE(3)\n",
+    "audio_components = decomposer.fit_transform(np.vstack([audio_vectors, text_embed]))"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 76,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "torch.Size([630, 1])"
+      ]
+     },
+     "execution_count": 76,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import plotly.express as px\n",
+    "(torch.tensor(text_embed) / torch.linalg.norm(torch.tensor(text_embed), dim=-1, keepdim=True))\n",
+    "torch.linalg.norm(torch.tensor(audio_vectors), dim=-1, keepdim=True).shape"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 48,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "array([[-0.0228844 , -0.01531727, -0.02099325, ...,  0.02487872,\n",
+       "        -0.04954423, -0.00967047],\n",
+       "       [ 0.05719518, -0.05181887,  0.02295402, ...,  0.04154928,\n",
+       "        -0.01846843,  0.0015544 ],\n",
+       "       [-0.01932843, -0.03254181, -0.02956614, ...,  0.04285421,\n",
+       "        -0.00225228, -0.00785008],\n",
+       "       ...,\n",
+       "       [-0.05801252,  0.05521129, -0.06229992, ..., -0.02577653,\n",
+       "         0.01485025, -0.04933776],\n",
+       "       [-0.01117569,  0.03371193,  0.03643309, ...,  0.03710453,\n",
+       "         0.06688339,  0.04197252],\n",
+       "       [ 0.02041594, -0.00629344, -0.10389867, ..., -0.02280687,\n",
+       "        -0.02982889,  0.03500484]], dtype=float32)"
+      ]
+     },
+     "execution_count": 48,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "np.vstack([audio_vectors, text_embed])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 45,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "songs = song_names.copy()\n",
+    "songs.append(\"text_embedding\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 46,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/html": [
+       "<div>\n",
+       "<style scoped>\n",
+       "    .dataframe tbody tr th:only-of-type {\n",
+       "        vertical-align: middle;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe tbody tr th {\n",
+       "        vertical-align: top;\n",
+       "    }\n",
+       "\n",
+       "    .dataframe thead th {\n",
+       "        text-align: right;\n",
+       "    }\n",
+       "</style>\n",
+       "<table border=\"1\" class=\"dataframe\">\n",
+       "  <thead>\n",
+       "    <tr style=\"text-align: right;\">\n",
+       "      <th></th>\n",
+       "      <th>0</th>\n",
+       "      <th>1</th>\n",
+       "      <th>2</th>\n",
+       "      <th>song_names</th>\n",
+       "      <th>is_text</th>\n",
+       "    </tr>\n",
+       "  </thead>\n",
+       "  <tbody>\n",
+       "    <tr>\n",
+       "      <th>626</th>\n",
+       "      <td>7.839409</td>\n",
+       "      <td>-2.566209</td>\n",
+       "      <td>25.651859</td>\n",
+       "      <td>Wilson Pickett - Hey Joe.wav</td>\n",
+       "      <td>False</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>627</th>\n",
+       "      <td>25.030121</td>\n",
+       "      <td>10.993256</td>\n",
+       "      <td>2.521003</td>\n",
+       "      <td>Bülent Ortaçgil - Değirmenler.wav</td>\n",
+       "      <td>False</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>628</th>\n",
+       "      <td>-11.422283</td>\n",
+       "      <td>6.371774</td>\n",
+       "      <td>-33.630165</td>\n",
+       "      <td>Sufle - Köprüaltı.wav</td>\n",
+       "      <td>False</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>629</th>\n",
+       "      <td>-21.544903</td>\n",
+       "      <td>7.802368</td>\n",
+       "      <td>-11.589836</td>\n",
+       "      <td>Keane - Somewhere Only We Know.wav</td>\n",
+       "      <td>False</td>\n",
+       "    </tr>\n",
+       "    <tr>\n",
+       "      <th>630</th>\n",
+       "      <td>-32.042549</td>\n",
+       "      <td>-20.397856</td>\n",
+       "      <td>-20.176537</td>\n",
+       "      <td>text_embedding</td>\n",
+       "      <td>True</td>\n",
+       "    </tr>\n",
+       "  </tbody>\n",
+       "</table>\n",
+       "</div>"
+      ],
+      "text/plain": [
+       "             0          1          2                          song_names  \\\n",
+       "626   7.839409  -2.566209  25.651859        Wilson Pickett - Hey Joe.wav   \n",
+       "627  25.030121  10.993256   2.521003   Bülent Ortaçgil - Değirmenler.wav   \n",
+       "628 -11.422283   6.371774 -33.630165               Sufle - Köprüaltı.wav   \n",
+       "629 -21.544903   7.802368 -11.589836  Keane - Somewhere Only We Know.wav   \n",
+       "630 -32.042549 -20.397856 -20.176537                      text_embedding   \n",
+       "\n",
+       "     is_text  \n",
+       "626    False  \n",
+       "627    False  \n",
+       "628    False  \n",
+       "629    False  \n",
+       "630     True  "
+      ]
+     },
+     "execution_count": 46,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "df_vectors = pd.DataFrame(audio_components).assign(song_names=songs, is_text=lambda x: x[\"song_names\"] == \"text_embedding\")\n",
+    "df_vectors.tail()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 47,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "application/vnd.plotly.v1+json": {
+       "config": {
+        "plotlyServerURL": "https://plot.ly"
+       },
+       "data": [
+        {
+         "customdata": [
+          [
+           "Teoman - Hayalperest.wav"
+          ],
+          [
+           "Jason Mraz - I'm Yours.wav"
+          ],
+          [
+           "AC_DC - Back In Black.wav"
+          ],
+          [
+           "Pirates Of New Providence - Dead Man's Chest.wav"
+          ],
+          [
+           "B.B. King - Ghetto Woman.wav"
+          ],
+          [
+           "Elle King - Playing For Keeps.wav"
+          ],
+          [
+           "Tame Impala - The Less I Know The Better.wav"
+          ],
+          [
+           "Procol Harum - A Whiter Shade Of Pale.wav"
+          ],
+          [
+           "Duman - Yanibasimdan.wav"
+          ],
+          [
+           "Coldplay - Adventure of a Lifetime.wav"
+          ],
+          [
+           "Müslüm Gürses - Affet.wav"
+          ],
+          [
+           "Stereo Avenue - Mad World.wav"
+          ],
+          [
+           "Coldplay - Sparks.wav"
+          ],
+          [
+           "Santana - Day of Celebration.wav"
+          ],
+          [
+           "Queen - I Want To Break Free - Remastered 2011.wav"
+          ],
+          [
+           "Guns N' Roses - This I Love.wav"
+          ],
+          [
+           "123 - binalar.wav"
+          ],
+          [
+           "Serena Ryder - Stompa.wav"
+          ],
+          [
+           "Passion Pit - Take a Walk.wav"
+          ],
+          [
+           "Marvin Gaye - Got To Give It Up - Pt. 1.wav"
+          ],
+          [
+           "Erik Truffaz - Let Me Go ! (feat. Sophie Hunger).wav"
+          ],
+          [
+           "Blonde Redhead - For the Damaged Coda.wav"
+          ],
+          [
+           "Israel Kamakawiwo'ole - Somewhere Over The Rainbow_What A Wonderful World.wav"
+          ],
+          [
+           "Daft Punk - Lose Yourself to Dance (feat. Pharrell Williams).wav"
+          ],
+          [
+           "Zülfü Livaneli - Gözlerin.wav"
+          ],
+          [
+           "Adamlar - İnsanın Düştüğü Durumlar.wav"
+          ],
+          [
+           "Foghat - Louisiana Blues.wav"
+          ],
+          [
+           "Amy Winehouse - Mr Magic (Through The Smoke).wav"
+          ],
+          [
+           "Pinhani - Dünyadan Uzak.wav"
+          ],
+          [
+           "Queen - Don't Stop Me Now - 2011 Remaster.wav"
+          ],
+          [
+           "Gevende - Sanki.wav"
+          ],
+          [
+           "Mirkelam - Hatıralar.wav"
+          ],
+          [
+           "Hüseyin Bitmez - Nazende Sevgilim.wav"
+          ],
+          [
+           "Sertab Erener - Kumsalda.wav"
+          ],
+          [
+           "Eric Clapton - Ramblin' on My Mind.wav"
+          ],
+          [
+           "Aretha Franklin - At Last.wav"
+          ],
+          [
+           "Bob Marley & The Wailers - No Woman, No Cry - Live At The Lyceum, London_1975.wav"
+          ],
+          [
+           "Maroon 5 - Harder To Breathe.wav"
+          ],
+          [
+           "James Brown - It's A Man's, Man's, Man's World - Mono.wav"
+          ],
+          [
+           "Buddy Guy - Mustang Sally (feat. Jeff Beck).wav"
+          ],
+          [
+           "Cem Karaca - Resimdeki Gözyaslari (Apaslar).wav"
+          ],
+          [
+           "Adamlar - Yanmış İçinden.wav"
+          ],
+          [
+           "Adamlar - Ah Benim Hayatım.wav"
+          ],
+          [
+           "Adamlar - Tın Tın.wav"
+          ],
+          [
+           "Ali Kızıltuğ - Sen Gel Diyorsun - Öf Öf.wav"
+          ],
+          [
+           "Wes Montgomery - In Your Own Sweet Way.wav"
+          ],
+          [
+           "Duman - Elleri Ellerime.wav"
+          ],
+          [
+           "Bob Marley & The Wailers - Three Little Birds.wav"
+          ],
+          [
+           "Bee Gees - Stayin' Alive - Remastered Version.wav"
+          ],
+          [
+           "Grover Washington, Jr. - Just the Two of Us.wav"
+          ],
+          [
+           "Queen - Save Me - 2011 Remaster.wav"
+          ],
+          [
+           "Adele - Hometown Glory.wav"
+          ],
+          [
+           "Dido - White Flag.wav"
+          ],
+          [
+           "Adamlar - Utanmazsan Unutmam.wav"
+          ],
+          [
+           "Dire Straits - Sultans Of Swing.wav"
+          ],
+          [
+           "Bülent Ortaçgil - Bozburun.wav"
+          ],
+          [
+           "Sting - Shape Of My Heart.wav"
+          ],
+          [
+           "Barcelona Gipsy Klezmer Orchestra - Hasta Siempre, Comandante.wav"
+          ],
+          [
+           "Jimi Hendrix - Purple Haze.wav"
+          ],
+          [
+           "Craig David - Rise & Fall - feat. Sting.wav"
+          ],
+          [
+           "Los Naranjos - Cantarte a Ti.wav"
+          ],
+          [
+           "Cem Karaca - Beni Siz Delirttiniz.wav"
+          ],
+          [
+           "Bob Dylan - You Belong To Me.wav"
+          ],
+          [
+           "Milky Chance - Stolen Dance.wav"
+          ],
+          [
+           "Nirvana - The Man Who Sold The World.wav"
+          ],
+          [
+           "Pink Floyd - Wish You Were Here - 2011 Remastered Version.wav"
+          ],
+          [
+           "Emilíana Torrini - Dead Duck.wav"
+          ],
+          [
+           "Johnnie Taylor - Running Out Of Lies.wav"
+          ],
+          [
+           "No Doubt - Don't Speak.wav"
+          ],
+          [
+           "Bob Marley & The Wailers - African Herbsman.wav"
+          ],
+          [
+           "MIKA - Grace Kelly.wav"
+          ],
+          [
+           "Coldplay - True Love.wav"
+          ],
+          [
+           "Enya - Only Time.wav"
+          ],
+          [
+           "Cem Karaca - Sevda Kuşun Kanadında.wav"
+          ],
+          [
+           "Replikas - Yaş Elli.wav"
+          ],
+          [
+           "John Mayer - New Light.wav"
+          ],
+          [
+           "Elvis Presley - All Shook Up.wav"
+          ],
+          [
+           "emir taha - Huyu Suyu.wav"
+          ],
+          [
+           "Foo Fighters - The Sky Is A Neighborhood.wav"
+          ],
+          [
+           "Lucio Quarantotto - Time To Say Goodbye (Con Te Partirò).wav"
+          ],
+          [
+           "123 - so much to say.wav"
+          ],
+          [
+           "fun. - We Are Young (feat. Janelle Monáe).wav"
+          ],
+          [
+           "Fallulah - I Lay My Head.wav"
+          ],
+          [
+           "TOTO - Hold the Line.wav"
+          ],
+          [
+           "Ray Charles - Mess Around.wav"
+          ],
+          [
+           "Queen - Innuendo - Remastered 2011.wav"
+          ],
+          [
+           "Aerosmith - Walk This Way.wav"
+          ],
+          [
+           "Foster The People - Pumped Up Kicks.wav"
+          ],
+          [
+           "Santana - Primavera.wav"
+          ],
+          [
+           "David Dallas - Runnin'.wav"
+          ],
+          [
+           "Eric Clapton - Worried Life Blues - Live; 2015 Remaster.wav"
+          ],
+          [
+           "Coldplay - Warning Sign.wav"
+          ],
+          [
+           "Joe Cocker - Unchain My Heart.wav"
+          ],
+          [
+           "Nur Yoldas - Sultan-ı Yegah.wav"
+          ],
+          [
+           "Nil Karaibrahimgil - Kanatlarım Var Ruhumda.wav"
+          ],
+          [
+           "Wilson Pickett - Hey Joe.wav"
+          ],
+          [
+           "Bülent Ortaçgil - Değirmenler.wav"
+          ],
+          [
+           "Sufle - Köprüaltı.wav"
+          ],
+          [
+           "Keane - Somewhere Only We Know.wav"
+          ]
+         ],
+         "hovertemplate": "is_text=False<br>0=%{x}<br>1=%{y}<br>2=%{z}<br>song_names=%{customdata[0]}<extra></extra>",
+         "legendgroup": "False",
+         "marker": {
+          "color": "#636efa",
+          "opacity": 0.8,
+          "symbol": "circle"
+         },
+         "mode": "markers",
+         "name": "False",
+         "scene": "scene",
+         "showlegend": true,
+         "type": "scatter3d",
+         "x": [
+          3.960073471069336,
+          6.63026237487793,
+          6.366035461425781,
+          35.9576301574707,
+          25.72879409790039,
+          -23.749359130859375,
+          -18.918292999267578,
+          20.815343856811523,
+          3.548778533935547,
+          -9.880792617797852,
+          -1.626604676246643,
+          -25.692523956298828,
+          -9.009389877319336,
+          21.91187858581543,
+          -8.398816108703613,
+          -13.183920860290527,
+          -16.33573341369629,
+          -20.739084243774414,
+          -16.058250427246094,
+          -5.901383876800537,
+          -29.334606170654297,
+          -21.556638717651367,
+          -32.477176666259766,
+          -10.107562065124512,
+          2.0257608890533447,
+          3.539175510406494,
+          9.495028495788574,
+          -8.56347370147705,
+          2.9173169136047363,
+          12.427820205688477,
+          22.154857635498047,
+          -9.044797897338867,
+          24.114742279052734,
+          -25.635881423950195,
+          19.347545623779297,
+          -19.076709747314453,
+          17.552734375,
+          -12.944130897521973,
+          29.22037696838379,
+          5.216680526733398,
+          4.647155284881592,
+          -5.739335536956787,
+          1.2586489915847778,
+          9.793363571166992,
+          22.0106143951416,
+          28.065549850463867,
+          13.323224067687988,
+          6.095829010009766,
+          -8.249640464782715,
+          5.1530609130859375,
+          -22.9552001953125,
+          -25.220529556274414,
+          -17.555011749267578,
+          26.815658569335938,
+          6.869404315948486,
+          34.0200080871582,
+          -10.640585899353027,
+          -29.66428565979004,
+          16.96628761291504,
+          -12.059541702270508,
+          18.312807083129883,
+          18.12636947631836,
+          -0.4769120514392853,
+          11.270349502563477,
+          15.4771089553833,
+          -9.384946823120117,
+          10.876140594482422,
+          -3.506535053253174,
+          -33.04511260986328,
+          21.82001304626465,
+          -20.24302101135254,
+          -21.872957229614258,
+          -26.808494567871094,
+          -4.571284770965576,
+          4.80881929397583,
+          16.436382293701172,
+          27.422374725341797,
+          -26.98200035095215,
+          -19.272430419921875,
+          -20.390697479248047,
+          -32.17781066894531,
+          -30.004587173461914,
+          -23.986347198486328,
+          -3.3886325359344482,
+          31.404930114746094,
+          -6.3269734382629395,
+          12.990680694580078,
+          3.9187815189361572,
+          4.270918369293213,
+          -28.737621307373047,
+          26.91798210144043,
+          -4.584568023681641,
+          25.394004821777344,
+          -27.019210815429688,
+          -24.920318603515625,
+          7.839408874511719,
+          25.030120849609375,
+          -11.422283172607422,
+          -21.544902801513672
+         ],
+         "y": [
+          -6.303861141204834,
+          15.232268333435059,
+          -18.639732360839844,
+          16.955612182617188,
+          -4.059649467468262,
+          5.14973258972168,
+          -2.5839288234710693,
+          0.775934636592865,
+          -6.10317325592041,
+          0.692775309085846,
+          -1.615648627281189,
+          4.755437850952148,
+          -0.393994003534317,
+          -19.156991958618164,
+          22.87925910949707,
+          11.058145523071289,
+          3.0950820446014404,
+          5.530603885650635,
+          -3.080402374267578,
+          13.579002380371094,
+          -4.78978157043457,
+          7.763420104980469,
+          -7.65438985824585,
+          23.316055297851562,
+          24.696044921875,
+          3.7123682498931885,
+          -21.037315368652344,
+          13.44579029083252,
+          -2.354578733444214,
+          -26.491195678710938,
+          -18.70002555847168,
+          13.3438720703125,
+          -2.367952823638916,
+          24.548198699951172,
+          -17.782995223999023,
+          14.160982131958008,
+          22.661109924316406,
+          -15.065990447998047,
+          14.719420433044434,
+          -16.994403839111328,
+          -2.528127431869507,
+          -9.89771556854248,
+          -6.246743202209473,
+          25.03219985961914,
+          -1.4081578254699707,
+          -2.3774948120117188,
+          -27.25217056274414,
+          23.652070999145508,
+          16.401748657226562,
+          20.747957229614258,
+          14.122920036315918,
+          15.15066909790039,
+          19.500532150268555,
+          3.8085296154022217,
+          -0.638141393661499,
+          -6.328094482421875,
+          2.543947696685791,
+          24.652851104736328,
+          -14.238359451293945,
+          19.87712287902832,
+          15.849624633789062,
+          9.969474792480469,
+          11.297757148742676,
+          -18.961462020874023,
+          -23.89181137084961,
+          6.338683128356934,
+          -6.532224178314209,
+          8.980716705322266,
+          10.482304573059082,
+          19.634340286254883,
+          -19.921180725097656,
+          -18.082565307617188,
+          -3.7091286182403564,
+          21.601825714111328,
+          -21.571517944335938,
+          -11.576738357543945,
+          19.804643630981445,
+          -2.666598081588745,
+          -23.158870697021484,
+          -8.012092590332031,
+          14.015975952148438,
+          8.013833999633789,
+          13.031487464904785,
+          18.049388885498047,
+          7.4357523918151855,
+          15.054443359375,
+          -21.247608184814453,
+          3.2007644176483154,
+          25.249967575073242,
+          -8.995502471923828,
+          -2.2580485343933105,
+          -9.61690616607666,
+          22.74266242980957,
+          27.359724044799805,
+          16.42228126525879,
+          -2.566208839416504,
+          10.993255615234375,
+          6.371773719787598,
+          7.8023681640625
+         ],
+         "z": [
+          -4.585926055908203,
+          7.6198248863220215,
+          11.197002410888672,
+          10.61091423034668,
+          -0.16559484601020813,
+          15.873065948486328,
+          6.092303276062012,
+          13.278153419494629,
+          -13.814813613891602,
+          -32.1518669128418,
+          -2.846285104751587,
+          -24.60615348815918,
+          -14.583600044250488,
+          -5.089763641357422,
+          17.827316284179688,
+          -8.723794937133789,
+          -0.46522191166877747,
+          22.079736709594727,
+          -27.905847549438477,
+          26.410999298095703,
+          -24.432289123535156,
+          -29.81077766418457,
+          -24.935848236083984,
+          14.596731185913086,
+          -11.877199172973633,
+          -23.376184463500977,
+          2.7458457946777344,
+          19.448259353637695,
+          -7.996399879455566,
+          9.06856918334961,
+          1.2749879360198975,
+          -12.594420433044434,
+          -22.486719131469727,
+          2.873701572418213,
+          -24.027612686157227,
+          2.2689332962036133,
+          21.957611083984375,
+          13.603313446044922,
+          16.907148361206055,
+          21.468961715698242,
+          23.266664505004883,
+          1.334670066833496,
+          -9.798734664916992,
+          5.956639766693115,
+          -24.757957458496094,
+          -16.810287475585938,
+          -0.19127635657787323,
+          17.18698501586914,
+          23.69522476196289,
+          11.511524200439453,
+          -16.232465744018555,
+          -11.024603843688965,
+          0.2891804277896881,
+          4.686855792999268,
+          4.955519676208496,
+          -11.170580863952637,
+          -9.173028945922852,
+          9.23276424407959,
+          20.439125061035156,
+          -21.887910842895508,
+          -25.068744659423828,
+          -2.6079728603363037,
+          -19.979209899902344,
+          -16.800277709960938,
+          -7.8158345222473145,
+          -20.36083984375,
+          -13.97440242767334,
+          13.297832489013672,
+          6.4031662940979,
+          16.68050193786621,
+          -7.421001434326172,
+          -20.746915817260742,
+          -17.661420822143555,
+          -16.111785888671875,
+          -6.790570259094238,
+          -10.68458080291748,
+          4.448904991149902,
+          5.806682586669922,
+          -18.60723114013672,
+          -32.03889465332031,
+          -3.2083699703216553,
+          8.138893127441406,
+          10.776344299316406,
+          15.558555603027344,
+          14.651917457580566,
+          -17.637117385864258,
+          20.528764724731445,
+          16.84113883972168,
+          10.138716697692871,
+          6.520318984985352,
+          8.3370361328125,
+          -3.437776565551758,
+          -6.115707874298096,
+          -4.344021797180176,
+          -2.9191675186157227,
+          25.651859283447266,
+          2.521003484725952,
+          -33.630165100097656,
+          -11.589836120605469
+         ]
+        },
+        {
+         "customdata": [
+          [
+           "text_embedding"
+          ]
+         ],
+         "hovertemplate": "is_text=True<br>0=%{x}<br>1=%{y}<br>2=%{z}<br>song_names=%{customdata[0]}<extra></extra>",
+         "legendgroup": "True",
+         "marker": {
+          "color": "#EF553B",
+          "opacity": 0.8,
+          "symbol": "circle"
+         },
+         "mode": "markers",
+         "name": "True",
+         "scene": "scene",
+         "showlegend": true,
+         "type": "scatter3d",
+         "x": [
+          -32.04254913330078
+         ],
+         "y": [
+          -20.397855758666992
+         ],
+         "z": [
+          -20.176536560058594
+         ]
+        }
+       ],
+       "layout": {
+        "legend": {
+         "title": {
+          "text": "is_text"
+         },
+         "tracegroupgap": 0
+        },
+        "margin": {
+         "t": 60
+        },
+        "scene": {
+         "domain": {
+          "x": [
+           0,
+           1
+          ],
+          "y": [
+           0,
+           1
+          ]
+         },
+         "xaxis": {
+          "title": {
+           "text": "0"
+          }
+         },
+         "yaxis": {
+          "title": {
+           "text": "1"
+          }
+         },
+         "zaxis": {
+          "title": {
+           "text": "2"
+          }
+         }
+        },
+        "template": {
+         "data": {
+          "bar": [
+           {
+            "error_x": {
+             "color": "#2a3f5f"
+            },
+            "error_y": {
+             "color": "#2a3f5f"
+            },
+            "marker": {
+             "line": {
+              "color": "#E5ECF6",
+              "width": 0.5
+             },
+             "pattern": {
+              "fillmode": "overlay",
+              "size": 10,
+              "solidity": 0.2
+             }
+            },
+            "type": "bar"
+           }
+          ],
+          "barpolar": [
+           {
+            "marker": {
+             "line": {
+              "color": "#E5ECF6",
+              "width": 0.5
+             },
+             "pattern": {
+              "fillmode": "overlay",
+              "size": 10,
+              "solidity": 0.2
+             }
+            },
+            "type": "barpolar"
+           }
+          ],
+          "carpet": [
+           {
+            "aaxis": {
+             "endlinecolor": "#2a3f5f",
+             "gridcolor": "white",
+             "linecolor": "white",
+             "minorgridcolor": "white",
+             "startlinecolor": "#2a3f5f"
+            },
+            "baxis": {
+             "endlinecolor": "#2a3f5f",
+             "gridcolor": "white",
+             "linecolor": "white",
+             "minorgridcolor": "white",
+             "startlinecolor": "#2a3f5f"
+            },
+            "type": "carpet"
+           }
+          ],
+          "choropleth": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "type": "choropleth"
+           }
+          ],
+          "contour": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "contour"
+           }
+          ],
+          "contourcarpet": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "type": "contourcarpet"
+           }
+          ],
+          "heatmap": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "heatmap"
+           }
+          ],
+          "heatmapgl": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "heatmapgl"
+           }
+          ],
+          "histogram": [
+           {
+            "marker": {
+             "pattern": {
+              "fillmode": "overlay",
+              "size": 10,
+              "solidity": 0.2
+             }
+            },
+            "type": "histogram"
+           }
+          ],
+          "histogram2d": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "histogram2d"
+           }
+          ],
+          "histogram2dcontour": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "histogram2dcontour"
+           }
+          ],
+          "mesh3d": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "type": "mesh3d"
+           }
+          ],
+          "parcoords": [
+           {
+            "line": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "parcoords"
+           }
+          ],
+          "pie": [
+           {
+            "automargin": true,
+            "type": "pie"
+           }
+          ],
+          "scatter": [
+           {
+            "fillpattern": {
+             "fillmode": "overlay",
+             "size": 10,
+             "solidity": 0.2
+            },
+            "type": "scatter"
+           }
+          ],
+          "scatter3d": [
+           {
+            "line": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scatter3d"
+           }
+          ],
+          "scattercarpet": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scattercarpet"
+           }
+          ],
+          "scattergeo": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scattergeo"
+           }
+          ],
+          "scattergl": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scattergl"
+           }
+          ],
+          "scattermapbox": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scattermapbox"
+           }
+          ],
+          "scatterpolar": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scatterpolar"
+           }
+          ],
+          "scatterpolargl": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scatterpolargl"
+           }
+          ],
+          "scatterternary": [
+           {
+            "marker": {
+             "colorbar": {
+              "outlinewidth": 0,
+              "ticks": ""
+             }
+            },
+            "type": "scatterternary"
+           }
+          ],
+          "surface": [
+           {
+            "colorbar": {
+             "outlinewidth": 0,
+             "ticks": ""
+            },
+            "colorscale": [
+             [
+              0,
+              "#0d0887"
+             ],
+             [
+              0.1111111111111111,
+              "#46039f"
+             ],
+             [
+              0.2222222222222222,
+              "#7201a8"
+             ],
+             [
+              0.3333333333333333,
+              "#9c179e"
+             ],
+             [
+              0.4444444444444444,
+              "#bd3786"
+             ],
+             [
+              0.5555555555555556,
+              "#d8576b"
+             ],
+             [
+              0.6666666666666666,
+              "#ed7953"
+             ],
+             [
+              0.7777777777777778,
+              "#fb9f3a"
+             ],
+             [
+              0.8888888888888888,
+              "#fdca26"
+             ],
+             [
+              1,
+              "#f0f921"
+             ]
+            ],
+            "type": "surface"
+           }
+          ],
+          "table": [
+           {
+            "cells": {
+             "fill": {
+              "color": "#EBF0F8"
+             },
+             "line": {
+              "color": "white"
+             }
+            },
+            "header": {
+             "fill": {
+              "color": "#C8D4E3"
+             },
+             "line": {
+              "color": "white"
+             }
+            },
+            "type": "table"
+           }
+          ]
+         },
+         "layout": {
+          "annotationdefaults": {
+           "arrowcolor": "#2a3f5f",
+           "arrowhead": 0,
+           "arrowwidth": 1
+          },
+          "autotypenumbers": "strict",
+          "coloraxis": {
+           "colorbar": {
+            "outlinewidth": 0,
+            "ticks": ""
+           }
+          },
+          "colorscale": {
+           "diverging": [
+            [
+             0,
+             "#8e0152"
+            ],
+            [
+             0.1,
+             "#c51b7d"
+            ],
+            [
+             0.2,
+             "#de77ae"
+            ],
+            [
+             0.3,
+             "#f1b6da"
+            ],
+            [
+             0.4,
+             "#fde0ef"
+            ],
+            [
+             0.5,
+             "#f7f7f7"
+            ],
+            [
+             0.6,
+             "#e6f5d0"
+            ],
+            [
+             0.7,
+             "#b8e186"
+            ],
+            [
+             0.8,
+             "#7fbc41"
+            ],
+            [
+             0.9,
+             "#4d9221"
+            ],
+            [
+             1,
+             "#276419"
+            ]
+           ],
+           "sequential": [
+            [
+             0,
+             "#0d0887"
+            ],
+            [
+             0.1111111111111111,
+             "#46039f"
+            ],
+            [
+             0.2222222222222222,
+             "#7201a8"
+            ],
+            [
+             0.3333333333333333,
+             "#9c179e"
+            ],
+            [
+             0.4444444444444444,
+             "#bd3786"
+            ],
+            [
+             0.5555555555555556,
+             "#d8576b"
+            ],
+            [
+             0.6666666666666666,
+             "#ed7953"
+            ],
+            [
+             0.7777777777777778,
+             "#fb9f3a"
+            ],
+            [
+             0.8888888888888888,
+             "#fdca26"
+            ],
+            [
+             1,
+             "#f0f921"
+            ]
+           ],
+           "sequentialminus": [
+            [
+             0,
+             "#0d0887"
+            ],
+            [
+             0.1111111111111111,
+             "#46039f"
+            ],
+            [
+             0.2222222222222222,
+             "#7201a8"
+            ],
+            [
+             0.3333333333333333,
+             "#9c179e"
+            ],
+            [
+             0.4444444444444444,
+             "#bd3786"
+            ],
+            [
+             0.5555555555555556,
+             "#d8576b"
+            ],
+            [
+             0.6666666666666666,
+             "#ed7953"
+            ],
+            [
+             0.7777777777777778,
+             "#fb9f3a"
+            ],
+            [
+             0.8888888888888888,
+             "#fdca26"
+            ],
+            [
+             1,
+             "#f0f921"
+            ]
+           ]
+          },
+          "colorway": [
+           "#636efa",
+           "#EF553B",
+           "#00cc96",
+           "#ab63fa",
+           "#FFA15A",
+           "#19d3f3",
+           "#FF6692",
+           "#B6E880",
+           "#FF97FF",
+           "#FECB52"
+          ],
+          "font": {
+           "color": "#2a3f5f"
+          },
+          "geo": {
+           "bgcolor": "white",
+           "lakecolor": "white",
+           "landcolor": "#E5ECF6",
+           "showlakes": true,
+           "showland": true,
+           "subunitcolor": "white"
+          },
+          "hoverlabel": {
+           "align": "left"
+          },
+          "hovermode": "closest",
+          "mapbox": {
+           "style": "light"
+          },
+          "paper_bgcolor": "white",
+          "plot_bgcolor": "#E5ECF6",
+          "polar": {
+           "angularaxis": {
+            "gridcolor": "white",
+            "linecolor": "white",
+            "ticks": ""
+           },
+           "bgcolor": "#E5ECF6",
+           "radialaxis": {
+            "gridcolor": "white",
+            "linecolor": "white",
+            "ticks": ""
+           }
+          },
+          "scene": {
+           "xaxis": {
+            "backgroundcolor": "#E5ECF6",
+            "gridcolor": "white",
+            "gridwidth": 2,
+            "linecolor": "white",
+            "showbackground": true,
+            "ticks": "",
+            "zerolinecolor": "white"
+           },
+           "yaxis": {
+            "backgroundcolor": "#E5ECF6",
+            "gridcolor": "white",
+            "gridwidth": 2,
+            "linecolor": "white",
+            "showbackground": true,
+            "ticks": "",
+            "zerolinecolor": "white"
+           },
+           "zaxis": {
+            "backgroundcolor": "#E5ECF6",
+            "gridcolor": "white",
+            "gridwidth": 2,
+            "linecolor": "white",
+            "showbackground": true,
+            "ticks": "",
+            "zerolinecolor": "white"
+           }
+          },
+          "shapedefaults": {
+           "line": {
+            "color": "#2a3f5f"
+           }
+          },
+          "ternary": {
+           "aaxis": {
+            "gridcolor": "white",
+            "linecolor": "white",
+            "ticks": ""
+           },
+           "baxis": {
+            "gridcolor": "white",
+            "linecolor": "white",
+            "ticks": ""
+           },
+           "bgcolor": "#E5ECF6",
+           "caxis": {
+            "gridcolor": "white",
+            "linecolor": "white",
+            "ticks": ""
+           }
+          },
+          "title": {
+           "x": 0.05
+          },
+          "xaxis": {
+           "automargin": true,
+           "gridcolor": "white",
+           "linecolor": "white",
+           "ticks": "",
+           "title": {
+            "standoff": 15
+           },
+           "zerolinecolor": "white",
+           "zerolinewidth": 2
+          },
+          "yaxis": {
+           "automargin": true,
+           "gridcolor": "white",
+           "linecolor": "white",
+           "ticks": "",
+           "title": {
+            "standoff": 15
+           },
+           "zerolinecolor": "white",
+           "zerolinewidth": 2
+          }
+         }
+        }
+       }
+      }
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
+   "source": [
+    "fig = px.scatter_3d(x=0, y=1, z=2, data_frame=df_vectors.tail(100), hover_data={\"song_names\": True}, color=\"is_text\", opacity=0.8)\n",
+    "fig.show()\n"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "playlist-curator",
+   "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.10.13"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

orchestrate_audio_data.py

@@ -3,6 +3,6 @@ from src.data.get_yt_links import collect_youtube_links
 from src.data.pytuber import start_download_process
 
 if __name__ == "__main__":
-    list_personal_saved_tracks()
-    collect_youtube_links()
+    # list_personal_saved_tracks()
+    # collect_youtube_links()
     start_download_process()

requirements.txt

@@ -86,4 +86,5 @@ validators==0.22.0
 wandb==0.16.0
 webdataset==0.2.77
 wget==3.2
-youtube-search-python==1.6.6
+youtube-search-python==1.6.6
+python-dotenv

src/data/pytuber.py

@@ -20,16 +20,25 @@ def download_mp3(link, download_path, track_full_name):
 
         new_file = track_full_name + '.wav'
         os.rename(download_path.joinpath(mp3.default_filename), data_dir.joinpath(new_file))
-    except AgeRestrictedError:
-        pass
+        return data_dir.joinpath(new_file)
 
+    except AgeRestrictedError:
+        return ""
+    
 
 def start_download_process():
     input_data = read_youtube_data()
+    reformatted_input_data = []
     done_pieces = os.listdir(ProjectPaths.DATA_DIR.joinpath("audio"))
     for i in tqdm(input_data):
         link = i["link"]
         full_name = f'{i["artist_name"]} - {i["track_name"]}'.replace("/", "_")
         if full_name + ".wav" in done_pieces:
             continue
-        download_mp3(link, full_name)
+        path = download_mp3(link, ProjectPaths.DATA_DIR.joinpath("audio") ,full_name)
+        new_data = i.copy()
+        new_data.update({"file_path": str(path)})
+        reformatted_input_data.append(new_data)
+        
+    with open(ProjectPaths.DATA_DIR.joinpath("json", "final_track_data.json"), "w") as wr:
+        json.dump(reformatted_input_data, wr, indent=4)

src/data/spotify.py

@@ -10,14 +10,17 @@ def list_personal_saved_tracks():
     sp = spotipy.Spotify(auth_manager=auth)
 
     tracks = []
+    downloaded_tracks = []
     offset_count = 0
     for _ in range(50):
         results = sp.current_user_saved_tracks(limit=50, offset=offset_count)
         for idx, item in enumerate(results['items']):
             track = item['track']
             data = {"artist": track['artists'][0]['name'], "track": track['name']}
-            tracks.append(data)
-            print(idx, track['artists'][0]['name'], " - ", track['name'])
+            if track not in downloaded_tracks:
+                tracks.append(data)
+                downloaded_tracks.append(track)
+                print(idx, track['artists'][0]['name'], " - ", track['name'])
         offset_count += 50
 
     with open(ProjectPaths.DATA_DIR.joinpath("json", "saved_tracks.json"), "w", encoding="UTF-8") as wr:

src/laion_clap/inference.py

@@ -38,4 +38,4 @@ class AudioEncoder(laion_clap.CLAP_Module):
     def extract_text_representation(self, text):
         text_data = [text]
         text_embed = self.get_text_embedding(text_data)
-        return text_embed
+        return text_embed