Update app.py

app.py

@@ -1,4 +1,4 @@
-
+import gradio as gr
 import requests
 import pandas as pd
 import time
@@ -121,6 +121,55 @@ def get_track_info(token, track_url):
         return [response.json()]
     return []
 
+def get_album_tracks(token, album_url):
+    """Get all tracks from an album URL."""
+    album_id = extract_id_from_url(album_url, "album")
+    print(f"Extracted album ID: {album_id}")
+    
+    headers = {'Authorization': f'Bearer {token}'}
+    tracks_url = f'https://api.spotify.com/v1/albums/{album_id}/tracks'
+    
+    all_tracks = []
+    next_url = tracks_url
+    
+    while next_url:
+        print(f"Fetching tracks from: {next_url}")
+        response = make_request_with_retry(next_url, headers)
+        if not response:
+            break
+            
+        data = response.json()
+        items = data.get('items', [])
+        
+        # Get album details for additional info
+        album_info = None
+        if len(all_tracks) == 0:  # Only need to get album info once
+            album_response = make_request_with_retry(f'https://api.spotify.com/v1/albums/{album_id}', headers)
+            if album_response:
+                album_info = album_response.json()
+        
+        for item in items:
+            if item:
+                # For album tracks, we need to add some missing information that's in the album
+                if album_info:
+                    item['album'] = {
+                        'name': album_info.get('name', 'Unknown'),
+                        'release_date': album_info.get('release_date', 'Not available'),
+                        'id': album_id
+                    }
+                
+                all_tracks.append(item)
+        
+        next_url = data.get('next')
+    
+    print(f"Found {len(all_tracks)} tracks in album")
+    
+    # Mark the source
+    for track in all_tracks:
+        track['playlist_source'] = album_url
+        
+    return all_tracks
+
 def get_artist_details(token, artist_id):
     """Get artist information including genres."""
     headers = {'Authorization': f'Bearer {token}'}