Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import spotipy
 from spotipy.oauth2 import SpotifyOAuth
 import toml
-import os
 from urllib.parse import urlparse, parse_qs
 import logging
 
@@ -44,7 +43,6 @@ def exchange_code_for_token(code):
     if not token_info:
         raise ValueError("Failed to retrieve token info.")
     return token_info['access_token']
-    print(token_info)
 
 def search_song(query, access_token):
     sp = spotipy.Spotify(auth=access_token)
@@ -68,66 +66,35 @@ def get_recommendations(seed_track_ids, access_token):
     global nor
     sp = spotipy.Spotify(auth=access_token)
     
-    # Log the input parameters (ensure access_token is masked)
-    logging.debug(f"Input seed_track_ids: {seed_track_ids[:5]}")  # Only log the first 5 tracks
-    logging.debug(f"Access token: {'*' * 10}...")  # Mask token for security
-    
     seed_track_ids = seed_track_ids[:5]
     
     try:
         logging.debug(f"Fetching recommendations based on {len(seed_track_ids)} seed tracks.")
-        
-        # Logging the actual API call
-        logging.debug(f"Calling Spotify API with seed_track_ids: {seed_track_ids} and limit: {nor}")
-        
         recommendations = sp.recommendations(seed_tracks=seed_track_ids, limit=nor)
         
-        # Log the full response to see if the data is returned as expected
-        logging.debug(f"Recommendations data: {recommendations}")
-
-    except spotipy.exceptions.SpotifyException as e:
-        logging.error(f"SpotifyException: {e}")
-        logging.debug("Error occurred while fetching recommendations. Returning default empty values.")
-        return f"Error fetching recommendations: {e}", [], [], [], []  # Ensure 5 values
-    except Exception as e:
-        logging.error(f"Exception: {e}")
-        logging.debug("General error occurred while fetching recommendations. Returning default empty values.")
-        return f"Error fetching recommendations: {e}", [], [], [], []  # Ensure 5 values
-
-    # If recommendations are fetched successfully, log the structure of the response
-    try:
         recommended_tracks = [track for track in recommendations['tracks']]
         recommended_track_ids = [track['id'] for track in recommended_tracks]
         recommended_tracks_names = [track['name'] for track in recommended_tracks]
 
-        # Log the recommended tracks
-        logging.debug(f"Recommended track IDs: {recommended_track_ids}")
-        logging.debug(f"Recommended track names: {recommended_tracks_names}")
-        
-        # Create Spotify embeds
         spotify_embeds = [
             f'<iframe src="https://open.spotify.com/embed/track/{track["id"]}" width="300" height="80" frameBorder="0" allowtransparency="true" allow="encrypted-media"></iframe>'
             for track in recommended_tracks
         ]
         
-        logging.debug(f"Generated Spotify embeds: {spotify_embeds}")
-
         return recommended_tracks, recommended_tracks_names, spotify_embeds, recommended_track_ids, recommended_tracks
     
-    except KeyError as e:
-        logging.error(f"KeyError: Missing expected key in API response: {e}")
-        logging.debug(f"API response structure: {recommendations}")
-        return "Error: Missing key in API response", [], [], [], []  # Ensure 5 values
+    except spotipy.exceptions.SpotifyException as e:
+        logging.error(f"SpotifyException: {e}")
+        return f"Error fetching recommendations: {e}", [], [], [], []  # Ensure 5 values
     except Exception as e:
-        logging.error(f"Exception while processing recommendations data: {e}")
-        return f"Error processing recommendations: {e}", [], [], [], []  # Ensure 5 values
+        logging.error(f"Exception: {e}")
+        return f"Error fetching recommendations: {e}", [], [], [], []  # Ensure 5 values
 
 def create_playlist(access_token, playlist_name):
     sp = spotipy.Spotify(auth=access_token)
     try:
         user_id = sp.current_user()['id']
         playlist = sp.user_playlist_create(user=user_id, name=playlist_name, public=True)
-        logging.info(f"Playlist '{playlist_name}' created successfully.")
         return playlist['id'], None
     except spotipy.exceptions.SpotifyException as e:
         logging.error(f"Failed to create playlist '{playlist_name}': {e}")
@@ -197,18 +164,16 @@ def gradio_interface():
             outputs=[access_token_state, login_section, app_section]
         )
 
-        # Updated add_song function
         def add_song(name, track_list, access_token):
             track = search_song(name, access_token)
             if track:
-                track_list.append(track)  # Add song to track list
+                track_list.append(track)
                 song_list_text = "\n".join(
                     [f'<div>{t["name"]}</div><iframe src="https://open.spotify.com/embed/track/{t["id"]}" width="300" height="80" frameBorder="0" allowtransparency="true" allow="encrypted-media"></iframe>' for t in track_list]
                 )
-                return track_list, song_list_text  # Return updated track list and HTML for the list
+                return track_list, song_list_text
             return track_list, "Song not found."
 
-        # Trigger the add_song function when the Add Song button is clicked
         add_song_button.click(
             fn=add_song,
             inputs=[song_name, track_list_state, access_token_state],
@@ -220,16 +185,12 @@ def gradio_interface():
                 track_ids = [t['id'] for t in track_list]
                 recommended_tracks, recommended_track_names, spotify_embeds, _, _ = get_recommendations(track_ids, access_token)
                 if recommended_tracks:
-                    # Store the recommended tracks in the state
                     recommended_tracks_state.value = recommended_tracks
-
-                    # Generate HTML with song names above their embeds
                     recommendations_text = "\n".join(
                         [f'<div>{name}</div><iframe src="https://open.spotify.com/embed/track/{track_id}" width="300" height="80" frameBorder="0" allowtransparency="true" allow="encrypted-media"></iframe>'
                         for name, track_id in zip(recommended_track_names, [track['id'] for track in recommended_tracks])]
                     )
                     return recommendations_text
-                    print(recommendations_text)
                 return "No recommendations found."
             return "Please add at least one song."