Update app.py

app.py

@@ -59,14 +59,22 @@ def get_audio_features(token, track_ids):
     url = 'https://api.spotify.com/v1/audio-features'
     headers = {'Authorization': f'Bearer {token}'}
     
+    # Print track IDs for debugging
+    print(f"Getting audio features for {len(track_ids)} tracks")
+    
     for i in range(0, len(track_ids), 100):
         batch_ids = track_ids[i:i+100]
         params = {'ids': ','.join(batch_ids)}
         response = make_request_with_retry(url, headers, params)
         if response:
-            for feature in response.json().get('audio_features', []):
-                if feature:
+            features_list = response.json().get('audio_features', [])
+            print(f"Received {len(features_list)} audio features")
+            for feature in features_list:
+                if feature and 'id' in feature:
                     audio_features[feature['id']] = feature
+    
+    # Print how many valid features we found
+    print(f"Found {len(audio_features)} valid audio features")
     return audio_features
 
 def get_tracks_and_features(token, url):
@@ -74,14 +82,34 @@ def get_tracks_and_features(token, url):
     track_ids = []
     
     if "track" in url:
-        track_id = url.split("/")[-1].split("?")[0]
-        track_ids = [track_id]
+        # Handle various URL formats
+        parts = url.split("/")
+        for part in parts:
+            if part and ("?" in part):
+                track_id = part.split("?")[0]
+                track_ids = [track_id]
+                break
+        if not track_ids and len(parts) > 0:
+            potential_id = parts[-1]
+            if potential_id:
+                track_ids = [potential_id]
     elif "playlist" in url:
-        playlist_id = url.split("/")[-1].split("?")[0]
-        tracks_url = f'https://api.spotify.com/v1/playlists/{playlist_id}/tracks'
-        response = make_request_with_retry(tracks_url, headers)
-        if response:
-            track_ids = [item['track']['id'] for item in response.json().get('items', []) if item.get('track')]
+        # Handle various URL formats
+        parts = url.split("/")
+        playlist_id = None
+        for part in parts:
+            if part and ("?" in part):
+                playlist_id = part.split("?")[0]
+                break
+        if not playlist_id and len(parts) > 0:
+            playlist_id = parts[-1]
+        
+        if playlist_id:
+            tracks_url = f'https://api.spotify.com/v1/playlists/{playlist_id}/tracks'
+            response = make_request_with_retry(tracks_url, headers)
+            if response and response.json().get('items'):
+                track_ids = [item['track']['id'] for item in response.json().get('items', []) 
+                            if item.get('track') and item['track'].get('id')]
     
     if not track_ids:
         return None, None
@@ -144,15 +172,35 @@ def interface(project_name, spotify_url, num_similar_songs=10):
         error_message = "Invalid URL format. Please enter a valid Spotify track or playlist URL."
         # Return empty DataFrame with error message and None for file
         return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+        
+    # Log the URL for debugging
+    print(f"Processing Spotify URL: {spotify_url}")
     
     token_spotify = get_token(client_ids[current_api_index], client_secrets[current_api_index])
     if not token_spotify:
         error_message = "Failed to authenticate with Spotify API. Please try again later."
         return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+    
+    print(f"Successfully obtained token")
+        
+    try:
+        track_ids, audio_features = get_tracks_and_features(token_spotify, spotify_url)
+        print(f"Track IDs: {track_ids[:5]}{'...' if len(track_ids) > 5 else ''}")
         
-    track_ids, audio_features = get_tracks_and_features(token_spotify, spotify_url)
-    if not track_ids or not audio_features:
-        error_message = "No valid tracks found for the provided URL."
+        if not track_ids:
+            error_message = "No valid tracks found for the provided URL. Please check the URL and try again."
+            return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+            
+        if not audio_features:
+            error_message = "Could not retrieve audio features for the tracks. Please try again."
+            return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+            
+        if len(audio_features) == 0:
+            error_message = "Received empty audio features. The track might not be available in your region."
+            return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+    except Exception as e:
+        error_message = f"An error occurred: {str(e)}"
+        print(f"Error: {str(e)}")
         return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
 
     similar_tracks_info = find_similar_tracks(token_spotify, audio_features, num_similar_songs)