{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "48a5cd13",
   "metadata": {},
   "source": [
    "# Test Input Generation"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "1b6c6c0a",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T19:40:31.797073Z",
     "start_time": "2023-09-18T19:40:31.432001Z"
    }
   },
   "outputs": [],
   "source": [
    "# import sys\n",
    "\n",
    "# sys.path.insert(0, \"/home/tony/Work/glockenspiel/studio_api\")\n",
    "\n",
    "import os\n",
    "import openai"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "5b984bc3",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T19:40:31.799990Z",
     "start_time": "2023-09-18T19:40:31.798459Z"
    }
   },
   "outputs": [],
   "source": [
    "openai.api_key = os.getenv(\"OPENAI_KEY\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "48d4e0ff",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T19:40:31.861627Z",
     "start_time": "2023-09-18T19:40:31.800966Z"
    }
   },
   "outputs": [],
   "source": [
    "def validate_and_clean_lines(lines: str, n_expected_tags: int = 2):\n",
    "    \"\"\"Limit the number of tags, and remove Note from ChatGPT.\"\"\"\n",
    "    # sanity checks\n",
    "    possible_chat_gpt_notes = [\"note: \", \"nb: \"]\n",
    "    split_lines = []\n",
    "    for line in lines.split(\"\\n\"):\n",
    "        lowered_line = line.lower().strip()\n",
    "        if all(\n",
    "            chat_gpt_note not in lowered_line\n",
    "            for chat_gpt_note in possible_chat_gpt_notes\n",
    "        ):\n",
    "            split_lines.append(line.strip())\n",
    "    tag_indces = [\n",
    "        i\n",
    "        for i, line in enumerate(split_lines)\n",
    "        if \"[\" and \"]\" in line and \"Note: \" not in line\n",
    "    ]\n",
    "    if len(tag_indces) > n_expected_tags:\n",
    "        # slice the list\n",
    "        split_lines = split_lines[: tag_indces[n_expected_tags]]\n",
    "    return \"\\n\".join(split_lines).strip()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "f4c55aee",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T19:40:31.913481Z",
     "start_time": "2023-09-18T19:40:31.863418Z"
    }
   },
   "outputs": [],
   "source": [
    "async def generate_test_song(prompt, language):\n",
    "    chat_completion = await openai.ChatCompletion.acreate(\n",
    "        model=\"gpt-3.5-turbo\",\n",
    "        messages=[\n",
    "            {\n",
    "                \"role\": \"user\",\n",
    "                \"content\": f\"make a {language} song about {prompt}, with one verse followed by one chorus, each four lines. put [verse] before the verse and [chorus] before the chorus. the song should be eight lines total, and only output the song without any extra notes\",\n",
    "            }\n",
    "        ],\n",
    "        max_tokens=200,\n",
    "    )\n",
    "    lines = chat_completion.choices[0].message.content\n",
    "    # sanity checks\n",
    "    lines = validate_and_clean_lines(lines)\n",
    "    return lines"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "b151c94e",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T20:01:33.818177Z",
     "start_time": "2023-09-18T20:01:33.814078Z"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "[('Pop', 134.3), ('Rap', 134.3), ('Rock', 106.3), ('commute', 102.9), ('energy boosters', 99.5), ('Romantic', 93.1), ('chill', 90.0), ('feel good', 87.1), ('hip-hop', 87.1), ('dance & electronic', 81.5), ('R&B', 76.2), ('holiday', 73.7), ('romance', 73.7), ('Baroque', 68.9), ('Classical', 68.9), ('En Español', 68.9), ('bollywood & indian', 66.7), ('piano', 66.7), ('rock', 66.7), ('France', 64.5)]\n"
     ]
    }
   ],
   "source": [
    "import json\n",
    "common_tags_path = \"/home/mikeys/bundle/common_tags.json\"\n",
    "with open(common_tags_path, \"r\") as fp:\n",
    "    common_tags = json.load(fp)\n",
    "tags = [(k, v) for k, v in common_tags.items()]\n",
    "tags.sort(key=lambda x: (-x[1], x[0]))\n",
    "print(tags[:20])"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "143e5a15",
   "metadata": {},
   "source": [
    "# Generate test input data\n",
    "\n",
    "We will use for each language, for each tag, do three themes generation."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 26,
   "id": "ca66d18b",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T20:47:37.646635Z",
     "start_time": "2023-09-18T20:47:37.644851Z"
    }
   },
   "outputs": [],
   "source": [
    "exampe_tags = [\"pop\", \"rap\", \"rock\", \"hip-hop\", \"edm\"]\n",
    "languages = [\"english\", \"french\", \"spanish\", \"chinese\", \"russian\"]\n",
    "themes = [\"life\", \"love\", \"friendship\"]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "id": "88458107",
   "metadata": {
    "ExecuteTime": {
     "end_time": "2023-09-18T20:47:37.643556Z",
     "start_time": "2023-09-18T20:41:20.663802Z"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'tag': 'pop', 'lan': 'english', 'theme': 'life', 'text': \"[Verse]\\nIn this journey we call life, we stumble and we fall\\nBut we rise with every challenge, standing tall\\nThrough the highs and lows, we learn to grow\\nEmbracing all the beauty life has to show\\n\\n[Chorus]\\nOh, life's a wild ride, a melody we dance along\\nThrough the laughter and tears, we find where we belong\\nWe'll navigate the storms, with hearts singing strong\\nLife's symphony plays on, forever we'll carry on\"}\n",
      "{'tag': 'pop', 'lan': 'english', 'theme': 'love', 'text': \"[Verse]\\nYou walked into my life, like a ray of golden sun\\nYou stole my heart, and now you're the only one\\nEvery moment with you feels like a dream come true\\nI'll hold onto this love, forever me and you\\n\\n[Chorus]\\nLove, oh love, you're the reason I believe\\nIn this world so big, you're the missing piece\\nTogether we'll dance through the storms, hand in hand\\nLove, oh love, let's paint our forever grand\"}\n",
      "{'tag': 'pop', 'lan': 'english', 'theme': 'friendship', 'text': \"[Verse]\\nIn this world filled with uncertainty,\\nYou're the one who brings the clarity.\\nThrough thick and thin, you're my sanctuary,\\nForever grateful for your friendship's legacy.\\n\\n[Chorus]\\nWe'll go hand in hand,\\nConquer any land.\\nTogether we'll withstand,\\nOur friendship will expand.\"}\n",
      "{'tag': 'rap', 'lan': 'english', 'theme': 'life', 'text': \"[Verse]\\nIn this journey we call life, we strive to survive,\\nEvery day's a challenge to keep our dreams alive,\\nWe navigate through hurdles, facing strife,\\nBut together we'll conquer, we'll thrive.\\n\\n[Chorus]\\nLife's a rollercoaster, gotta hold on tight,\\nThrough ups and downs, we'll find our light,\\nNo matter how hard, we'll rise above the fight,\\nIn this symphony of life, we'll ignite.\"}\n",
      "{'tag': 'rap', 'lan': 'english', 'theme': 'love', 'text': \"[Verse]\\nIn this crazy world of ours, love's a game we all play,\\nSometimes it brings us joy, sometimes it leads us astray,\\nFrom the highs to the lows, love's a rollercoaster ride,\\nBut I'll take my chances, 'cause with love, I won't hide.\\n\\n[Chorus]\\nLove, a symphony that echoes in our hearts,\\nThrough the ups and downs, it's where the magic starts,\\nLove, a fire that burns with a passionate flame,\\nIn this love game, I'll never be the same.\"}\n",
      "{'tag': 'rap', 'lan': 'english', 'theme': 'friendship', 'text': \"[Verse]\\nIn this world full of chaos and fear,\\nWe found each other, our souls drew near.\\nThrough thick and thin, we'll forever be,\\nTwo hearts united, bound eternally.\\n\\n[Chorus]\\nFriends till the end, through storm and strife,\\nSide by side, let's conquer life.\\nTogether we stand, forever strong,\\nA bond of friendship, we'll carry on.\"}\n",
      "{'tag': 'rock', 'lan': 'english', 'theme': 'life', 'text': \"[Verse]\\nIn the darkness of the night we find our way,\\nChasing dreams that seem so far away.\\nThrough the highs and lows we'll face the strife,\\nHoping to find purpose in this crazy life.\\n\\n[Chorus]\\nOh, we're climbing these mountains, oh so steep,\\nIn this journey we sow the memories we keep.\\nWith every step we take, we'll try to find,\\nA little piece of heaven in this life's design.\"}\n",
      "{'tag': 'rock', 'lan': 'english', 'theme': 'love', 'text': \"[verse]\\nYou walked into my life, like an earthquake's roar\\nShaking my foundations, I couldn't ignore\\nWith every beat of my heart, I knew it was true\\nA love like this, I've never felt before\\n\\n[chorus]\\nYou're the rhythm to my melody, the one I adore\\nTogether we'll conquer mountains, forevermore\"}\n",
      "{'tag': 'rock', 'lan': 'english', 'theme': 'friendship', 'text': \"[Verse 1]\\nIn this world of gray and endless strife,\\nFriendship brings color to our life,\\nThrough the highs and lows, we stand hand in hand,\\nTogether we'll conquer, a united band.\\n\\n[Chorus]\\nOh, friendship so true, a bond that will never bend,\\nWith you by my side, there's no challenge we can't transcend.\\nThrough stormy weather or under the shining sun,\\nOur friendship will prosper, forever strong and never undone.\"}\n",
      "{'tag': 'hip-hop', 'lan': 'english', 'theme': 'life', 'text': \"[Verse]\\nIn this journey we call life, we strive to find our way\\nThrough trials and tribulations, facing battles every day\\nBut we keep our heads up high, never letting hardship weigh\\nRising from the ashes, we know we'll seize the day\\n\\n[Chorus]\\nLife's a rollercoaster ride, we gotta hold on tight\\nThrough the struggles and the pain, we'll win the fight\\nWith every step we take, we'll break through and soar\\nIn this game of life, we'll always strive for more\"}\n",
      "{'tag': 'hip-hop', 'lan': 'english', 'theme': 'love', 'text': \"[Verse]\\nLove came knockin' on my door,\\nGot me feelin' like I want more,\\nHeart racin', emotions soar,\\nWith you, my love, I wanna explore.\\n\\n[Chorus]\\nYou're my sunshine in the darkest night,\\nTogether we'll make everything right,\\nOur love's a fire, burning so bright,\\nYou and I, we'll defy the fight.\"}\n",
      "{'tag': 'hip-hop', 'lan': 'english', 'theme': 'friendship', 'text': \"[Verse]\\nThrough the highs and lows, my friend, you'll always be near\\nBound by loyalty, we conquer our fears\\nSide by side, our bond is strong and clear\\nTogether we stand, shoulder to shoulder, year after year\\n\\n[Chorus]\\nFriends forever, we'll be there till the end\\nThrough thick and thin, we'll continue to ascend\\nThrough laughter and tears, our spirits transcend\\nUnited as one, our friendship will never bend\"}\n",
      "{'tag': 'edm', 'lan': 'english', 'theme': 'life', 'text': \"[Verse]\\nIn this journey we call life,\\nWe chase the dreams up high.\\nStruggles may dim our light,\\nBut we'll rise, never saying goodbye.\\n\\n[Chorus]\\nWe dance through the night,\\nFeel the rhythm ignite.\\nTogether we'll fight,\\nEmbracing life, shining bright.\"}\n",
      "{'tag': 'edm', 'lan': 'english', 'theme': 'love', 'text': \"[Verse]\\nIn the depths of the night, I see your face\\nOur hearts entwined, our love can't be erased\\nElectric sparks dance through the air\\nTogether forever, nothing can compare\\n\\n[Chorus]\\nOh, love's melody takes us higher\\nOur souls ignite, burning like fire\\nIn the rhythm of our hearts, we find solace\\nLost in the beat, love's sweet embrace\"}\n",
      "{'tag': 'edm', 'lan': 'english', 'theme': 'friendship', 'text': \"[verse]\\nIn the moonlight, we found each other's hand\\nThrough laughter and tears, we made a timeless bond\\nTogether we shine, like stars in the night sky\\nSide by side, we conquer mountains, oh so high\\n\\n[chorus]\\nWe're dancing through the fire, together we're strong\\nFriends forever, in our hearts they belong\\nThrough thick and thin, we'll never let go\\nThis friendship we cherish, a love that will grow\"}\n",
      "{'tag': 'pop', 'lan': 'french', 'theme': 'life', 'text': '[verse]\\nLa vie est belle, elle danse et chante\\nComme une mélodie enivrante\\nElle nous offre un rêve éternel\\nRemplissant nos cœurs de bonheur réel\\n\\n[chorus]\\nLa vie, oh la vie, elle est si précieuse\\nElle nous guide vers des moments merveilleux\\nDansons, chantons, savourons chaque instant\\nLa vie, oh la vie, un trésor éclatant'}\n",
      "{'tag': 'pop', 'lan': 'french', 'theme': 'love', 'text': \"[verse]\\nJe t'ai rencontré sous le ciel étoilé\\nTon sourire m'a ensorcelé\\nLe cœur qui bat, l'amour qui grandit\\nDans tes bras, je me sens si jolie\\n\\n[chorus]\\nL'amour nous emporte dans sa mélodie\\nLe temps s'arrête, c'est notre symphonie\\nSous le soleil ou un ciel de pluie\\nPour toujours, je suis à toi et tu es à moi.\"}\n",
      "{'tag': 'pop', 'lan': 'french', 'theme': 'friendship', 'text': \"[verse]\\nToi et moi, unis par l'amitié\\nUn lien si fort qui ne fait que grandir\\nDans nos cœurs, toujours prêts à s'entraider\\nNos rires et nos pleurs, nous savons nous soutenir\\n\\n[chorus]\\nDanser la vie, main dans la main\\nDes rêves plein les yeux, ensemble on avance\\nDans nos cœurs, une amitié sans fin\\nPour toujours unis, une chance\"}\n",
      "{'tag': 'rap', 'lan': 'french', 'theme': 'life', 'text': '[verse]\\nLa vie, c\\'est un parcours, sans aucun \"rewind\"\\nDes hauts et des bas, des rêves et des chagrins\\nOn tente de s\\'accrocher, même dans les moments difficiles\\nCar la persévérance est la clé de nos destinées fragiles\\n\\n[chorus]\\nC\\'est la vie, tant qu\\'on respire, faut continuer à danser\\nAvancer sans jamais s\\'arrêter, c\\'est notre manière d\\'exister\\nC\\'est la vie, on surmonte les obstacles avec fierté\\nEt dans chaque étape, on sait trouver la beauté'}\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'tag': 'rap', 'lan': 'french', 'theme': 'love', 'text': \"[verse]\\nL'amour brûle en moi comme une flamme ardente\\nNos cœurs en fusion, une passion démente\\nJe te veux près de moi à tout instant\\nEnsemble, on construit un amour étincelant\\n\\n[chorus]\\nMon amour, mon essentiel\\nToi et moi, c'est l'amour éternel\\nTous les jours, je pense à toi, je t'appelle\\nDans tes bras, je me sens invincible\"}\n",
      "{'tag': 'rap', 'lan': 'french', 'theme': 'friendship', 'text': \"[verse]\\nJe te parle de cette belle amitié\\nComme une évidence dans ma vie tracée\\nÀ travers les tempêtes, toi et moi, soudés\\nUne complicité qui ne pourra jamais se briser\\n\\n[chorus]\\nToi et moi, amis pour la vie\\nAvec toi, je me sens si épanoui\\nDans nos rires et nos fous rires infinis\\nNotre amitié, c'est un trésor précieux, oui\"}\n",
      "{'tag': 'rock', 'lan': 'french', 'theme': 'life', 'text': \"[verse]\\nDans la vie, tout peut changer\\nDes hauts et des bas, de l'amour et de la peine\\nChercher le sens, trouver la lumière\\nBraver les tempêtes, oublier nos frontières\\n\\n[chorus]\\nLa vie, elle tourne, elle danse\\nUn voyage tantôt doux, tantôt intense\\nLa vie, une mélodie éternelle\\nQui résonne en nous, belle et rebelle\"}\n",
      "{'tag': 'rock', 'lan': 'french', 'theme': 'love', 'text': \"[verse]\\nJe me suis perdu dans ton regard\\nTon amour me transperce, ça me tarde\\nTes mots doux résonnent dans ma tête\\nJe suis fou de toi, mon cœur s'arrête\\n\\n[chorus]\\nL'amour est rocailleux, mais nous persistons\\nNos cœurs battent à l'unisson\\nDans tes bras, je me sens vivant\\nUn amour si puissant, tellement grand\"}\n",
      "{'tag': 'rock', 'lan': 'french', 'theme': 'friendship', 'text': \"[verse]\\nOh mon ami, ensemble on brille\\nDans ce monde, on s'entille\\nNos cœurs unis dans une mélodie\\nAmis pour la vie, pour l'infini\\n\\n[chorus]\\nAmis pour toujours, l'amitié pure\\nNotre lien si fort, jamais ne se déchire\\nDans ce rock français, nous chantons haut et fier\\nL'amitié éclaire nos vies, c'est notre repère\"}\n",
      "{'tag': 'hip-hop', 'lan': 'french', 'theme': 'life', 'text': \"[verse]\\nLa vie est un combat, chaque jour une bataille\\nDans cette jungle urbaine où le temps déraille\\nLes rues sont le théâtre de nos vies en noir et blanc\\nOn cherche tous la lumière, la fuite est notre chant\\n\\n[chorus]\\nC'est la vie, c'est la vie, rien n'est jamais facile\\nLes hauts et les bas, notre destin se défile\\nOn avance, tête haute, avec nos cicatrices\\nC'est la vie, c'est la vie, notre unique artifice\"}\n",
      "{'tag': 'hip-hop', 'lan': 'french', 'theme': 'love', 'text': \"[verse]\\nDans mon cœur, l'amour brûle comme un feu,\\nLes étoiles dansent, c'est merveilleux,\\nJe t'aperçois, ma chérie, ma muse,\\nCette sensation, je ne m'en excuse.\\n\\n[chorus]\\nL'amour nous emporte, en un monde à part,\\nTout devient clair, même dans le noir.\\nJe suis accro à toi, mon étoile,\\nNotre amour brille, tel un phare qui éclaire.\"}\n",
      "{'tag': 'hip-hop', 'lan': 'french', 'theme': 'friendship', 'text': \"[Verse]\\nComme une famille, on est toujours ensemble\\nNos liens sont indélébiles, indestructibles\\nÀ travers les hauts et les bas, on reste solidaires\\nDans cette belle amitié, la joie est nécessaire\\n\\n[Chorus]\\nL'amitié, c'est une évidence\\nUn cadeau précieux, une chance\\nToujours présents, on se soutient\\nDans cette amitié, on s'épanouit bien\"}\n",
      "{'tag': 'edm', 'lan': 'french', 'theme': 'life', 'text': \"[verse]\\nLa vie est un voyage sans fin\\nDans le rythme, on danse sans faim\\nChaque pas nous mène vers l'infini\\nSur la piste, on s'envole, c'est ainsi\\n\\n[chorus]\\nVivre chaque instant, sans regret\\nLa vie est belle, on fait la fête\\nSur les vagues de la musique, on plane\\nLa vie est une danse, on s'évade sans peine\"}\n",
      "{'tag': 'edm', 'lan': 'french', 'theme': 'love', 'text': \"[verse]\\nDans la nuit, un amour enflammé\\nNos corps se cherchent, nos cœurs éperdus\\nLa musique nous guide, nos âmes s'unissent\\nUne danse enivrante, amour défendu\\n\\n[chorus]\\nL'amour brûle, il ne peut pas s'éteindre\\nLes étoiles chantent notre refrain\\nUnis pour toujours, nos vies se déclinent\\nDans ce rythme envoûtant qui soulève nos mains\"}\n",
      "{'tag': 'edm', 'lan': 'french', 'theme': 'friendship', 'text': \"[verse]\\nMes amis, nous dansons ensemble\\nNos cœurs liés pour toujours\\nDans la musique nous trouvons l'amour\\nNotre amitié, une danse éternelle\\n\\n[chorus]\\nAmis pour toujours, la musique guide nos pas\\nDans nos rires et nos pleurs, une relation sans fin\\nLa joie et l'amitié, une mélodie qui unit nos âmes\\nDans ce monde d'EDM, ensemble nous sommes divins\"}\n",
      "{'tag': 'pop', 'lan': 'spanish', 'theme': 'life', 'text': '[Verse 1]\\nLa vida es un sueño que hay que vivir,\\ncon cada amanecer quiero sonreír.\\nA veces es dulce, otras veces amarga,\\npero siempre nos enseña y nos carga.\\n\\n[Chorus]\\nLa vida es un viaje lleno de emociones,\\ncaminando juntos entre altos y bajos.\\nVamos, no te rindas, sigue adelante,\\nla vida está esperándote en cada instante.'}\n",
      "{'tag': 'pop', 'lan': 'spanish', 'theme': 'love', 'text': '[verse]\\nTe vi en la distancia, supe que era amor\\nTu sonrisa brillaba como el sol\\nNo puedo evitarlo, me cautivó tu voz\\nEres la razón por la que late mi corazón\\n\\n[chorus]\\nBaila conmigo, mi amor, sin temor\\nVen a mi lado, juntos por siempre, sí, soy tu rumor\\nEn tus ojos encuentro mi destino mejor\\nEres mi luz, mi pasión, mi eterno amor'}\n",
      "{'tag': 'pop', 'lan': 'spanish', 'theme': 'friendship', 'text': '[Verse]\\nTú y yo, amigos desde el principio\\nSiempre juntos, sin ningún resquicio\\nEn cada paso, siempre hay apoyo\\nNuestra amistad, es un tesoro\\n\\n[Chorus]\\nLa amistad es un lazo tan especial\\nUn abrazo sincero y leal\\nJuntos enfrentaremos cualquier situación\\nAmigos para siempre, en cada canción'}\n",
      "{'tag': 'rap', 'lan': 'spanish', 'theme': 'life', 'text': '[Verse]\\nLa vida es un viaje, lleno de altibajos\\nCaminando por senderos desconocidos\\nA veces caemos, pero siempre nos levantamos\\nPorque en cada caída hay una lección aprendida\\n\\n[Chorus]\\nLa vida es una lucha, pero hay que ser valiente\\nNo importa lo difícil que parezca, siempre sigue adelante\\nCon cada amanecer, una nueva oportunidad florece\\nVive cada momento, porque la vida es un romance\\n\\n(Only output the song without any extra notes)'}\n",
      "{'tag': 'rap', 'lan': 'spanish', 'theme': 'love', 'text': '[Verse]\\nEnredado en tus caricias, yo me pierdo sin medida,\\nTu amor es una droga, y yo soy tu adicto de por vida.\\nCada beso es un fuego, que enciende mi alma perdida,\\nEn este laberinto de amor, quiero perderte, mi querida.\\n\\n[Chorus]\\nEres mi razón de ser, mi amor eterno y sincero,\\nEn tus ojos encuentro el cielo, en tus brazos me quiero perder.\\nEres mi sueño hecho realidad, eres todo lo que quiero,\\nEn este amor prohibido, solo quiero contigo crecer.'}\n",
      "{'tag': 'rap', 'lan': 'spanish', 'theme': 'friendship', 'text': '[Verse 1]\\nLa amistad es un tesoro sin igual\\nSiempre juntos, nunca nos dejaremos de amar\\nEn los momentos difíciles, siempre apoyaré\\nUnidos por siempre, nada nos podrá separar\\n\\n[Chorus]\\nUnidos en esta gran amistad\\nComo hermanos, somos una unidad sin final\\nLa fuerza de nuestras almas nunca se apagará\\nEn el corazón, la amistad prevalecerá'}\n",
      "{'tag': 'rock', 'lan': 'spanish', 'theme': 'life', 'text': '[Verse]\\nLa vida es una montaña rusa\\nA veces arriba, a veces abajo\\nPero siempre sigue adelante\\nEn cada paso encontramos un nuevo lazo\\n\\n[Chorus]\\nLa vida es un viaje sin destino fijo\\nCon altibajos, pero no desisto\\nAprendiendo a volar en cada caída\\nEn cada momento, se renueva mi vida'}\n",
      "{'tag': 'rock', 'lan': 'spanish', 'theme': 'love', 'text': '[Verse]\\nEn el brillo de tus ojos me encuentro perdido,\\nLa pasión en tu voz me deja cautivado.\\nEn cada latido siento que estoy vivo,\\nContigo, amor, mi ser está renovado.\\n\\n[Chorus]\\nEn el ritmo de tu amor me quiero perder,\\nEres mi musa, mi razón de ser.\\nNuestro amor, un fuego que nunca se apagará,\\nBailando al compás de la música que nos hará volar.'}\n",
      "{'tag': 'rock', 'lan': 'spanish', 'theme': 'friendship', 'text': '[Verse]\\nEn la vida hay un amigo fiel\\nQue siempre está dispuesto a ayudar\\nCon su risa brillando como el sol\\nNuestra amistad nadie podrá separar\\n\\n[Chorus]\\nAmistad, un lazo verdadero\\nUnidos siempre, sin importar el tiempo\\nJuntos enfrentando los desafíos\\nNuestra música, un grito de compañerismo'}\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'tag': 'hip-hop', 'lan': 'spanish', 'theme': 'life', 'text': '[Verse]\\nLa vida es un viaje, llena de altos y bajos\\nLuchando por un sueño, siempre en movimiento\\nHaciendo malabares, buscando mi destino\\nPero sin renunciar, sigo camino al sonido\\n\\n[Chorus]\\nLa vida es un rap, un ritmo sin igual\\nCon letra improvisada, sin miedo a fallar\\nBailando al compás del hip-hop que me anima\\nViviendo cada día con pasión y rima'}\n",
      "{'tag': 'hip-hop', 'lan': 'spanish', 'theme': 'love', 'text': '[verse]\\nEn esta vida loca, nuestro amor brilla\\nDos almas unidas, sin prisa ni prilla\\nEl corazón me late, solo por ti\\nEres el fuego que enciende mi vivir\\n\\n[chorus]\\nTu amor es mi pasión, mi razón de existir\\nEn tus brazos encuentro paz y el verdadero sentir\\nEres mi refugio, mi dulce melodía\\nJuntos en este ritmo, hasta el amanecer cada día'}\n",
      "{'tag': 'hip-hop', 'lan': 'spanish', 'theme': 'friendship', 'text': '[Verse]\\nAmigos de verdad, hasta el final\\nUnidos en el hip-hop, sin rival\\nJuntos en las calles, siempre a volar\\nLa amistad eterna, nunca acabar\\n\\n[Chorus]\\nAmistad sincera, siempre a crecer\\nLeales en la vida, vamos a vencer\\nUnidos en el rap, sin desvanecer\\nAmigos por siempre, nunca perder'}\n",
      "{'tag': 'edm', 'lan': 'spanish', 'theme': 'life', 'text': '[verse]\\nEn el ritmo de la vida, encuentro mi pasión\\nCada día es una fiesta, llena de emoción\\nLa música me eleva, me hace soñar\\nEn cada beat, mi alma se enciende sin parar\\n\\n[chorus]\\nVivo la vida bailando al compás\\nDejo mis preocupaciones atrás\\nCon la música, soy libre y feliz\\nLa fiesta nunca acaba, eso es vivir'}\n",
      "{'tag': 'edm', 'lan': 'spanish', 'theme': 'love', 'text': '[verse]\\nEn el aire se siente una pasión\\nDos corazones bailando en canción\\nEl ritmo nos envuelve, nos hace vibrar\\nEntre luces de neón, te quiero encontrar\\n\\n[chorus]\\nAmor en cada latido\\nBailando juntos sin sentido\\nEl mundo desaparece al amanecer\\nEn tus brazos, solo quiero estar'}\n",
      "{'tag': 'edm', 'lan': 'spanish', 'theme': 'friendship', 'text': '[Verse]\\nEn la pista de baile nos encontramos\\nAmistad sincera, siempre nos apoyamos\\nJuntos bailando, como dos hermanos\\nEl ritmo nos une, somos invencibles enanos\\n\\n[Chorus]\\nLa amistad es eterna, no tiene final\\nCon música y pasión, nos disfrutamos sin igual\\nBailando junto a ti, hasta el amanecer\\nNuestra unión en la pista es motivo de enaltecer'}\n",
      "{'tag': 'pop', 'lan': 'chinese', 'theme': 'life', 'text': '[verse]\\n生活啊，多姿多彩如诗画\\n每天都是新的挑战\\n困难和希望交织美好\\n享受每个瞬间，让心充满爱\\n\\n[chorus]\\n我在这个多彩的人生乐园\\n舞动青春的旋律，不畏风云\\n每一天都是独一无二的精彩\\n用心去感受，人生的美好'}\n",
      "{'tag': 'pop', 'lan': 'chinese', 'theme': 'love', 'text': '[verse]\\n在这个世界上两个人相遇\\n眼中只剩下彼此的倾心\\n一见钟情写下爱的故事\\n心跳加速，情愫沸腾起\\n\\n[chorus]\\n爱情不问东西南北中\\n只要有你，我心满意足\\n手牵手一起走向未来\\n爱永远不变，永不停息\\n\\n(Song ends here)'}\n",
      "{'tag': 'pop', 'lan': 'chinese', 'theme': 'friendship', 'text': '[verse]\\n朋友是我最珍贵的宝贝\\n相互扶持 青春不会散开\\n一起度过了无数个日夜\\n友情永远 不会改变\\n\\n[chorus]\\n朋友像太阳温暖心房\\n手牵手共同实现梦想\\n不论风雨 我们在一起\\n友谊一生永不散场'}\n",
      "{'tag': 'rap', 'lan': 'chinese', 'theme': 'life', 'text': '[verse]\\n人生就像一曲RAP\\n奇幻起落 我在追逐\\n每次挫折都是个机会\\n坚持向前才是根本\\n\\n[chorus]\\n活在当下活出精彩\\n不怕困难 我心自在\\n无论曲折也别低头\\n歌唱人生 梦想无敌'}\n",
      "{'tag': 'rap', 'lan': 'chinese', 'theme': 'love', 'text': '[Verse]\\n爱在心中躁动，像冬日的暖阳\\n对你的情感，像海浪般翻涌澎湃\\n每个呼吸都充满你的气息\\n爱情之火在心中燃烧，不止一刻\\n\\n[Chorus]\\n你是我的信仰，让我找到了快乐\\n我会守护你的心，直到永远不变节\\n无论遇到什么挑战和艰难\\n爱的力量会给我力量去战胜'}\n",
      "{'tag': 'rap', 'lan': 'chinese', 'theme': 'friendship', 'text': '[verse]\\n朋友如兄弟 因缘如镜\\n相互扶持 直到天涯涯\\n时光无情 友情难忘\\n追逐梦想 在成长的道路上\\n\\n[chorus]\\n朋友啊 你是我的明日\\n手牵手 永不分离\\n友情密不可分 在心中永驻\\n这首歌 歌颂我们的友谊'}\n",
      "{'tag': 'rock', 'lan': 'chinese', 'theme': 'life', 'text': '[Verse]\\n生活不易，如同波涛起伏\\n追寻梦想，考验意志力\\n坚持不懈，努力奋斗着\\n逆风飞翔，展现勇气与力量\\n\\n[Chorus]\\n人生之路，苦尽甘来\\n奋发向前，信念永不崩\\n挑战自我，逆境中成长\\n勇往直前，活出我风采'}\n",
      "{'tag': 'rock', 'lan': 'chinese', 'theme': 'love', 'text': '[verse]\\n我不能停止思念\\n心中只有你的身影\\n爱上你我心悸不已\\n你的存在让我充满勇气\\n\\n[chorus]\\n这份爱如狂风暴雨般热烈\\n我们的心在爱中交织\\n你是我生命中的唯一\\n你的微笑让我感到幸福的喜悦'}\n",
      "{'tag': 'rock', 'lan': 'chinese', 'theme': 'friendship', 'text': '[verse]\\n朋友，是我一生的财富\\n把你当成我心中最珍贵的宝\\n一起征服人生的高山和大海\\n永远一起，不离不弃，坚不可摧\\n\\n[chorus]\\n友谊如山，永不磨灭\\n心连心，同舟共济\\n无论前路如何坎坷\\n朋友，在一起，走到永远'}\n",
      "{'tag': 'hip-hop', 'lan': 'chinese', 'theme': 'life', 'text': '[verse]\\n生活就像一场战斗\\n每天都要付出努力\\n无论风雨还是曙光\\n我坚持走在前方\\n\\n[chorus]\\n生命之舞 不停歇\\n不畏艰辛 直追梦想\\n流淌的節奏 跳动的心\\n这就是我的华语 Hip-Hop 真谛'}\n",
      "{'tag': 'hip-hop', 'lan': 'chinese', 'theme': 'love', 'text': '[verse]\\n我在这街头流浪，寻找那失去的爱\\n心痛如刀割，思念无尽的夜晚\\n追逐烟花绽放，找回我内心的光彩\\n唯有你的拥抱，让我感受爱的温暖\\n\\n[chorus]\\n你是我的宿命，爱穿越千山万水\\n心与心永相连，你是我唯一信仰\\n那些甜蜜的回忆，在夜里安慰我的灵魂\\n我们的爱，像烟火一样灿烂绚丽\\n\\n\\n['}\n",
      "{'tag': 'hip-hop', 'lan': 'chinese', 'theme': 'friendship', 'text': '[verse]\\n朋友不分昼夜\\n共度艰难时刻\\n手牵手一起闯荡\\n友情永不会被遗忘\\n\\n[chorus]\\n兄弟姐妹一起走\\n友爱无与伦比\\n我们是挚友一生一世\\n友谊像磁铁永不分离'}\n",
      "{'tag': 'edm', 'lan': 'chinese', 'theme': 'life', 'text': '[verse]\\n人生无常变幻多，睡觉醒来全是梦\\n天地间绽放的光，把我心中激荡\\n\\n[chorus]\\n放声歌唱，寻找自己的轨迹\\n跳动音符，奏响生命的旋律'}\n",
      "{'tag': 'edm', 'lan': 'chinese', 'theme': 'love', 'text': '[Verse]\\n我迷失在爱的海洋中\\n感觉心跳像是飞翔\\n你的微笑如阳光般灿烂\\n我的世界因你而绚烂\\n\\n[Chorus]\\n爱情的旋律 在心中回响\\n激荡着我灵魂的旋律\\n我们的爱情 像一曲动人的舞曲\\n在这世界上闪耀不灭'}\n",
      "{'tag': 'edm', 'lan': 'chinese', 'theme': 'friendship', 'text': '[verse]\\n我们的友情像烈火燃烧\\n一起追逐梦想的光芒\\n无论多远距离阻挡挑战\\n友谊的舞曲继续跳跃\\n\\n[chorus]\\n朋友啊，携手共度时光\\n心与心相连直至永久\\n这感觉让世界充满光荣\\n我们的友谊鼓舞万众'}\n",
      "{'tag': 'pop', 'lan': 'russian', 'theme': 'life', 'text': '[verse]\\nЖизнь кто-то красит красками,\\nИной – серым и пылью.\\nПрошлое осталось позади,\\nТолько будущим живи.\\n\\n[chorus]\\nО, жизнь, ты мудрая и сложная,\\nСловно волна, меняешь океана ход.\\nМы в ней игроки и звезды слишком скромные,\\nТы – русская мечта, песня народная моих год.'}\n",
      "{'tag': 'pop', 'lan': 'russian', 'theme': 'love', 'text': '[Verse]\\nСердце бьется сильно, как океан волнами,\\nТы – моё счастье, мой воздух и огонь,\\nТвоя любовь, словно краски яркие на палитре,\\nМы вместе плывём на корабле любви.\\n\\n[Chorus]\\nЛюбовь, любовь, она идёт сквозь года,\\nТы – мой ангел, ты – моя звезда.\\nМы вдвоём рядом, счастливы и свободны,\\nЛюбовью плетём незабываемые сны.'}\n",
      "{'tag': 'pop', 'lan': 'russian', 'theme': 'friendship', 'text': '[verse]\\nДруг мой, ты рядом всегда,\\nМы вместе, как солнце и небо.\\nСкачем по полям, как ветерок,\\nНаша дружба - это наш ключ.\\n\\n[chorus]\\nДрузья навеки, мы вместе рукой в руке,\\nСчастье и радость с тобой мы вместе делим.\\nКак птицы в небе, мы вольно летим,\\nНаша дружба навсегда, пусть так будет всегда.'}\n",
      "{'tag': 'rap', 'lan': 'russian', 'theme': 'life', 'text': '[verse]\\nЖизнь в ритме, брат, танцуем на грани,\\nВопреки всем преградам, идём по стезе.\\nНочные огни, дневной труд, нести горе,\\nДаже если падаем, всегда возвращаемся в бой.\\n\\n[chorus]\\nЖизнь - это бит, жёсткий и быстрый,\\nМы с ней в танце, никогда не снизим обороты.\\nДолбимся в плавник, следуя своему пути,\\nЖивём на максимуме, русский рэп здесь навсегда!\\n\\nHope you enjoy the Russian rap song about life!'}\n",
      "{'tag': 'rap', 'lan': 'russian', 'theme': 'love', 'text': '[verse]\\nЯ люблю тебя, словами не передать,\\nТы - солнце, что светит день и ночь.\\nТвои глаза словно звезды сияют,\\nВ твоей улыбке я нахожу покой.\\n\\n[chorus]\\nЛюбовь, великая и прекрасная,\\nОна в сердце сжигает огнем страстным.\\nС тобой рядом мир превращается в рай,\\nЛишь рядом с тобою я счастлив и свыше.'}\n",
      "{'tag': 'rap', 'lan': 'russian', 'theme': 'friendship', 'text': '[verse]\\nМы друзья, как родные, навсегда будем вместе,\\nЧерез тернии и преграды идти не разлучившись.\\nВместе веселиться, вместе грусти переживать,\\nЭта дружба неуязвима, в каждом сердце загорать.\\n\\n[chorus]\\nДрузья - это семья, что нас прочно объединяет,\\nБезмерная любовь, что нас вместе согревает.\\nСквозь время и расстояния нас дружба несет,\\nМы вместе по жизни и за друзей поднимаем бокалы яркие.\\n\\nNote'}\n",
      "{'tag': 'rock', 'lan': 'russian', 'theme': 'life', 'text': '[verse]\\nВ жизни моей, как в океане,\\nВолны судьбы меня качают,\\nНа берегах надежд мечтаю,\\nВлюблен в рок - сердце теребит.\\n\\n[chorus]\\nЖизнь - гитара рок-н-ролла,\\nСвою историю расскажет,\\nСвергая беды, тревоги скроет,\\nРусский рок прожигает душу.'}\n",
      "{'tag': 'rock', 'lan': 'russian', 'theme': 'love', 'text': '[verse]\\nТы - воздух, ты - пламень, моя любовь,\\nТвои глаза светятся, словно зарницы над небом.\\nВсе миры в твоих объятьях, восклицает сердце мое,\\nТы - лишь ты, и никогда не отпущу тебя, дорогая моя.\\n\\n[chorus]\\nТы – мой камень, мой взрыв, мое счастье на земле,\\nТвои поцелуи как ракеты в небе.\\nСердца стук только для нас двоих, это наш оркестр,\\nМоя любовь, моя радость, веч'}\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "{'tag': 'rock', 'lan': 'russian', 'theme': 'friendship', 'text': '[verse]\\nДруг, мой лучший друг, ты всегда со мной,\\nМы вместе с детства и до зари,\\nНи беды, ни горя нас не разлучит,\\nПусть дружба наша будет вековой.\\n\\n[chorus]\\nДрузья, всегда вместе мы идём,\\nСквозь трудности, вместе поднимаемся,\\nНаша дружба крепка и верна,\\nПусть звучит песня о дружбе вечной.\\n\\nTranslation:'}\n",
      "{'tag': 'hip-hop', 'lan': 'russian', 'theme': 'life', 'text': '[verse]\\nЖизнь, как река,\\nТо плавно течет, то сильно бьет.\\nНа пути преграды,\\nНо мы вместе их сносим, несгибаемо бьемся.\\n\\n[chorus]\\nЭто русский хип-хоп - наша жизнь, наш стиль,\\nМы покоряем миры, преодолеваем барьеры.\\nСила бита, слов смысл,\\nДержим путь, несмотря на все риски.'}\n",
      "{'tag': 'hip-hop', 'lan': 'russian', 'theme': 'love', 'text': '[verse]\\nВ городе моём, любовь есть в воздухе,\\nСердца страстью горят, тайно намекают.\\nМы двое, как феникс, в шуме ночи нашей,\\nРусским хип-хопом зажёглись, вместе плывают.\\n\\n[chorus]\\nТы и я, любовь в нас сольётся,\\nВ городах ты и я, вместе зажгутся.\\nТак давай наслаждаться этой любовью,\\nРусским хип-хопом танцуй со мною.'}\n",
      "{'tag': 'hip-hop', 'lan': 'russian', 'theme': 'friendship', 'text': '[verse]\\nДружба - это святое слово,\\nСердца сплотить смогает оно.\\nВместе мы смеемся и плачем,\\nВ дружбе счастливы, как в сказке мы.\\n\\n[chorus]\\nДруг мой, ты мне всегда рядом,\\nВместе дружить, вместе с надеждой.\\nВсе трудности мы вместе пройдём,\\nДружба вечна, на всю жизнь, дорогой.'}\n",
      "{'tag': 'edm', 'lan': 'russian', 'theme': 'life', 'text': '[verse]\\nВ жизни буря и свет, как вечный танец,\\nБоль уходит и приходит назад.\\nМы смеемся, плачем, бессонницей горим,\\nСердце бьется в ритме, зажигает костер.\\n\\n[chorus]\\nМы ведь мечтаем, всех бед лишиться,\\nСолнце нас осветить, светить навсегда.\\nВ душе у нас горит неразбитый огонь,\\nЖивем жизнью, словно вечность.'}\n",
      "{'tag': 'edm', 'lan': 'russian', 'theme': 'love', 'text': '[verse]\\nЛюбовь вибрирует в воздухе,\\nОна заполняет сердца на ночь.\\nТы и я, мы вместе в этом сне,\\nБезграничная страсть, наш зов печально слышен.\\n\\n[chorus]\\nЛюбовь наша, она так ярка и сильна,\\nОгонь в сердцах, искры пылают в огненном танце.\\nПусть мелодии звучат нежно в ночи,\\nМы двое, закутанные в пленительной романтике.'}\n",
      "{'tag': 'edm', 'lan': 'russian', 'theme': 'friendship', 'text': '[verse]\\nДрузья, мы вместе всегда,\\nРядом счастье и грусть;\\nС нами пройдём все преграды,\\nВместе мы сильны и юны.\\n\\n[chorus]\\nДружба, наш светлый путь,\\nС лучшими моментами тут;\\nДружба, никогда не прошла,\\nМы вместе до конца идем всегда.'}\n"
     ]
    }
   ],
   "source": [
    "for lan in languages:\n",
    "    for tag in exampe_tags:\n",
    "        c = 0\n",
    "        for theme in themes:\n",
    "            info = {\n",
    "                \"tag\": tag,\n",
    "                \"lan\": lan,\n",
    "                \"theme\": theme\n",
    "            }\n",
    "            response = await generate_test_song(f\"{theme}\", f\"{lan} {tag}\")\n",
    "            info[\"text\"] = response\n",
    "            print(info)\n",
    "            with open(f\"inputs/{lan}_{tag}_{c}.json\", \"w\") as fp:\n",
    "                json.dump(info, fp)\n",
    "            c += 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1c5503a5",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.12"
  },
  "toc": {
   "base_numbering": 1,
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "title_cell": "Table of Contents",
   "title_sidebar": "Contents",
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": true,
   "toc_window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}