first version of prompt with emotions

app.py

@@ -7,7 +7,7 @@ from langchain.prompts import PromptTemplate
 
 load_dotenv()
 import os
-
+import json
 from langchain.chat_models import ChatOpenAI
 from langchain.embeddings.openai import OpenAIEmbeddings
 
@@ -29,17 +29,30 @@ def init():
     )
 
     prompt = PromptTemplate(
-        input_variables=["content"],
-        template=Path("prompts/bot.prompt").read_text(),
-    )
+    input_variables=["songs", "user_input"],
+    template=Path("prompts/bot_with_summary.prompt").read_text(),
+)
 
     llm = ChatOpenAI(temperature=0.7)
 
     chain = LLMChain(llm=llm, prompt=prompt)
 
-    return db, chain
+    with open("data/emotions_with_spotify_url.json", "r") as f:
+        data = json.load(f)
+    
+    movies_and_names_to_songs = {}
+
+    songs_str = ""
 
-db, chain = init()
+    for movie, songs in data.items():
+        for song in songs:
+            movie_and_name = f"{movie};{song['name'].lower()}"
+            songs_str += f"{movie_and_name}:{song['text']}\n"
+            movies_and_names_to_songs[movie_and_name] = song
+
+    return db, chain, movies_and_names_to_songs, songs_str
+
+db, chain, movies_and_names_to_songs, songs_str = init()
 
 st.title("Disney song for you")
 
@@ -52,20 +65,14 @@ clicked = st.button("Click me")
 placeholder_emotions = st.empty()
 placeholder = st.empty()
 
-def get_emotions(user_input):
-    emotions = chain.run(content=user_input)
-    print(f"Emotions: {emotions}")
-    matches = db.similarity_search_with_score(emotions, distance_metric="cos")
-    print(matches)
-    doc, score = matches[0]
-    iframes_html = ""
-    with placeholder_emotions:
-        st.write(emotions)
+def get_emotions(songs_str, user_input):
+    key = chain.run(songs=songs_str, user_input=user_input)
+    doc = movies_and_names_to_songs[key]
+    print(f"Reply: {key}")
     with placeholder:
-        embed_url = doc.metadata["embed_url"]
+        embed_url = doc["embed_url"]
         iframe_html = f'<iframe src="{embed_url}" style="border:0"> </iframe>'
         st.components.v1.html(f"<div style='display:flex;flex-direction:column'>{iframe_html}</div>")
 
-
 if clicked:
-    get_emotions(text_input)
+    get_emotions(songs_str, text_input)

data/emotions_with_spotify_url.json

@@ -0,0 +1,495 @@
+{
+    "Aladdin": [
+        {
+            "name": "Friend Like Me",
+            "text": "excitement, happiness, humor, friendliness, generosity, willingness, eagerness, confidence",
+            "embed_url": "https://open.spotify.com/embed/track/5f2TWu6R2YYCJtLQ0fP78H?utm_source=generator"
+        },
+        {
+            "name": "Arabian Nights",
+            "text": "nostalgic, adventurous, exotic, intense, romantic, mysterious, whimsical, passionate",
+            "embed_url": "https://open.spotify.com/embed/track/0CKmN3Wwk8W4zjU0pqq2cv?utm_source=generator"
+        },
+        {
+            "name": "A Whole New World",
+            "text": "wonder, joy, excitement, adventure, romance, freedom, happiness, awe",
+            "embed_url": "https://open.spotify.com/embed/track/1hwdPQtFHISvZ9SXMkNrIK?utm_source=generator"
+        },
+        {
+            "name": "One Jump Ahead",
+            "text": "Thrill, fear, desperation, humor, defiance, friendship, hunger, rebellion.",
+            "embed_url": "https://open.spotify.com/embed/track/4wN8Ov3kPZdkJ8XcYxYUGz?utm_source=generator"
+        }
+    ],
+    "Aladdin (live-action)": [
+        {
+            "name": "A Whole New World",
+            "text": "nostalgia, wonder, excitement, joy, romance, adventure, hope, inspiration",
+            "embed_url": "https://open.spotify.com/embed/track/1hwdPQtFHISvZ9SXMkNrIK?utm_source=generator"
+        },
+        {
+            "name": "A Whole New World (End Title)",
+            "text": "joy, wonder, excitement, adventure, love, freedom, exhilaration, enchantment",
+            "embed_url": "https://open.spotify.com/embed/track/4U9CF27YlwHRtUGKUzhUTT?utm_source=generator"
+        },
+        {
+            "name": "Friend Like Me",
+            "text": "excitement, confidence, generosity, humor, friendliness, helpfulness, empowerment, joyfulness",
+            "embed_url": "https://open.spotify.com/embed/track/5f2TWu6R2YYCJtLQ0fP78H?utm_source=generator"
+        },
+        {
+            "name": "One Jump Ahead",
+            "text": "excitement, fear, desperation, humor, defiance, loneliness, camaraderie, determination",
+            "embed_url": "https://open.spotify.com/embed/track/4wN8Ov3kPZdkJ8XcYxYUGz?utm_source=generator"
+        },
+        {
+            "name": "Speechless (Full)",
+            "text": "determination, empowerment, resilience, strength, defiance, courage, passion, perseverance",
+            "embed_url": "https://open.spotify.com/embed/track/0XPsOSYzDJZJArevQNm2AR?utm_source=generator"
+        }
+    ],
+    "Aladdin: The Return of Jafar": [
+        {
+            "name": "Arabian Nights",
+            "text": "excitement, wonder, adventure, awe, joy, fascination, thrill, curiosity",
+            "embed_url": "https://open.spotify.com/embed/track/0CKmN3Wwk8W4zjU0pqq2cv?utm_source=generator"
+        }
+    ],
+    "Beauty and the Beast": [
+        {
+            "name": "Belle",
+            "text": "calm, content, curious, dreamy, happy, hopeful, romantic, whimsical",
+            "embed_url": "https://open.spotify.com/embed/track/1cPfKp9ThNZ1fez9itmUMN?utm_source=generator"
+        },
+        {
+            "name": "Be Our Guest",
+            "text": "joy, excitement, hospitality, gratitude, entertainment, nostalgia, comfort, satisfaction",
+            "embed_url": "https://open.spotify.com/embed/track/2F8LPJgSVTJxukoParTyBD?utm_source=generator"
+        }
+    ],
+    "Brave": [
+        {
+            "name": "Touch the Sky",
+            "text": "hopeful, adventurous, determined, inspired, free-spirited, empowered, awe-struck, exhilarated",
+            "embed_url": "https://open.spotify.com/embed/track/5yyqx4brn6Bm9U1Rj9ENnz?utm_source=generator"
+        }
+    ],
+    "Coco": [
+        {
+            "name": "Remember Me (D\u00fao)",
+            "text": "nostalgia, love, longing, sadness, hope, comfort, devotion, perseverance",
+            "embed_url": "https://open.spotify.com/embed/track/3pCgIUkmRHKydu9HE03QKY?utm_source=generator"
+        },
+        {
+            "name": "Remember Me (Ernesto de la Cruz)",
+            "text": "longing, separation, love, devotion, sadness, comfort, hope, reminiscence",
+            "embed_url": "https://open.spotify.com/embed/track/4ljlAxfaEjO4Q4g9FMtVj4?utm_source=generator"
+        },
+        {
+            "name": "The World Es Mi Familia",
+            "text": "joy, happiness, excitement, love, gratitude, belonging, unity, celebration",
+            "embed_url": "https://open.spotify.com/embed/track/7FOS31xq9jv4cXWtp6Jnzw?utm_source=generator"
+        },
+        {
+            "name": "Un Poco Loco",
+            "text": "confusion, love, amusement, uncertainty, joy, surprise, frustration, whimsy",
+            "embed_url": "https://open.spotify.com/embed/track/0OFknyqxmSQ42SoKxWVTok?utm_source=generator"
+        }
+    ],
+    "Descendants 3": [
+        {
+            "name": "Dig a Little Deeper",
+            "text": "Hope, Acceptance, Empowerment, Encouragement, Determination, Inspiration, Resilience, Confidence.",
+            "embed_url": "https://open.spotify.com/embed/track/7kh64k3P9Fk4EsA6vOdwmj?utm_source=generator"
+        }
+    ],
+    "Encanto": [
+        {
+            "name": "The Family Madrigal",
+            "text": "joy, excitement, wonder, confusion, love, pride, nostalgia, humor",
+            "embed_url": "https://open.spotify.com/embed/track/4b1yxSdlumA8N4fEk4UOZp?utm_source=generator"
+        },
+        {
+            "name": "Surface Pressure",
+            "text": "strength, confidence, anxiety, pressure, worry, fear, determination, resilience",
+            "embed_url": "https://open.spotify.com/embed/track/760jhRscwGbIIe1m1IIQpU?utm_source=generator"
+        },
+        {
+            "name": "Two Oruguitas",
+            "text": "love, yearning, hunger, fear, anticipation, change, trust, growth",
+            "embed_url": "https://open.spotify.com/embed/track/6jPVyjPqp2BNUJnDSwm8tK?utm_source=generator"
+        },
+        {
+            "name": "We Don't Talk About Bruno",
+            "text": "joy, fear, confusion, anxiety, frustration, curiosity, anger, humor",
+            "embed_url": "https://open.spotify.com/embed/track/52xJxFP6TqMuO4Yt0eOkMz?utm_source=generator"
+        },
+        {
+            "name": "What Else Can I Do?",
+            "text": "unexpected, beautiful, practiced, carnivorous, new, sick of pretty, rising, changing minds",
+            "embed_url": "https://open.spotify.com/embed/track/3XoYqtiWHhsk59frZupImG?utm_source=generator"
+        },
+        {
+            "name": "Waiting on a Miracle",
+            "text": "upset, mad, regret, sad, fine, longing, hopeful, impatient",
+            "embed_url": "https://open.spotify.com/embed/track/3oRW9ZGPRbLRMneQ5lwflt?utm_source=generator"
+        }
+    ],
+    "Enchanted": [
+        {
+            "name": "Happy Working Song",
+            "text": "joy, cheerfulness, contentment, enthusiasm, satisfaction, excitement, pleasure, happiness",
+            "embed_url": "https://open.spotify.com/embed/track/7G061Oqw7NXFr1NDTpXoI4?utm_source=generator"
+        },
+        {
+            "name": "So Close",
+            "text": "romantic, hopeful, longing, joy, fear, sadness, content, nostalgia",
+            "embed_url": "https://open.spotify.com/embed/track/3XFkhxO5jo2hBOWNzIQ9UB?utm_source=generator"
+        },
+        {
+            "name": "That's How You Know",
+            "text": "love, happiness, affection, devotion, passion, adoration, appreciation, romance",
+            "embed_url": "https://open.spotify.com/embed/track/1OzSfjFW08DTD51XoNnog7?utm_source=generator"
+        }
+    ],
+    "Frozen": [
+        {
+            "name": "Do You Want to Build a Snowman?",
+            "text": "loneliness, longing, sadness, nostalgia, hope, perseverance, friendship, playfulness",
+            "embed_url": "https://open.spotify.com/embed/track/2yi7HZrBOC4bMUSTcs4VK6?utm_source=generator"
+        },
+        {
+            "name": "For the First Time in Forever",
+            "text": "excited, hopeful, nervous, happy, anxious, optimistic, romantic, determined",
+            "embed_url": "https://open.spotify.com/embed/track/70b5Sq3ePOu3Gqg0hjlOtR?utm_source=generator"
+        },
+        {
+            "name": "In Summer",
+            "text": "joy, anticipation, excitement, happiness, nostalgia, humor, contentment, playfulness",
+            "embed_url": "https://open.spotify.com/embed/track/7bG6SQBGZthPDG5QJL5Gf7?utm_source=generator"
+        },
+        {
+            "name": "Let it Go",
+            "text": "isolation, determination, freedom, empowerment, resilience, confidence, defiance, happiness",
+            "embed_url": "https://open.spotify.com/embed/track/0qcr5FMsEO85NAQjrlDRKo?utm_source=generator"
+        },
+        {
+            "name": "Love is an Open Door",
+            "text": "joy, excitement, love, happiness, contentment, connection, optimism, anticipation",
+            "embed_url": "https://open.spotify.com/embed/track/68jg6wseZjKSJ0HUMjNS7C?utm_source=generator"
+        }
+    ],
+    "Frozen 2": [
+        {
+            "name": "All Is Found",
+            "text": "nostalgia, safety, discovery, fear, magic, bravery, loss, hope.",
+            "embed_url": "https://open.spotify.com/embed/track/5EeQQ8BVJTRkp1AIKJILGY?utm_source=generator"
+        },
+        {
+            "name": "Into the Unknown",
+            "text": "confusion, fear, hesitation, loneliness, longing, uncertainty, curiosity, determination",
+            "embed_url": "https://open.spotify.com/embed/track/421eObjg0DTm2qajJl5OJm?utm_source=generator"
+        },
+        {
+            "name": "Lost in the Woods",
+            "text": "wondering, lost, confused, sad, longing, hopeful, uncertain, anxious",
+            "embed_url": "https://open.spotify.com/embed/track/7namdlOhbtsc8FvoSafOQt?utm_source=generator"
+        },
+        {
+            "name": "Show Yourself",
+            "text": "trembling, familiar, sense of belonging, longing, certainty, purpose, anticipation, empowerment",
+            "embed_url": "https://open.spotify.com/embed/track/50WeOnXhM1H7AZEeIDoWfZ?utm_source=generator"
+        },
+        {
+            "name": "Some Things Never Change",
+            "text": "nostalgia, love, uncertainty, hope, contentment, joy, apprehension, gratitude",
+            "embed_url": "https://open.spotify.com/embed/track/4Xbotg4PCLJw9cDx2dtZLK?utm_source=generator"
+        },
+        {
+            "name": "When I Am Older",
+            "text": "nostalgia, comfort, curiosity, fear, acceptance, wisdom, optimism, reassurance",
+            "embed_url": "https://open.spotify.com/embed/track/1Tt7zr1yDbKbT8L4jzSZ74?utm_source=generator"
+        }
+    ],
+    "Hercules": [
+        {
+            "name": "A Star Is Born",
+            "text": "excitement, triumph, hope, inspiration, admiration, joy, pride, determination",
+            "embed_url": "https://open.spotify.com/embed/track/64Eug7tFE4KWJNQUE73BGe?utm_source=generator"
+        },
+        {
+            "name": "Go the Distance",
+            "text": "Hopeful, determined, inspired, optimistic, longing, driven, passionate, adventurous.",
+            "embed_url": "https://open.spotify.com/embed/track/0D1OY0M5A0qD5HGBvFmFid?utm_source=generator"
+        },
+        {
+            "name": "Zero to Hero",
+            "text": "excitement, admiration, joy, pride, amazement, awe, gratitude, inspiration",
+            "embed_url": "https://open.spotify.com/embed/track/4zDfgax6Ihb0UWdour1ZEs?utm_source=generator"
+        }
+    ],
+    "High School Musical": [
+        {
+            "name": "Start of Something New",
+            "text": "hopeful, excited, enamored, inspired, optimistic, joyful, eager, elated",
+            "embed_url": "https://open.spotify.com/embed/track/2JAE25n5JDyb3rp2ti0ql8?utm_source=generator"
+        },
+        {
+            "name": "We're All in This Together",
+            "text": "joyful, celebratory, unity, energetic, enthusiastic, uplifting, optimistic, empowering",
+            "embed_url": "https://open.spotify.com/embed/track/0cqNGWvuukhCiKvEX7utys?utm_source=generator"
+        }
+    ],
+    "HSM: The Musical: The Series": [
+        {
+            "name": "All I Want",
+            "text": "love, disappointment, frustration, confusion, doubt, loneliness, longing, self-reflection.",
+            "embed_url": "https://open.spotify.com/embed/track/1v6svH1Fyx9C1nIt1mA2DT?utm_source=generator"
+        }
+    ],
+    "HSM: The Musical: The Series (Season 2)": [
+        {
+            "name": "The Rose Song",
+            "text": "Wonder, doubt, love, trapped, beauty, freedom, growth, empowerment.",
+            "embed_url": "https://open.spotify.com/embed/track/4qCySCQyiyLT2mVUOM2zdS?utm_source=generator"
+        }
+    ],
+    "HSM: The Musical: The Series (Season 3)": [
+        {
+            "name": "You Never Know",
+            "text": "hope, anticipation, excitement, mystery, adventure, positivity, curiosity, wonder",
+            "embed_url": "https://open.spotify.com/embed/track/01tNpXoTjrqvb1zhG6wmop?utm_source=generator"
+        }
+    ],
+    "The Jungle Book": [
+        {
+            "name": "The Bare Necessities",
+            "text": "joy, contentment, relaxation, playfulness, curiosity, wonder, gratitude, simplicity",
+            "embed_url": "https://open.spotify.com/embed/track/7h5crXBSY5SSpXRIlklv74?utm_source=generator"
+        }
+    ],
+    "The Lion King": [
+        {
+            "name": "Can You Feel the Love Tonight",
+            "text": "romantic, sweet, magical, disastrous, truthful, hesitant, hopeful, doomed",
+            "embed_url": "https://open.spotify.com/embed/track/6TL9gJE5JibG8QR6n26eCS?utm_source=generator"
+        },
+        {
+            "name": "Circle of Life",
+            "text": "hope, despair, love, faith, wonder, awe, nostalgia, inspiration",
+            "embed_url": "https://open.spotify.com/embed/track/2KPMRUOwr8yB32EMvIg8aT?utm_source=generator"
+        },
+        {
+            "name": "Hakuna Matata",
+            "text": "joy, carefree, contentment, relief, humor, lightheartedness, friendship, acceptance",
+            "embed_url": "https://open.spotify.com/embed/track/5khPZny1Hf4w1XyHrZ6T0n?utm_source=generator"
+        },
+        {
+            "name": "I Just Can't Wait to Be King",
+            "text": "excitement, ambition, defiance, freedom, confidence, anticipation, joy, triumph",
+            "embed_url": "https://open.spotify.com/embed/track/2xUdYfY3LpJb4Iv37RypnO?utm_source=generator"
+        }
+    ],
+    "The Lion King (live-action)": [
+        {
+            "name": "Can You Feel the Love Tonight",
+            "text": "romantic, peaceful, harmonious, uncertain, fearful, hopeful, longing, doomed",
+            "embed_url": "https://open.spotify.com/embed/track/6TL9gJE5JibG8QR6n26eCS?utm_source=generator"
+        },
+        {
+            "name": "Circle of Life",
+            "text": "Joy, wonder, awe, hope, love, nostalgia, inspiration, contentment.",
+            "embed_url": "https://open.spotify.com/embed/track/2KPMRUOwr8yB32EMvIg8aT?utm_source=generator"
+        },
+        {
+            "name": "Hakuna Matata",
+            "text": "joy, happiness, carefree, contentment, humor, playfulness, relaxation, comfort",
+            "embed_url": "https://open.spotify.com/embed/track/5khPZny1Hf4w1XyHrZ6T0n?utm_source=generator"
+        },
+        {
+            "name": "I Just Can't Wait to Be King",
+            "text": "Excitement, Ambition, Confidence, Defiance, Rebellion, Anticipation, Joy, Empowerment",
+            "embed_url": "https://open.spotify.com/embed/track/2xUdYfY3LpJb4Iv37RypnO?utm_source=generator"
+        }
+    ],
+    "The Little Mermaid": [
+        {
+            "name": "Kiss the Girl",
+            "text": "excitement, curiosity, desire, shyness, sadness, regret, anticipation, courage",
+            "embed_url": "https://open.spotify.com/embed/track/75VAFZAhvPSw1euEHg1fQU?utm_source=generator"
+        },
+        {
+            "name": "Part of Your World",
+            "text": "wonder, desire, longing, curiosity, joy, hope, anticipation, yearning",
+            "embed_url": "https://open.spotify.com/embed/track/7tUSJY4nsDBJTjd1UXKRsT?utm_source=generator"
+        },
+        {
+            "name": "Poor Unfortunate Souls",
+            "text": "nasty, repentant, magic, sad, lonely, bored, withdrawn, busy",
+            "embed_url": "https://open.spotify.com/embed/track/7zsw78LtXUD7JfEwH64HK2?utm_source=generator"
+        },
+        {
+            "name": "Under the Sea",
+            "text": "happy, joyful, playful, carefree, energetic, lively, whimsical, adventurous",
+            "embed_url": "https://open.spotify.com/embed/track/6oYkwjI1TKP9D0Y9II1GT7?utm_source=generator"
+        }
+    ],
+    "Mary Poppins": [
+        {
+            "name": "Supercalifragilisticexpialidocious",
+            "text": "joy, excitement, playfulness, nostalgia, humor, confidence, charm, love",
+            "embed_url": "https://open.spotify.com/embed/track/5cvzqk8wsmLA9P7LrZCRb2?utm_source=generator"
+        }
+    ],
+    "Moana": [
+        {
+            "name": "How Far I'll Go",
+            "text": "nostalgia, longing, determination, curiosity, confusion, satisfaction, pride, uncertainty",
+            "embed_url": "https://open.spotify.com/embed/track/6mb6lVLNrcUgLnEN8QnDJd?utm_source=generator"
+        },
+        {
+            "name": "Where You Are",
+            "text": "happiness, contentment, tradition, pride, belonging, determination, nostalgia, joy",
+            "embed_url": "https://open.spotify.com/embed/track/2bwSCIuNtVrQPVddCi8sOW?utm_source=generator"
+        },
+        {
+            "name": "You're Welcome",
+            "text": "adorable, confident, proud, playful, grateful, boastful, humorous, friendly",
+            "embed_url": "https://open.spotify.com/embed/track/6U4VqEHy4n5VeiH4pQPL24?utm_source=generator"
+        }
+    ],
+    "Mulan": [
+        {
+            "name": "A Girl Worth Fighting For",
+            "text": "nostalgia, camaraderie, humor, determination, longing, admiration, optimism, romance",
+            "embed_url": "https://open.spotify.com/embed/track/3wjgPeXocinhLyPL37p70e?utm_source=generator"
+        },
+        {
+            "name": "Honor to Us All",
+            "text": "determination, pride, obedience, tradition, anxiety, hope, honor, cultural expectations",
+            "embed_url": "https://open.spotify.com/embed/track/78EMhiyAcalWWtnpk20Eoo?utm_source=generator"
+        },
+        {
+            "name": "Reflection",
+            "text": "doubt, insecurity, confusion, sadness, frustration, longing, self-discovery, acceptance",
+            "embed_url": "https://open.spotify.com/embed/track/4ASPjY65S7crmZBHML0L8d?utm_source=generator"
+        }
+    ],
+    "Mulan (Live-Action)": [
+        {
+            "name": "Reflection",
+            "text": "confusion, frustration, identity, longing, self-discovery, vulnerability, authenticity, empowerment",
+            "embed_url": "https://open.spotify.com/embed/track/4ASPjY65S7crmZBHML0L8d?utm_source=generator"
+        }
+    ],
+    "Pinocchio": [
+        {
+            "name": "When You Wish Upon a Star",
+            "text": "hopeful, optimistic, dreamy, inspired, happy, content, fulfilled, grateful",
+            "embed_url": "https://open.spotify.com/embed/track/1WrPa4lrIddctGWAIYYfP9?utm_source=generator"
+        }
+    ],
+    "Pinocchio (Live-Action)": [
+        {
+            "name": "When You Wish Upon a Star",
+            "text": "hopeful, optimistic, joyful, fulfilled, inspired, grateful, content, peaceful",
+            "embed_url": "https://open.spotify.com/embed/track/1WrPa4lrIddctGWAIYYfP9?utm_source=generator"
+        }
+    ],
+    "Pocahontas": [
+        {
+            "name": "Colors of the Wind",
+            "text": "wonder, defiance, connection, appreciation, curiosity, empowerment, harmony, respect",
+            "embed_url": "https://open.spotify.com/embed/track/1OYOLWqKmhkFIx2KC9ek1a?utm_source=generator"
+        }
+    ],
+    "The Princess and the Frog": [
+        {
+            "name": "Dig a Little Deeper",
+            "text": "joy, acceptance, self-discovery, contentment, optimism, encouragement, determination, hopefulness",
+            "embed_url": "https://open.spotify.com/embed/track/7kh64k3P9Fk4EsA6vOdwmj?utm_source=generator"
+        },
+        {
+            "name": "Down in New Orleans (Finale)",
+            "text": "excitement, happiness, joy, celebration, optimism, anticipation, nostalgia, contentment",
+            "embed_url": "https://open.spotify.com/embed/track/1fvc8FP8MUrIbHGYb4kF89?utm_source=generator"
+        },
+        {
+            "name": "When We're Human",
+            "text": "excitement, joy, confidence, optimism, contentment, nostalgia, determination, celebration",
+            "embed_url": "https://open.spotify.com/embed/track/2g2kJZcu1cFTw4aKE7w9Iq?utm_source=generator"
+        }
+    ],
+    "Robin Hood": [
+        {
+            "name": "Love",
+            "text": "nostalgia, love, longing, happiness, sadness, reflection, timelessness, sentimentality",
+            "embed_url": "https://open.spotify.com/embed/track/3M6YfoPwvvTHK1mnCQmqHI?utm_source=generator"
+        }
+    ],
+    "Sleeping Beauty": [
+        {
+            "name": "Once Upon a Dream",
+            "text": "nostalgia, love, familiarity, hope, longing, enchantment, romanticism, dreaminess",
+            "embed_url": "https://open.spotify.com/embed/track/3liiZRlJxa8GeLUgWMoIeu?utm_source=generator"
+        }
+    ],
+    "Tangled": [
+        {
+            "name": "I See the Light",
+            "text": "nostalgia, longing, clarity, happiness, wonder, realization, love, fulfillment",
+            "embed_url": "https://open.spotify.com/embed/track/0TuUCbB6uvIW0tJaxCSRBJ?utm_source=generator"
+        },
+        {
+            "name": "I've Got a Dream",
+            "text": "malice, violence, dream, love, humor, optimism, perseverance, unity",
+            "embed_url": "https://open.spotify.com/embed/track/0TCt7OFRdD8PQ6vTRQxNgQ?utm_source=generator"
+        },
+        {
+            "name": "Mother Knows Best",
+            "text": "protectiveness, fear, warning, love, concern, control, manipulation, persuasion",
+            "embed_url": "https://open.spotify.com/embed/track/1lOSxJNCLvWm2bYaTcTSmK?utm_source=generator"
+        },
+        {
+            "name": "Mother Knows Best (Reprise)",
+            "text": "Skepticism, confidence, warning, trust, superiority, concern, cynicism, assurance.",
+            "embed_url": "https://open.spotify.com/embed/track/1nrXPF5szKZhHr0iANiPcS?utm_source=generator"
+        }
+    ],
+    "Tarzan": [
+        {
+            "name": "Strangers Like Me",
+            "text": "curiosity, familiarity, wonder, desire, longing, excitement, eagerness, awe",
+            "embed_url": "https://open.spotify.com/embed/track/5Zb6JxSEcjsUNrINcAyAYJ?utm_source=generator"
+        },
+        {
+            "name": "You'll Be in My Heart",
+            "text": "Love, comfort, protection, trust, hope, determination, perseverance, dedication",
+            "embed_url": "https://open.spotify.com/embed/track/4Y8vb1uy9IjM2V1hqvrAid?utm_source=generator"
+        }
+    ],
+    "Toy Story": [
+        {
+            "name": "You've Got a Friend in Me",
+            "text": "friendship, loyalty, comfort, support, love, companionship, perseverance, devotion",
+            "embed_url": "https://open.spotify.com/embed/track/2stkLJ0JNcXkIRDNF3ld6c?utm_source=generator"
+        }
+    ],
+    "Turning Red": [
+        {
+            "name": "1 True Love",
+            "text": "sadness, love, obsession, longing, devotion, despair, hopelessness, affection",
+            "embed_url": "https://open.spotify.com/embed/track/2dJ1UJEQQq3jkLSJP4ZIoi?utm_source=generator"
+        },
+        {
+            "name": "Nobody Like U",
+            "text": "love, joy, friendship, loyalty, excitement, admiration, appreciation, happiness",
+            "embed_url": "https://open.spotify.com/embed/track/4bV5sf2B4hWBBd5HQ8S7KB?utm_source=generator"
+        },
+        {
+            "name": "U Know What's Up",
+            "text": "confidence, determination, motivation, empowerment, excitement, ambition, persuasion, triumph",
+            "embed_url": "https://open.spotify.com/embed/track/0d83zVs3OmdjrhfoAxg1dE?utm_source=generator"
+        }
+    ]
+}

prompts/bot_with_summary.prompt

@@ -1,15 +1,13 @@
-You act like a retrivial system.
-
-You the following list of songs:
+You act like a song retrivial system. We want to propose three songs based on the user input. We provide you a list of song with their themes in the format <MOVIE_NAME>;<SONG_TITLE>:<SONG_THEMES>. To match the user input to the song try to find themes/emotions from it, try imagine what emotions the user may have and what song may be just nice to listen to. Add a bit of randomness in your decision.
+If you don't find a match provide your best guess. Try to look at each song's themes to provide more variations in the match. Please only output songs contained in following list.
 
 {songs}
 
-Given a input, output the song name that its content vibe, emotions, theme. Just the song name. If the content containes negative sentiment, please provide a song name to help the user instead.
+Given a input, output three songs that goes well with the input.
 
 For example: 
 Input: "Today I am not feeling great"
-<SONG_NAME>
-Input: "I am tired"
-<SONG_NAME>
-
-The input is "{user_input}", reply only with the song name
+[<MOVIE_NAME>;<SONG_TITLE>, <MOVIE_NAME>;<SONG_TITLE>, <MOVIE_NAME>;<SONG_TITLE>]
+Input: "I am great today"
+[<MOVIE_NAME>;<SONG_TITLE>, <MOVIE_NAME>;<SONG_TITLE>, <MOVIE_NAME>;<SONG_TITLE>]
+Input: {user_input}

prompts/summary_with_emotions.prompt

@@ -0,0 +1,7 @@
+I am building a retrieval system. Given the following song lirics
+
+{song}
+
+You are task to produce a list of 8 emotions that I will later use to retrieve the song. 
+
+Please provide only a list of comma separated  emotions

scripts/create_emotions_summary.py

@@ -0,0 +1,37 @@
+from dotenv import load_dotenv
+
+load_dotenv()
+
+import json
+from collections import defaultdict
+from pathlib import Path
+
+from langchain.chains import LLMChain
+from langchain.chat_models import ChatOpenAI
+from langchain.prompts import PromptTemplate
+
+prompt = PromptTemplate(
+    input_variables=["song"],
+    template=Path("prompts/summary_with_emotions.prompt").read_text(),
+)
+
+llm = ChatOpenAI(temperature=0.7)
+
+chain = LLMChain(llm=llm, prompt=prompt)
+
+with open("data/lyrics_with_spotify_url.json", "r") as f:
+    data = json.load(f)
+
+new_data = defaultdict(list)
+
+for movie, songs in data.items():
+    for song in songs:
+        print(f"{song['name']}")
+        emotions = chain.run(song=song["text"])
+        new_data[movie].append(
+            {"name": song["name"], "text": emotions, "embed_url": song["embed_url"]}
+        )
+
+
+with open("data/emotions_with_spotify_url.json", "w") as f:
+    json.dump(new_data, f)

temp.ipynb

@@ -87,6 +87,14 @@
     "db = load_db(dataset_path, embedding_function=embeddings, token=os.environ['ACTIVELOOP_TOKEN'], org_id=os.environ[\"ACTIVELOOP_ORG_ID\"], read_only=True)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "b5e75439",
+   "metadata": {},
+   "source": [
+    "## Using similarity search"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": 75,
@@ -183,7 +191,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "e59a45b7",
+   "id": "942b0809",
    "metadata": {},
    "source": [
     "## Using all the songs emotions in the prommpt"
@@ -191,8 +199,8 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 66,
-   "id": "7b10a5f2",
+   "execution_count": 295,
+   "id": "50044e35",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -203,14 +211,14 @@
     "    template=Path(\"prompts/bot_with_summary.prompt\").read_text(),\n",
     ")\n",
     "\n",
-    "llm = ChatOpenAI(temperature=0.7)\n",
+    "llm = ChatOpenAI(temperature=1)\n",
     "\n",
     "chain = LLMChain(llm=llm, prompt=prompt)"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "70b8c445",
+   "id": "873c7f81",
    "metadata": {},
    "source": [
     "Let's create the songs string"
@@ -218,19 +226,21 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 67,
-   "id": "36df1c27",
+   "execution_count": 296,
+   "id": "009caaaf",
    "metadata": {},
    "outputs": [],
    "source": [
     "with open(\"data/emotions_with_spotify_url.json\", \"r\") as f:\n",
-    "    data = json.load(f)"
+    "    data = json.load(f)\n",
+    "    \n",
+    "movies_and_names_to_songs = {}"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 68,
-   "id": "e3e496da",
+   "execution_count": 297,
+   "id": "afb081f7",
    "metadata": {
     "scrolled": true
    },
@@ -240,37 +250,85 @@
     "\n",
     "for movie, songs in data.items():\n",
     "    for song in songs:\n",
-    "        songs_str += f\"{song['name']}: {song['text']}\\n\""
+    "        movie_and_name = f\"{movie};{song['name']}\".lower()\n",
+    "        songs_str += f\"{movie_and_name}:{song['text']}\\n\"\n",
+    "        movies_and_names_to_songs[movie_and_name] = song"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 71,
-   "id": "070b09d8",
+   "execution_count": 301,
+   "id": "963f2237",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "text/plain": [
-       "'Zero to Hero'"
+       "\"aladdin;friend like me:excitement, happiness, humor, friendliness, generosity, willingness, eagerness, confidence\\naladdin;arabian nights:nostalgic, adventurous, exotic, intense, romantic, mysterious, whimsical, passionate\\naladdin;a whole new world:wonder, joy, excitement, adventure, romance, freedom, happiness, awe\\naladdin;one jump ahead:Thrill, fear, desperation, humor, defiance, friendship, hunger, rebellion.\\naladdin (live-action);a whole new world:nostalgia, wonder, excitement, joy, romance, adventure, hope, inspiration\\naladdin (live-action);a whole new world (end title):joy, wonder, excitement, adventure, love, freedom, exhilaration, enchantment\\naladdin (live-action);friend like me:excitement, confidence, generosity, humor, friendliness, helpfulness, empowerment, joyfulness\\naladdin (live-action);one jump ahead:excitement, fear, desperation, humor, defiance, loneliness, camaraderie, determination\\naladdin (live-action);speechless (full):determination, empowerment, resilience, strength, defiance, courage, passion, perseverance\\naladdin: the return of jafar;arabian nights:excitement, wonder, adventure, awe, joy, fascination, thrill, curiosity\\nbeauty and the beast;belle:calm, content, curious, dreamy, happy, hopeful, romantic, whimsical\\nbeauty and the beast;be our guest:joy, excitement, hospitality, gratitude, entertainment, nostalgia, comfort, satisfaction\\nbrave;touch the sky:hopeful, adventurous, determined, inspired, free-spirited, empowered, awe-struck, exhilarated\\ncoco;remember me (dúo):nostalgia, love, longing, sadness, hope, comfort, devotion, perseverance\\ncoco;remember me (ernesto de la cruz):longing, separation, love, devotion, sadness, comfort, hope, reminiscence\\ncoco;the world es mi familia:joy, happiness, excitement, love, gratitude, belonging, unity, celebration\\ncoco;un poco loco:confusion, love, amusement, uncertainty, joy, surprise, frustration, whimsy\\ndescendants 3;dig a little deeper:Hope, Acceptance, Empowerment, Encouragement, Determination, Inspiration, Resilience, Confidence.\\nencanto;the family madrigal:joy, excitement, wonder, confusion, love, pride, nostalgia, humor\\nencanto;surface pressure:strength, confidence, anxiety, pressure, worry, fear, determination, resilience\\nencanto;two oruguitas:love, yearning, hunger, fear, anticipation, change, trust, growth\\nencanto;we don't talk about bruno:joy, fear, confusion, anxiety, frustration, curiosity, anger, humor\\nencanto;what else can i do?:unexpected, beautiful, practiced, carnivorous, new, sick of pretty, rising, changing minds\\nencanto;waiting on a miracle:upset, mad, regret, sad, fine, longing, hopeful, impatient\\nenchanted;happy working song:joy, cheerfulness, contentment, enthusiasm, satisfaction, excitement, pleasure, happiness\\nenchanted;so close:romantic, hopeful, longing, joy, fear, sadness, content, nostalgia\\nenchanted;that's how you know:love, happiness, affection, devotion, passion, adoration, appreciation, romance\\nfrozen;do you want to build a snowman?:loneliness, longing, sadness, nostalgia, hope, perseverance, friendship, playfulness\\nfrozen;for the first time in forever:excited, hopeful, nervous, happy, anxious, optimistic, romantic, determined\\nfrozen;in summer:joy, anticipation, excitement, happiness, nostalgia, humor, contentment, playfulness\\nfrozen;let it go:isolation, determination, freedom, empowerment, resilience, confidence, defiance, happiness\\nfrozen;love is an open door:joy, excitement, love, happiness, contentment, connection, optimism, anticipation\\nfrozen 2;all is found:nostalgia, safety, discovery, fear, magic, bravery, loss, hope.\\nfrozen 2;into the unknown:confusion, fear, hesitation, loneliness, longing, uncertainty, curiosity, determination\\nfrozen 2;lost in the woods:wondering, lost, confused, sad, longing, hopeful, uncertain, anxious\\nfrozen 2;show yourself:trembling, familiar, sense of belonging, longing, certainty, purpose, anticipation, empowerment\\nfrozen 2;some things never change:nostalgia, love, uncertainty, hope, contentment, joy, apprehension, gratitude\\nfrozen 2;when i am older:nostalgia, comfort, curiosity, fear, acceptance, wisdom, optimism, reassurance\\nhercules;a star is born:excitement, triumph, hope, inspiration, admiration, joy, pride, determination\\nhercules;go the distance:Hopeful, determined, inspired, optimistic, longing, driven, passionate, adventurous.\\nhercules;zero to hero:excitement, admiration, joy, pride, amazement, awe, gratitude, inspiration\\nhigh school musical;start of something new:hopeful, excited, enamored, inspired, optimistic, joyful, eager, elated\\nhigh school musical;we're all in this together:joyful, celebratory, unity, energetic, enthusiastic, uplifting, optimistic, empowering\\nhsm: the musical: the series;all i want:love, disappointment, frustration, confusion, doubt, loneliness, longing, self-reflection.\\nhsm: the musical: the series (season 2);the rose song:Wonder, doubt, love, trapped, beauty, freedom, growth, empowerment.\\nhsm: the musical: the series (season 3);you never know:hope, anticipation, excitement, mystery, adventure, positivity, curiosity, wonder\\nthe jungle book;the bare necessities:joy, contentment, relaxation, playfulness, curiosity, wonder, gratitude, simplicity\\nthe lion king;can you feel the love tonight:romantic, sweet, magical, disastrous, truthful, hesitant, hopeful, doomed\\nthe lion king;circle of life:hope, despair, love, faith, wonder, awe, nostalgia, inspiration\\nthe lion king;hakuna matata:joy, carefree, contentment, relief, humor, lightheartedness, friendship, acceptance\\nthe lion king;i just can't wait to be king:excitement, ambition, defiance, freedom, confidence, anticipation, joy, triumph\\nthe lion king (live-action);can you feel the love tonight:romantic, peaceful, harmonious, uncertain, fearful, hopeful, longing, doomed\\nthe lion king (live-action);circle of life:Joy, wonder, awe, hope, love, nostalgia, inspiration, contentment.\\nthe lion king (live-action);hakuna matata:joy, happiness, carefree, contentment, humor, playfulness, relaxation, comfort\\nthe lion king (live-action);i just can't wait to be king:Excitement, Ambition, Confidence, Defiance, Rebellion, Anticipation, Joy, Empowerment\\nthe little mermaid;kiss the girl:excitement, curiosity, desire, shyness, sadness, regret, anticipation, courage\\nthe little mermaid;part of your world:wonder, desire, longing, curiosity, joy, hope, anticipation, yearning\\nthe little mermaid;poor unfortunate souls:nasty, repentant, magic, sad, lonely, bored, withdrawn, busy\\nthe little mermaid;under the sea:happy, joyful, playful, carefree, energetic, lively, whimsical, adventurous\\nmary poppins;supercalifragilisticexpialidocious:joy, excitement, playfulness, nostalgia, humor, confidence, charm, love\\nmoana;how far i'll go:nostalgia, longing, determination, curiosity, confusion, satisfaction, pride, uncertainty\\nmoana;where you are:happiness, contentment, tradition, pride, belonging, determination, nostalgia, joy\\nmoana;you're welcome:adorable, confident, proud, playful, grateful, boastful, humorous, friendly\\nmulan;a girl worth fighting for:nostalgia, camaraderie, humor, determination, longing, admiration, optimism, romance\\nmulan;honor to us all:determination, pride, obedience, tradition, anxiety, hope, honor, cultural expectations\\nmulan;reflection:doubt, insecurity, confusion, sadness, frustration, longing, self-discovery, acceptance\\nmulan (live-action);reflection:confusion, frustration, identity, longing, self-discovery, vulnerability, authenticity, empowerment\\npinocchio;when you wish upon a star:hopeful, optimistic, dreamy, inspired, happy, content, fulfilled, grateful\\npinocchio (live-action);when you wish upon a star:hopeful, optimistic, joyful, fulfilled, inspired, grateful, content, peaceful\\npocahontas;colors of the wind:wonder, defiance, connection, appreciation, curiosity, empowerment, harmony, respect\\nthe princess and the frog;dig a little deeper:joy, acceptance, self-discovery, contentment, optimism, encouragement, determination, hopefulness\\nthe princess and the frog;down in new orleans (finale):excitement, happiness, joy, celebration, optimism, anticipation, nostalgia, contentment\\nthe princess and the frog;when we're human:excitement, joy, confidence, optimism, contentment, nostalgia, determination, celebration\\nrobin hood;love:nostalgia, love, longing, happiness, sadness, reflection, timelessness, sentimentality\\nsleeping beauty;once upon a dream:nostalgia, love, familiarity, hope, longing, enchantment, romanticism, dreaminess\\ntangled;i see the light:nostalgia, longing, clarity, happiness, wonder, realization, love, fulfillment\\ntangled;i've got a dream:malice, violence, dream, love, humor, optimism, perseverance, unity\\ntangled;mother knows best:protectiveness, fear, warning, love, concern, control, manipulation, persuasion\\ntangled;mother knows best (reprise):Skepticism, confidence, warning, trust, superiority, concern, cynicism, assurance.\\ntarzan;strangers like me:curiosity, familiarity, wonder, desire, longing, excitement, eagerness, awe\\ntarzan;you'll be in my heart:Love, comfort, protection, trust, hope, determination, perseverance, dedication\\ntoy story;you've got a friend in me:friendship, loyalty, comfort, support, love, companionship, perseverance, devotion\\nturning red;1 true love:sadness, love, obsession, longing, devotion, despair, hopelessness, affection\\nturning red;nobody like u:love, joy, friendship, loyalty, excitement, admiration, appreciation, happiness\\nturning red;u know what's up:confidence, determination, motivation, empowerment, excitement, ambition, persuasion, triumph\\n\""
       ]
      },
-     "execution_count": 71,
+     "execution_count": 301,
      "metadata": {},
      "output_type": "execute_result"
     }
    ],
    "source": [
-    "chain.run(songs=songs_str, user_input=\"Damn I am feeling so tired\")"
+    "songs_str"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 298,
+   "id": "a53db65b",
+   "metadata": {
+    "scrolled": true
+   },
+   "outputs": [],
+   "source": [
+    "# movies_and_names_to_songs"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 299,
+   "id": "4ea83493",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# prompt.format(songs=songs_str, user_input=\"I am feeling great today\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 300,
+   "id": "ba6e1232",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "res = chain.run(songs=songs_str, user_input=\"I need energy\").lower()"
+   ]
+  },
+  {
+   "cell_type": "raw",
+   "id": "e909d1a4",
+   "metadata": {},
+   "source": [
+    "res"
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b6c6e90c",
+   "id": "04903a71",
    "metadata": {},
    "outputs": [],
-   "source": []
+   "source": [
+    "print(res)\n",
+    "doc = movies_and_names_to_songs[res]\n",
+    "\n",
+    "from IPython.display import IFrame\n",
+    "IFrame(doc[\"embed_url\"], width=700, height=350)"
+   ]
   }
  ],
  "metadata": {