ym88659208ym87991671
Получение рекомендаций в Spotify с помощью GigaChat | Документация для разработчиков

Получение рекомендаций в Spotify с помощью GigaChat

Обновлено 1 марта 2024

Этот ноутбук показывает, как можно получать рекомендаций в Spotify c помощью GigaChat. Получить Client Id и Client Secret в Spotify можно тут

!pip install spotipy gigachain
import json
import os
import sys

import spotipy
from langchain.agents import AgentExecutor, ZeroShotAgent
from langchain.chat_models import GigaChat
from langchain.tools import Tool
from spotipy.oauth2 import SpotifyClientCredentials
os.environ["SP_ID"] = "<Spotify Client ID>"
os.environ["SP_SECRET"] = "<Spotify Client Secret>"

llm = GigaChat() # Use big model of GigaChat

spotify = spotipy.Spotify(
client_credentials_manager=SpotifyClientCredentials(
client_id=os.getenv("SP_ID"), client_secret=os.getenv("SP_SECRET")
)
)
def create_playlist(data):
"""Функция получения рекомендаций"""
data = json.loads(data)
kwargs = {}
if "energy" in data:
kwargs["target_energy"] = data["energy"] / 100
if "happiness" in data:
kwargs["target_valence"] = data["happiness"] / 100
if "dance" in data:
kwargs["target_danceability"] = data["dance"] / 100
if "genres" in data:
kwargs["seed_genres"] = data["genres"][:5]
else:
kwargs["seed_genres"] = ["pop", "hip-hop", "rock", "r-n-b"]

results = spotify.recommendations(limit=10, **kwargs)
tracks = []
for track in results["tracks"]:
name = f"{track['artists'][0]['name']}{track['name']}"
os.write(1, f"{name}\n".encode())
tracks.append(name)

return "\n".join(tracks)


tools = [
Tool.from_function(
func=create_playlist,
return_direct=True,
name="CREATE_PLAYLIST",
description="""Описание: создает плейлисты \n Параметры:
{{"happiness": число которое обозначает настроение запроса где 0 - грусть, печаль, агрессивность и 100 - счастье, веселье
"energy": число которое обозначает энергичность запроса где 0 — спокойствие и 100 — энергичность не обязательный параметр,
"dance": число которое обозначает танцевальность где 0 - спокойствие и 100 — танцевальная музыка не обязательный параметр,
"genres": список от 1 до 5 элементов выбранных из этого списка ["acoustic", "afrobeat", "alt-rock", "alternative", "ambient", "anime", "black-metal", "bluegrass", "blues", "bossanova", "brazil", "breakbeat", "british", "cantopop", "chicago-house", "children", "chill", "classical", "club", "comedy", "country", "dance", "dancehall", "death-metal", "deep-house", "detroit-techno", "disco", "disney", "drum-and-bass", "dub", "dubstep", "edm", "electro", "electronic", "emo", "folk", "forro", "french", "funk", "garage", "german", "gospel", "goth", "grindcore", "groove", "grunge", "guitar", "happy", "hard-rock", "hardcore", "hardstyle", "heavy-metal", "hip-hop", "holidays", "honky-tonk", "house", "idm", "indian", "indie", "indie-pop", "industrial", "iranian", "j-dance", "j-idol", "j-pop", "j-rock", "jazz", "k-pop", "kids", "latin", "latino", "malay", "mandopop", "metal", "metal-misc", "metalcore", "minimal-techno", "movies", "mpb", "new-age", "new-release", "opera", "pagode", "party", "philippines-opm", "piano", "pop", "pop-film", "post-dubstep", "power-pop", "progressive-house", "psych-rock", "punk", "punk-rock", "r-n-b", "rainy-day", "reggae", "reggaeton", "road-trip", "rock", "rock-n-roll", "rockabilly", "romance", "sad", "salsa", "samba", "sertanejo", "show-tunes", "singer-songwriter", "ska", "sleep", "songwriter", "soul", "soundtracks", "spanish", "study", "summer", "swedish", "synth-pop", "tango", "techno", "trance", "trip-hop", "turkish", "work-out", "world-music"]}}''
""", # noqa: E501,
),
]

agent = ZeroShotAgent.from_llm_and_tools(llm=llm, tools=tools)
agent_executor = AgentExecutor.from_agent_and_tools(
agent=agent, tools=tools, verbose=True
)

stdout = sys.stdout

sys.stdout = stdout

print(
agent_executor.run(
"Создай мне грустный плейлист, когда хочется сбежать от всех проблем"
)
)
    

> Entering new AgentExecutor chain...
-----
Thought: Создать плейлист с настроением грусть и жанрами, которые подходят для такого настроения
Action: CREATE_PLAYLIST
Action Input: {"happiness": 0, "genres": ["sad", "ballads", "love-songs"]}

Observation: John Murphy — In the House - In a Heartbeat
Lana Del Rey — Carmen
Céline Dion — My Heart Will Go On (Love Theme from "Titanic")
Lifehouse — Everything
Touché Amoré — Condolences
Of Mice & Men — My Understandings
Coldplay — Warning Sign
Sigur Rós — Untitled #1 (Vaka)
Troye Sivan — Happy Little Pill
B.o.B — Both of Us (feat. Taylor Swift)


> Finished chain.
John Murphy — In the House - In a Heartbeat
Lana Del Rey — Carmen
Céline Dion — My Heart Will Go On (Love Theme from "Titanic")
Lifehouse — Everything
Touché Amoré — Condolences
Of Mice & Men — My Understandings
Coldplay — Warning Sign
Sigur Rós — Untitled #1 (Vaka)
Troye Sivan — Happy Little Pill
B.o.B — Both of Us (feat. Taylor Swift)
ПАО Сбербанк использует cookie для персонализации сервисов и удобства пользователей.
Вы можете запретить сохранение cookie в настройках своего браузера.