import base64
import acrcloud
import os
import eyed3
import eyed3.id3.frames
from eyed3.id3.frames import UserTextFrame
import requests
import json
import re
from bs4 import BeautifulSoup
from genius_api import GeniusApi
from my_shazam_utility import shazam_recognize_song
from applemusic_api import AppleMusicApi
from Acrcloudretrieve import recognize_song, set_id3_tags_mp3
from Retrieve_lyrics import get_lyrics
from erhalten_alb_covers import save_and_embed_album_cover

def load_config():
    with open('D:/Eurydice/Encompassing Data by discerning/config/config.json', 'r') as config_file:
        config_data = json.load(config_file)
    return config_data

# Load the configuration on script start
config = load_config()

# Now also load Spotify credentials
CLIENT_ID = config['Spotify']['CLIENT_ID']
CLIENT_SECRET = config['Spotify']['CLIENT_SECRET']

# Instantiate Genius Api
genius_api = GeniusApi()

def get_user_choice():
    # Display a header
    print("=" * 50)
    print("Welcome to the Song Recognition Service!")
    print("=" * 50)

    # Provide instructions and options
    print("\nPlease select the recognition service you'd like to use:\n")
    print("  1: YoutubeACR - Fast and accurate music recognition")
    print("  2: Shazam - Discover music, artists, and lyrics in seconds")
    
    # Separator for aesthetic purposes
    print("-" * 50)

    # Input prompt
    choice = input("Enter your choice (1 or 2) and press Enter: ")
    
    # More flair to indicate processing/input received
    print("\n" + "." * 25 + " Processing " + "." * 25 + "\n")

    return choice

def add_or_update_txxx_frame(audiofile, description, value):
    found = False
    frames = audiofile.tag.frame_set.get(eyed3.id3.frames.USERTEXT_FID, [])
    for frame in frames:
        if frame.description == description:
            frame.text = value
            found = True
            break

    if not found:
        # Create a new TXXX frame without specifying encoding
        new_frame = eyed3.id3.frames.UserTextFrame(description=description, text=value)
        # Previously: When encoding was being passed
        # Now: Encoding isn't specified as it's not required or not supported based on the error
        if not frames:  # If it's the first frame of this type
            audiofile.tag.frame_set[eyed3.id3.frames.USERTEXT_FID] = [new_frame]
        else:
            frames.append(new_frame)  # Append to exisiting list of USERTEXT frames

# Your Spotify authentication and song search functions:
def authenticate_spotify(client_id, client_secret):
    auth_url = 'https://accounts.spotify.com/api/token'
    client_creds = f"{client_id}:{client_secret}"
    client_creds_b64 = base64.b64encode(client_creds.encode())
    headers = {'Authorization': f'Basic {client_creds_b64.decode()}'}
    data = {'grant_type': 'client_credentials'}
    response = requests.post(auth_url, headers=headers, data=data)
    access_token = response.json().get('access_token')
    return access_token

def search_spotify_for_song(access_token, artist_name, title):
    base_url = "https://api.spotify.com/v1/search"
    query = f"{title} artist:{artist_name}"
    headers = {"Authorization": f"Bearer {access_token}"}
    params = {"q": query, "type": "track", "limit": 1}
    response = requests.get(base_url, headers=headers, params=params)
    results = response.json()
    try:
        track_info = results['tracks']['items'][0]
        return track_info
    except IndexError:
        print("Song not found on Spotify.")
        return None

# Function to fetch lyrics from Genius
def get_lyrics_from_genius(artist_name, title):
    # Search for the song on Genius
    results = genius_api.get_search_by_songs(f"{artist_name} {title}")
    if results:
        song_info = results[0]['result']  # Take the most relevant result
        song_id = str(song_info['id'])
        song_details = genius_api.get_song_by_id(song_id, text_format='plain')
        return song_details.get('lyrics', "Lyrics not available.")
    return "Song not found on Genius."

def save_lyrics_to_lrc(lyrics, lrc_path):
    with open(lrc_path, 'w', encoding='utf-8') as lrc_file:
        lrc_file.write(lyrics)

def get_high_quality_album_art_url(song_info):
    images = song_info['album']['images']  # Get the list of image dicts
    if not images:
        return None  # No images available
    # Sort the images by size, pick the largest
    highest_quality_image = max(images, key=lambda x: x['width']*x['height'])
    return highest_quality_image['url']

def save_high_quality_album_art(image_url, file_path):
    try:
        response = requests.get(image_url, stream=True)
        if response.status_code == 200:
            with open(file_path, 'wb') as out_file:
                for chunk in response.iter_content(1024):
                    out_file.write(chunk)
            print(f"High quality album art saved: {file_path}")
            return True  # Indicate success
        else:
            print("Could not download the album art.")
    except Exception as e:
        print(f"Error saving high-quality album art: {e}")
    return False  # Indicate failure

def embed_album_art_to_song(file_path, image_path):
    try:
        audiofile = eyed3.load(file_path)
        if audiofile.tag is None:  # If the file has no tags, create a new tag
            audiofile.initTag()
        with open(image_path, 'rb') as img_file:
            audiofile.tag.images.set(3, img_file.read(), 'image/jpeg')
        audiofile.tag.save()
        print("High quality album art embedded into song.")
    except FileNotFoundError:
        print(f"Failed to embed album art - No such file: {image_path}")

def process_audio_file_with_spotify_search(audio_file_path):
    shazam_data = shazam_recognize_song(audio_file_path)
    if shazam_data:
        artist_name = shazam_data['track']['subtitle']
        title = shazam_data['track']['title']
        print(f"Identified Song: {artist_name} - {title}")
        
        access_token = authenticate_spotify(CLIENT_ID, CLIENT_SECRET)
        song_info = search_spotify_for_song(access_token, artist_name, title)
        
        if song_info:
            print(json.dumps(song_info, indent=4))  # For debugging
            print("\n///////////////////////////////\n")
            
            album_name = song_info['album']['name']
            album_url = song_info['album']['external_urls']['spotify']
            track_number = song_info['track_number']
            release_date = song_info['album']['release_date']
            isrc = song_info.get('external_ids', {}).get('isrc', "Not Available")
            label = song_info['label'] if 'label' in song_info else "Not Available"
            explicit = str(song_info['explicit']) if 'explicit' in song_info else "Not Available"  # Convert to string
            genres = ", ".join(song_info['genres']) if 'genres' in song_info else "Not Available"
            author_url = song_info['artists'][0]['external_urls']['spotify'] if 'artists' in song_info else "Not Available"
            spotify_url = song_info['external_urls']['spotify']
            
            print(f"Track Number on Spotify: {track_number}")
            
            audiofile = eyed3.load(audio_file_path)
            if audiofile.tag is None:  # If the file has no tags, create a new tag
                audiofile.initTag(version=eyed3.id3.ID3_V2_3)
            
            # Set standard tags
            audiofile.tag.artist = artist_name
            audiofile.tag.album = album_name
            audiofile.tag.album_artist = artist_name
            audiofile.tag.title = title
            audiofile.tag.recording_date = release_date
            
            # Using helper function to add or update TXXX frames
            add_or_update_txxx_frame(audiofile, "Album URL", album_url)
            add_or_update_txxx_frame(audiofile, "Eurydice", "True")
            add_or_update_txxx_frame(audiofile, "Compilation", "KK")
            add_or_update_txxx_frame(audiofile, "Genre", genres)
            add_or_update_txxx_frame(audiofile, "Author URL", author_url)
            add_or_update_txxx_frame(audiofile, "Label", label)
            add_or_update_txxx_frame(audiofile, "Explicit", explicit)
            add_or_update_txxx_frame(audiofile, "ISRC", isrc)
            add_or_update_txxx_frame(audiofile, "Spotify URL", spotify_url)
            
            audiofile.tag.comments.set(f"ISRC: {isrc}, Label: {label}, Explicit: {explicit}")
            
            audiofile.tag.save()  # Save the metadata to the file
            
            print(f"Metadata embedded into the file: {audio_file_path}")

            # Fetch high-quality album art URL
            high_res_image_url = get_high_quality_album_art_url(song_info)
            
            if high_res_image_url:
                # Determine paths
                image_file_path = os.path.splitext(audio_file_path)[0] + ".jpg"

                # Save and embed album art
                if save_high_quality_album_art(high_res_image_url, image_file_path):
                    embed_album_art_to_song(audio_file_path, image_file_path)
                else:
                    print("Skipping album art embed due to download failure.")
            else:
                print("No album art available.")

            new_file_name = f"{track_number:02d}. {title} - {artist_name} - {album_name} - {isrc}.mp3"
            new_file_name = re.sub(r'[/:*?"<>|]', '', new_file_name)  # Clean up characters not allowed in file names
            new_file_path = os.path.join(os.path.dirname(audio_file_path), new_file_name)
            os.rename(audio_file_path, new_file_path)  # Rename file
            print(f"File has been renamed to: {new_file_name}")

            new_image_file_path = os.path.splitext(new_file_path)[0] + ".jpg"
            os.rename(image_file_path, new_image_file_path)
            print(f"Album art file has been renamed to: {os.path.basename(new_image_file_path)}")

            new_lyrics_file_path = os.path.split(new_file_path)

            lyrics = get_lyrics_from_genius(artist_name, title)
            if 'plain' in lyrics:
                lyrics_plain_text = lyrics['plain']
                print("Lyrics:\n", lyrics_plain_text)
                # base_directory = os.path.dirname(audio_file_path)
                lyrics_file_path = os.path.splitext(audio_file_path)[0] + ".lrc"
                with open(lyrics_file_path, 'w', encoding='utf-8') as lrc_file:
                    lrc_file.write(lyrics_plain_text)
                # save_lyrics_to_lrc(lyrics_plain_text, track_number, title, artist_name, album_name, isrc, base_directory)
                # save_lyrics_to_lrc(lyrics_plain_text, new_file_name)
                # print(f"Lyrics saved to: {new_file_name}")
                print(f"Saved LRC lyrics to: {lyrics_file_path}")
            else:
                print("No lyrics available to save.")

        else:
            print("Song not found on Spotify.")
    else:
        print("Song could not be identified.")

if __name__ == "__main__":
    user_choice = get_user_choice()
    audio_file_path = 'D:/Eurydice/Encompassing Data by discerning/Test_file/Unknown_file.mp3'
    if user_choice == '1':
        print("\n" + "." * 15 + " ᴜsɪɴɢ YᴏᴜᴛᴜʙᴇACR " + "." * 15 + "\n")
        song_tags = recognize_song(audio_file_path)

        if song_tags:
            print(f'Song identified: {song_tags}')
            set_id3_tags_mp3(audio_file_path, song_tags)

            artist_name = song_tags.get('artists')[0].get('name')
            song_title = song_tags.get('title')
        
            safe_artist_name = re.sub(r'[/\:?"<>|]', '', artist_name)
            safe_song_title = re.sub(r'[/\:?"<>|]', '', song_title)
            new_file_name = f"{safe_artist_name} - {safe_song_title}.mp3"
        
            new_file_path = os.path.join(os.path.dirname(audio_file_path), new_file_name)
            os.rename(audio_file_path, new_file_path)
            print(f"File has been renamed to: {new_file_name}")

        else:
            print('Could not identify the song in YᴏᴜᴛᴜʙᴇACR.')

        apple_music_api = AppleMusicApi(Exception)  # Initialize AppleMusicApi with necessary authentication
        apple_music_api.get_access_token()
        track_results = apple_music_api.search('songs', f"{artist_name} - {song_title}")
        if track_results:
            track_id = track_results[0]['id']
            album_artwork_url_template = track_results[0]['attributes']['artwork']['url']

            save_and_embed_album_cover(new_file_path, artist_name, song_title, album_artwork_url_template)
        else:
            print("Song not found on Apple Music.")
        
        lrc_lyrics = get_lyrics(safe_artist_name, safe_song_title)
        if lrc_lyrics:
            lrc_file_path = os.path.join(os.path.dirname(audio_file_path), f"{safe_artist_name} - {safe_song_title}.lrc")
            with open(lrc_file_path, 'w', encoding='utf-8') as lrc_file:
                lrc_file.write(lrc_lyrics)
            print(f"Saved LRC file to: {lrc_file_path}")
        else:
            print("Could not get the lyrics.")

    elif user_choice == '2':
        print("\n" + "." * 15 + " ᴜsɪɴɢ Sʜᴀᴢᴀᴍ " + "." * 15 + "\n")
        song_tags = shazam_recognize_song(audio_file_path)
        print(song_tags)
        process_audio_file_with_spotify_search(audio_file_path)

    else:
        print("Invalid choice. Exiting....")
        exit()

after fetching lyrics from genius can you rename the file to track number, track name, artist name, album name, isrc tag