Update app.py

app.py

@@ -113,25 +113,75 @@ def get_track_info(token, track_url):
         return [response.json()]
     return []
 
-def extract_track_details(tracks):
-    """Extract relevant information from track objects."""
+def get_artist_details(token, artist_id):
+    """Get artist information including genres."""
+    headers = {'Authorization': f'Bearer {token}'}
+    url = f'https://api.spotify.com/v1/artists/{artist_id}'
+    
+    response = make_request_with_retry(url, headers)
+    if response:
+        return response.json()
+    return None
+
+def extract_track_details(tracks, token):
+    """Extract relevant information from track objects including artist details."""
     tracks_info = []
+    artists_cache = {}  # Cache artist details to reduce API calls
+    
     for track in tracks:
         if not track:
             continue
+        
+        # Get artist details if we have an artist ID
+        artist_id = None
+        artist_name = 'Unknown'
+        genres = []
+        
+        if track.get('artists') and len(track['artists']) > 0:
+            artist_id = track['artists'][0].get('id')
+            artist_name = track['artists'][0].get('name', 'Unknown')
+            
+            # Only fetch artist details if we have an ID and haven't cached it already
+            if artist_id and artist_id not in artists_cache:
+                artist_data = get_artist_details(token, artist_id)
+                if artist_data:
+                    artists_cache[artist_id] = {
+                        'genres': artist_data.get('genres', []),
+                        'popularity': artist_data.get('popularity', 'Not available'),
+                        'followers': artist_data.get('followers', {}).get('total', 'Not available')
+                    }
+            
+            # Get genres from cache
+            if artist_id in artists_cache:
+                genres = artists_cache[artist_id].get('genres', [])
+        
+        # Calculate duration in minutes:seconds format
+        duration_ms = track.get('duration_ms', 0)
+        if duration_ms:
+            minutes = duration_ms // 60000
+            seconds = (duration_ms % 60000) // 1000
+            duration_formatted = f"{minutes}:{seconds:02d}"
+        else:
+            duration_formatted = 'Not available'
             
         tracks_info.append({
-            'artist': track['artists'][0]['name'] if track.get('artists') else 'Unknown',
+            'artist': artist_name,
             'title': track.get('name', 'Unknown'),
             'album': track.get('album', {}).get('name', 'Unknown'),
             'isrc': track.get('external_ids', {}).get('isrc', 'Not available'),
-            'popularity': track.get('popularity', 'Not available'),
+            'track_popularity': track.get('popularity', 'Not available'),
+            'genres': ', '.join(genres) if genres else 'Not available',
+            'artist_popularity': artists_cache.get(artist_id, {}).get('popularity', 'Not available') if artist_id else 'Not available',
+            'artist_followers': artists_cache.get(artist_id, {}).get('followers', 'Not available') if artist_id else 'Not available',
             'release_date': track.get('album', {}).get('release_date', 'Not available'),
-            'duration_ms': track.get('duration_ms', 'Not available'),
+            'duration': duration_formatted,
+            'duration_ms': duration_ms,
+            'explicit': 'Yes' if track.get('explicit', False) else 'No',
             'spotify_url': track.get('external_urls', {}).get('spotify', 'Not available'),
             'preview_url': track.get('preview_url', 'Not available'),
             'playlist_source': getattr(track, 'playlist_source', 'Direct Track')
         })
+    
     return tracks_info
 
 # Main Interface Function
@@ -192,8 +242,8 @@ def interface(project_name, spotify_urls, include_all_info=True):
         error_message = "Could not find any tracks in the provided URLs."
         return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
     
-    # Extract track details
-    tracks_info = extract_track_details(all_tracks)
+    # Extract track details including artist information
+    tracks_info = extract_track_details(all_tracks, token_spotify)
     
     # Remove duplicate tracks (based on ISRC or title+artist if ISRC not available)
     df = pd.DataFrame(tracks_info)
@@ -212,7 +262,7 @@ def interface(project_name, spotify_urls, include_all_info=True):
     
     # Filter columns if not include_all_info
     if not include_all_info:
-        columns_to_keep = ['artist', 'title', 'album', 'release_date', 'popularity', 'spotify_url']
+        columns_to_keep = ['artist', 'title', 'album', 'genres', 'release_date', 'track_popularity', 'explicit', 'spotify_url']
         df = df[columns_to_keep]
     
     # Save DataFrame to an Excel file