Update app.py

app.py

@@ -3,10 +3,8 @@ import requests
 import pandas as pd
 import time
 import shutil
-import numpy as np
 import os
 from tempfile import NamedTemporaryFile
-from openpyxl import Workbook
 
 # Spotify API credentials from environment variables
 client_ids = os.getenv("SPOTIFY_CLIENT_IDS")
@@ -73,7 +71,7 @@ def get_audio_features(token, track_ids):
 
 def get_tracks_and_features(token, url):
     headers = {'Authorization': f'Bearer {token}'}
-    track_ids = None
+    track_ids = []
     
     if "track" in url:
         track_id = url.split("/")[-1].split("?")[0]
@@ -83,7 +81,7 @@ def get_tracks_and_features(token, url):
         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['track']]
+            track_ids = [item['track']['id'] for item in response.json().get('items', []) if item.get('track')]
     
     if not track_ids:
         return None, None
@@ -141,13 +139,28 @@ def get_track_information(token, track_ids):
 
 # Main Interface Function
 def interface(project_name, spotify_url, num_similar_songs=10):
+    # Input validation
+    if not spotify_url or not ("spotify.com" in spotify_url and ("track" in spotify_url or "playlist" in spotify_url)):
+        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
+    
     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
+        
     track_ids, audio_features = get_tracks_and_features(token_spotify, spotify_url)
     if not track_ids or not audio_features:
-        return "Invalid URL or no songs found.", None
+        error_message = "No valid tracks found for the provided URL."
+        return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
 
     similar_tracks_info = find_similar_tracks(token_spotify, audio_features, num_similar_songs)
     
+    if not similar_tracks_info:
+        error_message = "Could not find similar tracks. Please try with a different track or playlist."
+        return gr.Dataframe(value=pd.DataFrame({"Error": [error_message]})), None
+    
     # Create DataFrame
     df = pd.DataFrame(similar_tracks_info)
 
@@ -156,7 +169,7 @@ def interface(project_name, spotify_url, num_similar_songs=10):
     df.to_excel(tmpfile.name, index=False)
 
     # Rename the file with the project name
-    project_file_name = f"{project_name}.xlsx"
+    project_file_name = f"{project_name if project_name else 'spotify_similar_tracks'}.xlsx"
     shutil.move(tmpfile.name, project_file_name)
 
     return df, project_file_name  # Returns the DataFrame and the link to the Excel file
@@ -165,12 +178,22 @@ def interface(project_name, spotify_url, num_similar_songs=10):
 iface = gr.Interface(
     fn=interface,
     inputs=[
-        gr.Textbox(label="Project Name"),
-        gr.Textbox(label="Spotify URL (Track or Playlist)"),
-        gr.Number(label="Number of Similar Songs", value=10, minimum=1, maximum=100)
+        gr.Textbox(label="Project Name", placeholder="Enter a name for your project"),
+        gr.Textbox(label="Spotify URL (Track or Playlist)", placeholder="https://open.spotify.com/track/... or https://open.spotify.com/playlist/..."),
+        gr.Slider(label="Number of Similar Songs", minimum=1, maximum=100, value=10, step=1)
+    ],
+    outputs=[
+        gr.Dataframe(),
+        gr.File(label="Download Excel")
     ],
-    outputs=[gr.Dataframe(), gr.File(label="Download Excel")],
     title="Spotify Similar Track Finder",
-    description="Enter a Spotify URL to find similar songs based on their features."
+    description="Enter a Spotify URL to find similar songs based on their audio features.",
+    examples=[
+        ["Pop Hits", "https://open.spotify.com/track/1mWdTewIgB3gtBM3TOSFhB", 10],
+        ["Rock Classics", "https://open.spotify.com/playlist/37i9dQZF1DWXRqgorJj26U", 15]
+    ],
+    allow_flagging="never"
 )
-iface.launch()
+
+if __name__ == "__main__":
+    iface.launch()