main.py

import streamlit as st
from streamlit_option_menu import option_menu
import streamlit.components.v1 as components
import os
import time
import spotipy
from spotipy.oauth2 import SpotifyOAuth, SpotifyClientCredentials
import requests
from spotifysearch import *
from model import *


#Authentication - without user
#client_credentials_manager = SpotifyClientCredentials(client_id=Client_id, client_secret=client_secret)
#sp = spotipy.Spotify(client_credentials_manager = client_credentials_manager)

Client_id=os.environ['Client_ID']
client_secret=os.environ['Client_secret']
auth_manager = SpotifyClientCredentials(client_id=Client_id, client_secret=client_secret)
sp = spotipy.client.Spotify(auth_manager=auth_manager) 






if 'model' not in st.session_state:
    st.session_state.model = 'Similarity Model'
def update_radio2():
    st.session_state.model=st.session_state.radio2
if 'genre' not in st.session_state:
    st.session_state.genre=3
def update_num_genre():
    st.session_state.genre=st.session_state.num_genre
if 'artist' not in st.session_state:
    st.session_state.artist=5
def update_same_art():
    st.session_state.artist=st.session_state.same_art
if 'model2' not in st.session_state:
    st.session_state.model2= 'Similarity Model'
def update_radio1():
    st.session_state.model2 =st.session_state.radio1

if 'Region' not in st.session_state:
    st.session_state.rg="US"
def update_Region():
    st.session_state.rg=st.session_state.Region
if 'radio' not in st.session_state:
    st.session_state.feature="Song"
def update_radio0():
    st.session_state.feature=st.session_state.radio

if 'p_url' not in st.session_state:
    st.session_state.p_url = 'Example: https://open.spotify.com/playlist/37i9dQZF1DX8FwnYE6PRvL?si=06ff6b38d4124af0'
def update_playlist_url():
    st.session_state.p_url = st.session_state.playlist_url

if 's_url' not in st.session_state:
    st.session_state.s_url = 'Example: https://open.spotify.com/track/5CQ30WqJwcep0pYcV4AMNc?si=ed4b04f153a24531'
def update_song_url():
    st.session_state.s_url = st.session_state.song_url

if 'sn_url' not in st.session_state:
    st.session_state.sn_url = 'In The End'
def update_song_name_url():
    st.session_state.sn_url = st.session_state.songname_url

if 'a_url' not in st.session_state:
    st.session_state.a_url = 'Example: https://open.spotify.com/artist/3RNrq3jvMZxD9ZyoOZbQOD?si=UNAsX20kRpG89bxOO8o7ew'
def update_artist_url():
    st.session_state.a_url = st.session_state.artist_url


def play_recomm():
    if 'rs' in st.session_state:
        del st.session_state.rs,st.session_state.err
    try:
        if len(pd.read_csv('Data/new_tracks.csv')) >= 200:
            with st.spinner('Updating the dataset...'):
                x=update_dataset()
                st.success('{} New tracks were added to the dataset.'.format(x))
    except:
        st.error("The dataset update failed. ")
    with st.spinner('Getting Recommendations...'):
        res,err = playlist_model(st.session_state.p_url,st.session_state.model,st.session_state.genre,st.session_state.artist)
        st.session_state.rs=res
        st.session_state.err=err
    if len(st.session_state.rs)>=1:
        if st.session_state.model == 'ReModel' or st.session_state.model == 'Experimental Model':
            st.success('Go to the Result page to view the top {} recommendations, Thanks for taking the time to use this :D'.format(len(st.session_state.rs)))
            st.success('- RedValis'.format(len(st.session_state.rs)))

        else:
            st.success('Go to the Result page to view the  Spotify recommendations')
            st.success('- RedValis')
    else:
        st.error('Model failed. Check the log for more information.')   

def art_recomm():
    if 'rs' in st.session_state:
        del st.session_state.rs,st.session_state.err
    with st.spinner('Getting Recommendations...'):
        res,err = top_tracks(st.session_state.a_url,st.session_state.rg)
        st.session_state.rs=res
        st.session_state.err=err
    if len(st.session_state.rs)>=1:
        st.success("Go to the Result page to view the Artist's top tracks, Thank you for taking the time to use this :D")
        st.success("- RedValis")
    else:
        st.error('Model failed. Check the log for more information.')

def song_recomm():
    if 'rs' in st.session_state:
        del st.session_state.rs,st.session_state.err
    with st.spinner('Getting Recommendations...'):
        res,err = song_model(st.session_state.s_url,st.session_state.model,st.session_state.genre,st.session_state.artist)
        st.session_state.rs=res
        st.session_state.err=err
    if len(st.session_state.rs)>=1:
        if st.session_state.model == 'ReModel' or st.session_state.model == 'Experimental Model':
            st.success('Go to the Result page to view the top {} recommendations, Thank you for taking the time to use this :D'.format(len(st.session_state.rs)))
            st.success('- RedValis'.format(len(st.session_state.rs)))
        else:
            st.success('Go to the Result page to view the  Spotify recommendations')
            st.success('- RedValis')
    else:
        st.error('Model failed. Check the log for more information.')

def playlist_page():
    st.subheader("User Playlist")
    st.markdown('---')
    playlist_uri = (st.session_state.playlist_url).split('/')[-1].split('?')[0]
    uri_link = 'https://open.spotify.com/embed/playlist/' + playlist_uri
    components.iframe(uri_link, height=300)
    return

def song_page():
    st.subheader("User Song")
    st.markdown('---')
    song_uri = (st.session_state.song_url).split('/')[-1].split('?')[0]
    uri_link = 'https://open.spotify.com/embed/track/' + song_uri
    components.iframe(uri_link, height=100)

def artist_page():
    st.subheader("User Artist")
    st.markdown('---')
    artist_uri = (st.session_state.artist_url).split('/')[-1].split('?')[0]
    uri_link = 'https://open.spotify.com/embed/artist/' + artist_uri
    components.iframe(uri_link, height=80)


def spr_sidebar():
    menu=option_menu(
        menu_title=None,
        options=['Home','Result','About','Log'],
        icons=['house','book','info-square','terminal'],
        menu_icon='cast',
        default_index=0,
        orientation='horizontal'
    )
    if menu=='Home':
        st.session_state.app_mode = 'Home'
    elif menu=='Result':
        st.session_state.app_mode = 'Result'
    elif menu=='About':
        st.session_state.app_mode = 'About'
    elif menu=='Log':
        st.session_state.app_mode = 'Log'
    
def home_page():
    st.session_state.radio=st.session_state.feature
    st.session_state.radio2=st.session_state.model
    st.session_state.num_genre=st.session_state.genre
    st.session_state.same_art=st.session_state.artist
    st.session_state.Region=st.session_state.rg

    
    st.title('Helical Recommendation System')
    col,col2,col3=st.columns([2,2,3])
    radio=col.radio("Feature",options=("Playlist","Song","Artist Top Tracks"),key='radio',on_change=update_radio0)
    if radio =="Artist Top Tracks":
        radio1=col2.radio("Model",options=["Similarity Model"],key='radio1',on_change=update_radio1)
        Region=col3.selectbox("Please Choose Region",index=58,key='Region',on_change=update_Region,options=('AD', 'AR', 'AU', 'AT', 'BE', 'BO', 'BR', 'BG', 'CA', 'CL', 'CO', 'CR', 'CY', 'CZ', 'DK', 'DO', 'EC', 'SV', 'EE', 'FI', 'FR', 'DE', 'GR', 'GT', 'HN', 'HK', 'HU', 'IS', 'ID', 'IE', 'IT', 'JP', 'LV', 'LI', 'LT', 'LU', 'MY', 'MT', 'MX', 'MC', 'NL', 'NZ', 'NI', 'NO', 'PA', 'PY', 'PE', 'PH', 'PL', 'PT', 'SG', 'ES', 'SK', 'SE', 'CH', 'TW', 'TR', 'GB', 'US', 'UY'))
    elif radio =="Playlist" or radio =="Song" :
        radio2=col2.radio("Model",options=("ReModel","Experimental Model","Similarity Model"),key='radio2',on_change=update_radio2)
        if st.session_state.radio2=="ReModel" or st.session_state.radio2=="Experimental Model":
            num_genre=col3.selectbox("choose a number of genres to focus on",options=(1,2,3,4,5,6,7),index=2,key='num_genre',on_change=update_num_genre)
            same_art=col3.selectbox("How many recommendations by the same artist",options=(1,2,3,4,5,7,10,15),index=3,key='same_art',on_change=update_same_art)


    st.markdown("<br>", unsafe_allow_html=True)
    
    if radio == "Playlist" :
        st.session_state.playlist_url = st.session_state.p_url
        Url = st.text_input(label="Playlist Url",key='playlist_url',on_change=update_playlist_url)
        playlist_page()
        state =st.button('Get Recommendations')
        with st.expander("Here's how to find any Playlist URL in Spotify"):
            st.write(""" 
                - Search for Playlist on the Spotify app
                - Right Click on the Playlist you like
                - Click "Share"
                - Choose "Copy link to playlist"
            """)
            st.markdown("<br>", unsafe_allow_html=True)
            st.image('spotify_get_playlist_url.png')
        if state:
            play_recomm()
    elif radio == "Song" :
        st.session_state.songname_url = st.session_state.sn_url
        Url_a = st.text_input(label="Type a songs name to get the url for",key='get_url',on_change=update_song_name_url)

        from spotifysearch.client import Client
        myclient = Client(Client_id, client_secret)
        results = myclient.search(Url_a)
        tracks = results.get_tracks()

        track = tracks
        
        result_length = len(tracks)

        track_song_list = []
        track_url_list = []
        track_actual_url_list = []
        anotherlist=[]


        genrec = st.radio(
            "Choose an option to search in:",
            ["5x4"])


        if genrec == "Full":
        
            for leupz in range(result_length):
                thetrack = tracks[leupz]
                st.write(thetrack.name)
                name_desu = thetrack.name
                track_song_list.append(name_desu)
            
                st.write(thetrack.url)            
                urlname_desu = thetrack.name           
                track_url_list.append(urlname_desu)      
            
                song_uri = (thetrack.url).split('/')[-1].split('?')[0]
                uri_link = 'https://open.spotify.com/embed/track/' + song_uri
                components.iframe(uri_link, height=100)


            st.write(result_length)

        elif genrec == "5x4":
            for leupz in range(result_length):
                thetrack = tracks[leupz]

                name_desu = thetrack.name
                track_song_list.append(name_desu)
            
       
                urlname_desu = thetrack.url          
                track_url_list.append(urlname_desu)      
            
                song_uri = (thetrack.url).split('/')[-1].split('?')[0]
                uri_link = 'https://open.spotify.com/embed/track/' + song_uri
                dem = uri_link

                track_actual_url_list.append(dem)
            notneeded_chkbox = st.checkbox('Click on this to see the results')

            if notneeded_chkbox:
########### WARNING: MAY CAUSE SOME TO FAINT :/
                
                st.write('Feature activated!')

                pages = st.radio("Page",["1","2","3","4",], horizontal=True)

                if pages == "1":
                    Le_songs = st.radio(
                        "Its time to choose...",
                        [track_song_list[0],track_song_list[1],track_song_list[2],track_song_list[3],track_song_list[4]])
                    
                    if Le_songs == track_song_list[0]:
                        st.session_state.s_url = track_actual_url_list[0]
                        
                    elif Le_songs == track_song_list[1]:
                        st.session_state.s_url = track_actual_url_list[1]
                        
                    elif Le_songs == track_song_list[2]:
                        st.session_state.s_url = track_actual_url_list[2]
                    
                    elif Le_songs == track_song_list[3]:
                        st.session_state.s_url = track_actual_url_list[3]
                        
                    elif Le_songs == track_song_list[4]:
                        st.session_state.s_url = track_actual_url_list[4]
                        
                        
                    st.write('Previews')
                
                    components.iframe(track_actual_url_list[0], height=100)
                    components.iframe(track_actual_url_list[1], height=100)
                    components.iframe(track_actual_url_list[2], height=100)
                    components.iframe(track_actual_url_list[3], height=100)
                    components.iframe(track_actual_url_list[4], height=100)
                    

                if pages == "2":
                    Le_songs = st.radio(
                        "Its time to choose...",
                        [track_song_list[5],track_song_list[6],track_song_list[7],track_song_list[8],track_song_list[9]])
                    if Le_songs == track_song_list[5]:
                        st.session_state.s_url = track_actual_url_list[5]
                        
                    elif Le_songs == track_song_list[6]:
                        st.session_state.s_url = track_actual_url_list[6]
                        
                    elif Le_songs == track_song_list[7]:
                        st.session_state.s_url = track_actual_url_list[7]
                    
                    elif Le_songs == track_song_list[8]:
                        st.session_state.s_url = track_actual_url_list[8]
                        
                    elif Le_songs == track_song_list[9]:
                        st.session_state.s_url = track_actual_url_list[9]
                        
                    st.write('Previews')
                
                    components.iframe(track_actual_url_list[5], height=100)
                    components.iframe(track_actual_url_list[6], height=100)
                    components.iframe(track_actual_url_list[7], height=100)
                    components.iframe(track_actual_url_list[8], height=100)
                    components.iframe(track_actual_url_list[9], height=100)
                if pages == "3":
                    Le_songs = st.radio(
                        "Its time to choose...",
                        [track_song_list[10],track_song_list[11],track_song_list[12],track_song_list[13],track_song_list[14]])

                    if Le_songs == track_song_list[10]:
                        st.session_state.s_url = track_actual_url_list[10]
                        
                    elif Le_songs == track_song_list[11]:
                        st.session_state.s_url = track_actual_url_list[11]
                        
                    elif Le_songs == track_song_list[12]:
                        st.session_state.s_url = track_actual_url_list[12]
                    
                    elif Le_songs == track_song_list[13]:
                        st.session_state.s_url = track_actual_url_list[13]
                        
                    elif Le_songs == track_song_list[14]:
                        st.session_state.s_url = track_actual_url_list[14]
                    
                    st.write('Previews')
                
                    components.iframe(track_actual_url_list[10], height=100)
                    components.iframe(track_actual_url_list[11], height=100)
                    components.iframe(track_actual_url_list[12], height=100)
                    components.iframe(track_actual_url_list[13], height=100)
                    components.iframe(track_actual_url_list[14], height=100)
                if pages == "4":
                    Le_songs = st.radio(
                        "Its time to choose...",
                        [track_song_list[15],track_song_list[16],track_song_list[17],track_song_list[18],track_song_list[19]])

                    if Le_songs == track_song_list[15]:
                        st.session_state.s_url = track_actual_url_list[15]
                        
                    elif Le_songs == track_song_list[16]:
                        st.session_state.s_url = track_actual_url_list[16]
                        
                    elif Le_songs == track_song_list[17]:
                        st.session_state.s_url = track_actual_url_list[17]
                    
                    elif Le_songs == track_song_list[18]:
                        st.session_state.s_url = track_actual_url_list[18]
                        
                    elif Le_songs == track_song_list[19]:
                        st.session_state.s_url = track_actual_url_list[19]
                    
                    st.write('Previews')
                
                    components.iframe(track_actual_url_list[15], height=100)
                    components.iframe(track_actual_url_list[16], height=100)
                    components.iframe(track_actual_url_list[17], height=100)
                    components.iframe(track_actual_url_list[18], height=100)
                    components.iframe(track_actual_url_list[19], height=100)
                    
                else:
                    st.write("""If you have clicked on an option, click on "Get recommendations" button to get recommendations""")

     
                st.write(result_length)
        
        st.session_state.song_url = st.session_state.s_url
        Url = st.text_input(label="Song Url ",key='song_url',on_change=update_song_url)
    
        song_page()
        state =st.button('Get Recommendations')
        with st.expander("Here's how to find any Song URL in Spotify"):
            st.write(""" 
                - Search for Song on the Spotify app
                - Right Click on the Song you like
                - Click "Share"
                - Choose "Copy link to Song"
            """)
            st.markdown("<br>", unsafe_allow_html=True)
            st.image('spotify_get_song_url.png')

            
        if state:
            song_recomm()
    elif radio == "Artist Top Tracks" :
        st.session_state.artist_url = st.session_state.a_url
        Url = st.text_input(label="Artist Url",key='artist_url',on_change=update_artist_url)
        artist_page()
        state =st.button('Get Recommendations')
        with st.expander("Here's how to find any Artist URL in Spotify"):
            st.write(""" 
                - Search for Artist on the Spotify app
                - Right Click on the Artist you like
                - Click "Share"
                - Choose "Copy link to Artist"
            """)
            st.markdown("<br>", unsafe_allow_html=True)
            st.image('spotify_get_artist_url.png')
        if state:
            art_recomm()
    
def result_page():
    if 'rs' not in st.session_state:
        st.error('Please select a model on the Home page and run Get Recommendations')
    else:
        st.success('Top {} recommendations'.format(len(st.session_state.rs)))
        i=0
        for uri in st.session_state.rs:
         uri_link = "https://open.spotify.com/embed/track/" + uri + "?utm_source=generator&theme=0"
         components.iframe(uri_link, height=80)
         i+=1
         if i%5==0:
            time.sleep(1)
def Log_page():
    log=st.checkbox('Display Output', True, key='display_output')
    if log == True:
     if 'err' in st.session_state:
        st.write(st.session_state.err)
    with open('Data/streamlit.csv') as f:
        st.download_button('Download Dataset', f,file_name='streamlit.csv')
def About_page():
    st.header('Development')
    """
    Made by RedValis
    Massive thanks to ruby  "AbdelRahman" skies from github for code snippets that helped in the creation of our project
    """
    st.subheader('Spotify Million Playlist Dataset')
    """
    We're using the Million Playlist Dataset, which, as its name implies, consists of one million playlists.
    contains a number of songs, and some metadata is included as well, such as the name of the playlist, duration, number of songs, number of artists, etc.
    This allows for a great accuracy score and a larger number of songs scored according to how people organize their public playlists
    """

    """
    It is created by sampling playlists from the billions of playlists that Spotify users have created over the years. 
    Playlists that meet the following criteria were selected at random:
    - Created by a user that resides in the United States and is at least 13 years old
    - Was a public playlist at the time the MPD was generated
    - Contains at least 10 tracks
    - Contains no more than 300 tracks
    - Contains at least 3 unique artists
    - Contains at least 2 unique albums
    - Has no local tracks (local tracks are non-Spotify tracks that a user has on their local device
    - Has at least one follower (not including the creator
    - Was created after January 1, 2010 and before December 1, 2020
    - Does not have an offensive title
    - Does not have an adult-oriented title if the playlist was created by a user under 18 years of age
    Information about the Dataset [here](https://www.aicrowd.com/challenges/spotify-million-playlist-dataset-challenge)
    """
    st.subheader('Audio Features Explanation')
    """
    This is the list of audio features that we use to determine the similarity between songs.
    For anyone who would like to inspect the source code 
    
    | Variable | Description |
    | :----: | :---: |
    | Acousticness | A confidence measure from 0.0 to 1.0 of whether the track is acoustic. 1.0 represents high confidence the track is acoustic. |
    | Danceability | Danceability describes how suitable a track is for dancing based on a combination of musical elements including tempo, rhythm stability, beat strength, and overall regularity. A value of 0.0 is least danceable and 1.0 is most danceable. |
    | Energy | Energy is a measure from 0.0 to 1.0 and represents a perceptual measure of intensity and activity. Typically, energetic tracks feel fast, loud, and noisy. For example, death metal has high energy, while a Bach prelude scores low on the scale. Perceptual features contributing to this attribute include dynamic range, perceived loudness, timbre, onset rate, and general entropy. |
    | Instrumentalness | Predicts whether a track contains no vocals. "Ooh" and "aah" sounds are treated as instrumental in this context. Rap or spoken word tracks are clearly "vocal". The closer the instrumentalness value is to 1.0, the greater likelihood the track contains no vocal content. Values above 0.5 are intended to represent instrumental tracks, but confidence is higher as the value approaches 1.0. |
    | Key | The key the track is in. Integers map to pitches using standard Pitch Class notation. E.g. 0 = C, 1 = C♯/D♭, 2 = D, and so on. If no key was detected, the value is -1. |
    | Liveness | Detects the presence of an audience in the recording. Higher liveness values represent an increased probability that the track was performed live. A value above 0.8 provides strong likelihood that the track is live. |
    | Loudness | The overall loudness of a track in decibels (dB). Loudness values are averaged across the entire track and are useful for comparing relative loudness of tracks. Loudness is the quality of a sound that is the primary psychological correlate of physical strength (amplitude). Values typically range between -60 and 0 db. |
    | Mode | Mode indicates the modality (major or minor) of a track, the type of scale from which its melodic content is derived. Major is represented by 1 and minor is 0. |
    | Speechiness | Speechiness detects the presence of spoken words in a track. The more exclusively speech-like the recording (e.g. talk show, audio book, poetry), the closer to 1.0 the attribute value. Values above 0.66 describe tracks that are probably made entirely of spoken words. Values between 0.33 and 0.66 describe tracks that may contain both music and speech, either in sections or layered, including such cases as rap music. Values below 0.33 most likely represent music and other non-speech-like tracks. |
    | Tempo | The overall estimated tempo of a track in beats per minute (BPM). In musical terminology, tempo is the speed or pace of a given piece and derives directly from the average beat duration. |
    | Time Signature | An estimated time signature. The time signature (meter) is a notational convention to specify how many beats are in each bar (or measure). The time signature ranges from 3 to 7 indicating time signatures of "3/4", to "7/4". |
    | Valence | A measure from 0.0 to 1.0 describing the musical positiveness conveyed by a track. Tracks with high valence sound more positive (e.g. happy, cheerful, euphoric), while tracks with low valence sound more negative (e.g. sad, depressed, angry). |
    
    Information about features: [here](https://developer.spotify.com/documentation/web-api/reference/#/operations/get-audio-features)
    """

def main():
    spr_sidebar()        
    if st.session_state.app_mode == 'Home':
        home_page()
    if st.session_state.app_mode == 'Result':
        result_page()
    if st.session_state.app_mode == 'About' :
        About_page()
    if st.session_state.app_mode == 'Log':
        Log_page()
# Run main()
#if __name__ == '__main__': this doesnt allow reletive imports >:()
main()