import os
from tkinter import N

from spotifyclient import SpotifyClient

def main():
    plnum=0
    spotify_client = SpotifyClient(os.getenv(" "),
                                   os.getenv(" "))

    # get last played tracks
    #num_tracks_to_visualise = int(input("How many tracks would you like to visualise? "))
    num_tracks_to_visualise = 5 #(Max of 50)
    last_played_tracks = spotify_client.get_last_played_tracks(num_tracks_to_visualise)

    print(f"\nHere are the last {num_tracks_to_visualise} tracks you listened to on Spotify:")
    for index, track in enumerate(last_played_tracks):
        print(f"{index+1}- {track}")
        if index == 0:
            with open("Seed Tracks.txt", 'w') as f:
                f.write(f"{index+1}- {track} \n")
        else:
            with open("Seed Tracks.txt", 'a') as f:
                f.write(f"{index+1}- {track} \n")

        #f.close
    # choose which tracks to use as a seed to generate a playlist
    with open("indexs.txt") as f:
        for line in f:
            indexes = line
    ##indexes = input("\nEnter a list of up to 5 tracks you'd like to use as seeds. Use indexes separated by a space: ")
            indexes = indexes.split()
            seed_tracks = [last_played_tracks[int(index)-1] for index in indexes]

    # get recommended tracks based off seed tracks
            recommended_tracks = spotify_client.get_track_recommendations(seed_tracks)
            print("\nHere are the recommended tracks which will be included in your new playlist:")
            for index, track in enumerate(recommended_tracks):
                print(f"{index+1}- {track}")

    # get playlist name from user and create playlist
            playlist_name = "Tracks from index(s) " + line
            playlist = spotify_client.create_playlist(playlist_name)
            print(f"\nPlaylist '{playlist.name}' was created successfully.")

    # populate playlist with recommended tracks
            spotify_client.populate_playlist(playlist, recommended_tracks)
            print(f"\nRecommended tracks successfully uploaded to playlist '{playlist.name}'.")


if __name__ == "__main__":
    main()