Update app.py

app.py

@@ -1,4 +1,4 @@
-
+import gradio as gr
 import requests
 import pandas as pd
 import time
@@ -198,6 +198,22 @@ def get_artist_details(token, artist_id):
         return response.json()
     return None
 
+def get_track_isrc(token, track):
+    """Get ISRC code for a track if not already present."""
+    if track.get('external_ids', {}).get('isrc'):
+        return track['external_ids']['isrc']
+        
+    # If track doesn't have ISRC, try to get it from the API
+    if track.get('id'):
+        headers = {'Authorization': f'Bearer {token}'}
+        url = f'https://api.spotify.com/v1/tracks/{track["id"]}'
+        
+        response = make_request_with_retry(url, headers)
+        if response and response.json().get('external_ids', {}).get('isrc'):
+            return response.json()['external_ids']['isrc']
+    
+    return 'Not available'
+
 def extract_track_details(tracks, token):
     """Extract relevant information from track objects including artist details."""
     tracks_info = []
@@ -299,94 +315,4 @@ def interface(project_name, spotify_urls, include_all_info=True):
     
     print(f"Successfully authenticated with Spotify API")
     
-    all_tracks = []
-    
-    # Process each URL
-    for url in valid_urls:
-        try:
-            print(f"Processing URL: {url}")
-            if "playlist" in url:
-                tracks = get_playlist_tracks(token_spotify, url)
-                # Add source information
-                for track in tracks:
-                    track['playlist_source'] = url
-                all_tracks.extend(tracks)
-            elif "track" in url:
-                track = get_track_info(token_spotify, url)
-                if track:
-                    track[0]['playlist_source'] = url
-                    all_tracks.extend(track)
-            elif "album" in url:
-                album_tracks = get_album_tracks(token_spotify, url)
-                all_tracks.extend(album_tracks)
-        except Exception as e:
-            print(f"Error processing URL {url}: {str(e)}")
-            continue
-    
-    if not all_tracks:
-        error_message = "Could not find any tracks in the provided URLs."
-        return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
-    
-    # Extract track details including artist information
-    print("Extracting detailed track information including ISRCs...")
-    tracks_info = extract_track_details(all_tracks, token_spotify)
-    
-    # Remove duplicate tracks (based on ISRC or title+artist if ISRC not available)
-    print("Creating DataFrame and removing duplicates...")
-    df = pd.DataFrame(tracks_info)
-    
-    # Create a key for deduplication
-    df['dedup_key'] = df.apply(
-        lambda row: row['isrc'] if row['isrc'] != 'Not available' else f"{row['artist']}_{row['title']}", 
-        axis=1
-    )
-    
-    # Drop duplicates
-    df = df.drop_duplicates(subset='dedup_key')
-    df = df.drop(columns=['dedup_key'])
-    
-    print(f"Found {len(df)} unique tracks after deduplication")
-    
-    # Filter columns if not include_all_info
-    if not include_all_info:
-        columns_to_keep = ['artist', 'title', 'isrc', 'album', 'genres', 'release_date', 'track_popularity', 'explicit', 'spotify_url']
-        df = df[columns_to_keep]
-    
-    # Save DataFrame to an Excel file
-    tmpfile = NamedTemporaryFile(delete=False, suffix='.xlsx')
-    df.to_excel(tmpfile.name, index=False)
-    
-    # Rename the file with the project name
-    project_file_name = f"{project_name}.xlsx"
-    shutil.move(tmpfile.name, project_file_name)
-    
-    return df, project_file_name
-
-# Gradio Interface Configuration
-iface = gr.Interface(
-    fn=interface,
-    inputs=[
-        gr.Textbox(label="Project Name", placeholder="Enter a name for your export"),
-        gr.Textbox(
-            label="Spotify URLs (Tracks, Albums or Playlists)", 
-            placeholder="Enter one Spotify URL per line (tracks, albums or playlists)",
-            lines=5
-        ),
-        gr.Checkbox(label="Include All Track Information", value=True)
-    ],
-    outputs=[
-        gr.Dataframe(),
-        gr.File(label="Download Excel")
-    ],
-    title="Spotify Track Collector",
-    description="Extract tracks from multiple Spotify playlists, albums, and tracks into a single Excel file.",
-    examples=[
-        ["Pop Collection", "https://open.spotify.com/playlist/37i9dQZF1DXcBWIGoYBM5M\nhttps://open.spotify.com/track/4cOdK2wGLETKBW3PvgPWqT", True],
-        ["Rock Collection", "https://open.spotify.com/playlist/37i9dQZF1DWXRqgorJj26U", False],
-        ["Album Tracks", "https://open.spotify.com/album/1R5BORZZxNUg8QMgbqt0nd", True]
-    ],
-    allow_flagging="never"
-)
-
-if __name__ == "__main__":
-    iface.launch()
+    all_tracks = []