spotify_recommender.py

import os
import spotipy
from spotipy.oauth2 import SpotifyClientCredentials
from dotenv import load_dotenv


load_dotenv()

CLIENT_ID = os.getenv('SPOTIPY_CLIENT_ID')
CLIENT_SECRET = os.getenv('SPOTIPY_CLIENT_SECRET')

if not CLIENT_ID or not CLIENT_SECRET:
    print("Error: Spotify API credentials not loaded. Make sure SPOTIPY_CLIENT_ID and SPOTIPY_CLIENT_SECRET are set in your .env file.")
    exit()

sp = spotipy.Spotify(auth_manager=SpotifyClientCredentials(client_id=CLIENT_ID,
                                                           client_secret=CLIENT_SECRET))


EMOTION_TO_SEARCH_MAP = {
    "happy": {
        "keywords": ["happy pop", "upbeat dance", "joyful songs"],
        "genres": ["pop", "dance", "funk"] 
    },
    "sad": {
        "keywords": ["sad acoustic", "melancholy piano", "heartbreak songs"],
        "genres": ["acoustic", "ambient", "ballad"]
    },
    "angry": {
        "keywords": ["angry rock", "intense metal", "rage rap"],
        "genres": ["rock", "metal", "hip-hop"]
    },
    "neutral": {
        "keywords": ["chill lo-fi", "relaxing instrumental", "calm background music"],
        "genres": ["lo-fi", "ambient", "chill"]
    },
    "surprise": {
        "keywords": ["exciting electronic", "unpredictable indie", "bouncy pop"],
        "genres": ["electronic", "indie-pop", "alternative"]
    },
    "fear": {
        "keywords": ["dark ambient", "suspenseful instrumental", "eerie soundscapes"],
        "genres": ["ambient", "dark-ambient"]
    },
    "disgust": {
        "keywords": ["heavy metal", "punk rock", "industrial music"],
        "genres": ["heavy-metal", "punk"]
    }
}


def get_music_recommendations(emotion, limit=5):
    emotion = emotion.lower()
    search_terms = EMOTION_TO_SEARCH_MAP.get(emotion, EMOTION_TO_SEARCH_MAP["neutral"])

    
    keywords = search_terms.get("keywords")
    genres_for_search = search_terms.get("genres")

    query = keywords[0] if keywords else genres_for_search[0] 
    if not query:
        print(f"Warning: No valid search query for emotion '{emotion}'.")
        return []

    try:
        
        results = sp.search(q=query, type='track', limit=limit)
        tracks = []
        for track in results['tracks']['items']:
            tracks.append({
                'name': track['name'],
                'artist': track['artists'][0]['name'],
                'url': track['external_urls']['spotify'],
                'embed_url': f"https://open.spotify.com/embed/track/{track['id']}"
            })
        return tracks
    except spotipy.SpotifyException as e:
        print(f"Spotify API Error during search: {e}")
        return []
    except Exception as e:
        print(f"An unexpected error occurred during search: {e}")
        return []

if __name__ == "__main__":
    print("--- Spotify Music Recommender Test (Search-Based) ---")
    emotions_to_test = ["happy", "sad", "angry", "neutral", "surprise"]

    for emotion in emotions_to_test:
        print(f"\nRecommending for '{emotion}' emotion:")
        recommendations = get_music_recommendations(emotion, limit=3)
        if recommendations:
            for i, track in enumerate(recommendations):
                print(f"  {i+1}. {track['name']} by {track['artist']}")
                print(f"     URL: {track['url']}")
                print(f"     Embed URL: {track['embed_url']}")
        else:
            print(f"  No recommendations found for '{emotion}'.")

    print("\nRecommending for 'confused' emotion (should fallback to neutral):")
    recommendations = get_music_recommendations("confused", limit=2)
    if recommendations:
        for i, track in enumerate(recommendations):
            print(f"  {i+1}. {track['name']} by {track['artist']}")
    else:
        print("  No recommendations found for 'confused'.")