i am back

i made changes to this script that detects if the current song is from a playlist or album, before I made changes, it was playlist only

main.py

@@ -32,20 +32,28 @@ if response is not None and response.get('is_playing', False):
     track_url = track['external_urls']['spotify']
 
     # Initialize playlist_url
-    playlist_url = None
+    context_name = None
+    context_url = None
 
     # Check if the track is from a context (playlist)
-    if response.get('context') and response['context'].get('type') == 'playlist':
-        playlist_uri = response['context']['uri']
-        playlist_id = playlist_uri.split(':')[-1]
-        playlist = sp.playlist(playlist_id)
-        playlist_name = playlist['name']
-        playlist_url = playlist['external_urls']['spotify']
+    if response.get('context'):
+        context_type = response['context'].get('type')
+        context_uri = response['context']['uri']
+        context_id = context_uri.split(':')[-1]
+    
+    if context_type == 'playlist':
+        context = sp.playlist(context_id)
+        context_name = context['name']
+        context_url = context['external_urls']['spotify']
+    elif context_type == 'album':
+        context = sp.album(context_id)
+        context_name = context['name']
+        context_url = context['external_urls']['spotify']
 
     # Format the message
     description = f"**Artists**\n{artists}\n**Album**\n{album_name}\n\n[Listen on Spotify]({track_url})"
-    if playlist_name and playlist_url:
-        description += f" | [{playlist_name}]({playlist_url})"
+    if context_name and context_url:
+        description += f" | [{context_name}]({context_url})"
     description += f" | [visit the source](https://git.rintyuu.dev/rintyuu/discord-nowplaying)"
     
     message = {