Upload 2 files

main.py

@@ -5,6 +5,7 @@ import time
 from model import *
 from Explanations.mistral import MistralExplainer, mistral_2_display
 from Explanations.llama2 import Llama2Explainer
+import json
 
 if 'model' not in st.session_state:
     st.session_state.model = 'Model 1'
@@ -65,23 +66,23 @@ if 'explanation_length' not in st.session_state:
 if 'selected_song_index' not in st.session_state:
     st.session_state.selected_song_index = 0
     
-song_list = [{"title": "The Strokes - Someday", "uri" : "https://open.spotify.com/track/7hm4HTk9encxT0LYC0J6oI?si=9d23a7470dfb4330", "next_five": ['Figure It Out by Royal Blood', 'Joker And The Thief by Wolfmother', 'Heads Will Roll - A-Trak Remix Radio Edit by Yeah Yeah Yeahs', 'Hate To Say I Told You So by The Hives', 'Get Free by The Vines']}, 
-                 {"title": "Doja Cat - Paint the Town Red", "uri" : "https://open.spotify.com/track/4dHRp03oxqJQfjGU8ECo7v?si=75a8b6b9ae304963", "next_five":["I'm Good (Blue) by David Guetta", 'Dance The Night - From Barbie The Album by Dua Lipa', 'Super Shy by NewJeans', 'Snooze by SZA', 'Watermelon Sugar by Harry Styles']},
-                 {"title": "Beyonce - Single Ladies","uri":"https://open.spotify.com/track/5R9a4t5t5O0IsznsrKPVro?si=d7f7f6b342b446c8", "next_five": ['Locked out of Heaven by Bruno Mars', '...Baby One More Time by Britney Spears', 'I Wanna Dance with Somebody (Who Loves Me) by Whitney Houston', 'Disturbia by Rihanna', 'Too Good At Goodbyes by Sam Smith']},
-                 {"title": "Dolly Parton - Jolene","uri":"https://open.spotify.com/track/2SpEHTbUuebeLkgs9QB7Ue?si=ab915579c1244a5f", "next_five":['Wide Open Spaces by The Chicks', 'Blue Eyes Crying In the Rain by Willie Nelson', 'Need You Now by Lady A', "Mama's Broken Heart by Miranda Lambert", "She's In Love With The Boy - Single Version by Trisha Yearwood"]},
-                 {"title": "Taylor Swift - Lavender Haze","uri":"https://open.spotify.com/track/5jQI2r1RdgtuT8S3iG8zFC?si=5922d57584b84382", "next_five":['Dance The Night - From Barbie The Album by Dua Lipa', "I'm Good (Blue) by David Guetta", 'Perfect by Ed Sheeran', "Doin' Time by Lana Del Rey", 'Sweater Weather by The Neighbourhood']},
-                 {"title": "Lincoln Park - What I've Done","uri":"https://open.spotify.com/track/2nLtzopw4rPReszdYBJU6h?si=1b48d972e12d43e7", "next_five":['Higher by Creed', 'Drive by Incubus', 'I Hate Everything About You by Three Days Grace', 'The Reason by Hoobastank', 'Butterfly by Crazy Town']},
-                 {"title": "Avicii - Hey Brother","uri":"https://open.spotify.com/track/4lhqb6JvbHId48OUJGwymk?si=d148732daa8e4897", "next_five": ['Silence by Marshmello', 'Sugar (feat. Francesco Yates) by Robin Schulz', 'Where Did You Go? (feat. MNEK) by Jax Jones', 'Rockabye (feat. Sean Paul & Anne-Marie) by Clean Bandit', 'Lush Life by Zara Larsson']},
-                 {"title": "Simon & Garfunkel - Sound of Silence","uri":"https://open.spotify.com/track/3YfS47QufnLDFA71FUsgCM?si=5bff00bb3b2e4a67", "next_five": ['American Pie by Don McLean', 'The Weight - Remastered 2000 by The Band', "You're So Vain by Carly Simon", "Do You Believe in Magic? by The Lovin' Spoonful", "You've Got a Friend by Carole King"]},
-                 {"title": "Queen - Bohemian Rhapsody","uri":"https://open.spotify.com/track/6l8GvAyoUZwWDgF1e4822w?si=9639498e177d46e6", "next_five": ['You Really Got Me by The Kinks', 'Walk On the Wild Side by Lou Reed', 'In The Army Now by Status Quo', 'Surrender by Cheap Trick', 'More Than This by Roxy Music']},
-                 {"title": "Frédéric Chopin - Nocturnes Op. 9 N. 2","uri":"https://open.spotify.com/track/61YM5SkqqeUjIBL7It56cs?si=01a05e921c61409b", "next_five":['5 Stücke im Volkston', 'Op. 102: II. Langsam by Robert Schumann', 'Paganini: 24 Caprices', 'Op. 1: No. 24 in A Minor by Niccolò Paganini', 'I puritani / Act I: A te', 'o cara by Vincenzo Bellini', 'Strauss I: Radetzky March', 'Op. 228 by Johann Strauss I', 'Clarinet Concerto No. 1 in F Minor', 'Op. 73: III. Rondo: Allegretto by Carl Maria von Weber', 'Scherzo-tarantelle in G Minor', 'Op.16 - 1987 Remastered Version by Henryk Wieniawski']}
+song_list = [{"title": "Someday by The Strokes", "uri" : "https://open.spotify.com/track/7hm4HTk9encxT0LYC0J6oI?si=9d23a7470dfb4330", "next_five": [{'full_track_name': 'Figure It Out by Royal Blood', 'uri': '3MjrueDQKVr6xDDseZwhEd'}, {'full_track_name': 'Joker And The Thief by Wolfmother', 'uri': '7ISL3LO8AWP3fKIXunvqTa'}, {'full_track_name': 'Heads Will Roll - A-Trak Remix Radio Edit by Yeah Yeah Yeahs', 'uri': '5qGIfWDRdyO67LCTqA2sEn'}, {'full_track_name': 'Hate To Say I Told You So by The Hives', 'uri': '20j7B6mL9gnNvG3ic6Vwh5'}, {'full_track_name': 'Get Free by The Vines', 'uri': '1JcGNoiwifg0MdJMVgJQYx'}]}, 
+                 {"title": "Paint the Town Red by Doja Cat", "uri" : "https://open.spotify.com/track/4dHRp03oxqJQfjGU8ECo7v?si=75a8b6b9ae304963", "next_five":[{'full_track_name': 'Locked out of Heaven by Bruno Mars', 'uri': '3w3y8KPTfNeOKPiqUTakBh'}, {'full_track_name': '...Baby One More Time by Britney Spears', 'uri': '3MjUtNVVq3C8Fn0MP3zhXa'}, {'full_track_name': 'I Wanna Dance with Somebody (Who Loves Me) by Whitney Houston', 'uri': '2tUBqZG2AbRi7Q0BIrVrEj'}, {'full_track_name': 'Disturbia by Rihanna', 'uri': '2VOomzT6VavJOGBeySqaMc'}, {'full_track_name': 'Too Good At Goodbyes by Sam Smith', 'uri': '1mXVgsBdtIVeCLJnSnmtdV'}]},
+                 {"title": "Single Ladies by Beyonce","uri":"https://open.spotify.com/track/5R9a4t5t5O0IsznsrKPVro?si=d7f7f6b342b446c8", "next_five":[{'full_track_name': "I'm Good (Blue) by David Guetta", 'uri': '4uUG5RXrOk84mYEfFvj3cK'}, {'full_track_name': 'Dance The Night - From Barbie The Album by Dua Lipa', 'uri': '1vYXt7VSjH9JIM5oRRo7vA'}, {'full_track_name': 'Super Shy by NewJeans', 'uri': '5sdQOyqq2IDhvmx2lHOpwd'}, {'full_track_name': 'Snooze by SZA', 'uri': '4iZ4pt7kvcaH6Yo8UoZ4s2'}, {'full_track_name': 'Watermelon Sugar by Harry Styles', 'uri': '6UelLqGlWMcVH1E5c4H7lY'}] },
+                 {"title": "Jolene by Dolly Parton","uri":"https://open.spotify.com/track/2SpEHTbUuebeLkgs9QB7Ue?si=ab915579c1244a5f", "next_five":[{'full_track_name': 'Wide Open Spaces by The Chicks', 'uri': '6cjwec9ii5uLK7CDfPBYt1'}, {'full_track_name': 'Blue Eyes Crying In the Rain by Willie Nelson', 'uri': '2uRVPeQbsEpRQD0DKr1WTo'}, {'full_track_name': 'Need You Now by Lady A', 'uri': '11EX5yhxr9Ihl3IN1asrfK'}, {'full_track_name': "Mama's Broken Heart by Miranda Lambert", 'uri': '2mXsZsJBThTaJXbNcg5QD4'}, {'full_track_name': "She's In Love With The Boy - Single Version by Trisha Yearwood", 'uri': '2ulBBx6YQ3qY3ci34RadtN'}]},
+                 {"title": "Lavender Haze by Taylor Swift","uri":"https://open.spotify.com/track/5jQI2r1RdgtuT8S3iG8zFC?si=5922d57584b84382", "next_five":[{'full_track_name': 'Dance The Night - From Barbie The Album by Dua Lipa', 'uri': '1vYXt7VSjH9JIM5oRRo7vA'}, {'full_track_name': "I'm Good (Blue) by David Guetta", 'uri': '4uUG5RXrOk84mYEfFvj3cK'}, {'full_track_name': 'Perfect by Ed Sheeran', 'uri': '0tgVpDi06FyKpA1z0VMD4v'}, {'full_track_name': "Doin' Time by Lana Del Rey", 'uri': '0Oqc0kKFsQ6MhFOLBNZIGX'}, {'full_track_name': 'Sweater Weather by The Neighbourhood', 'uri': '2QjOHCTQ1Jl3zawyYOpxh6'}]},
+                 {"title": "What I've Done by Lincoln Park","uri":"https://open.spotify.com/track/2nLtzopw4rPReszdYBJU6h?si=1b48d972e12d43e7", "next_five":[{'full_track_name': 'Higher by Creed', 'uri': '1ZozJfi8u9cO2Ob8KwiwNT'}, {'full_track_name': 'Drive by Incubus', 'uri': '7nnWIPM5hwE3DaUBkvOIpy'}, {'full_track_name': 'I Hate Everything About You by Three Days Grace', 'uri': '0M955bMOoilikPXwKLYpoi'}, {'full_track_name': 'The Reason by Hoobastank', 'uri': '77loZpT5Y5PRP1S451P9Yz'}, {'full_track_name': 'Butterfly by Crazy Town', 'uri': '4BggEwLhGfrbrl7JBhC8EC'}]},
+                 {"title": "Hey Brother by Avicii","uri":"https://open.spotify.com/track/4lhqb6JvbHId48OUJGwymk?si=d148732daa8e4897", "next_five":[{'full_track_name': 'Silence by Marshmello', 'uri': '7vGuf3Y35N4wmASOKLUVVU'}, {'full_track_name': 'Sugar (feat. Francesco Yates) by Robin Schulz', 'uri': '5tf1VVWniHgryyumXyJM7w'}, {'full_track_name': 'Where Did You Go? (feat. MNEK) by Jax Jones', 'uri': '3sa06xVNmLLYIxdNNmVQN8'}, {'full_track_name': 'Rockabye (feat. Sean Paul & Anne-Marie) by Clean Bandit', 'uri': '5knuzwU65gJK7IF5yJsuaW'}, {'full_track_name': 'Lush Life by Zara Larsson', 'uri': '1rIKgCH4H52lrvDcz50hS8'}] },
+                 {"title": "Sound of Silence by Simon & Garfunkel","uri":"https://open.spotify.com/track/3YfS47QufnLDFA71FUsgCM?si=5bff00bb3b2e4a67", "next_five": [{'full_track_name': 'American Pie by Don McLean', 'uri': '1fDsrQ23eTAVFElUMaf38X'}, {'full_track_name': 'The Weight - Remastered 2000 by The Band', 'uri': '0P7DoyGrr4Wp9w5TotEtUC'}, {'full_track_name': "You're So Vain by Carly Simon", 'uri': '2DnJjbjNTV9Nd5NOa1KGba'}, {'full_track_name': "Do You Believe in Magic? by The Lovin' Spoonful", 'uri': '1udKn1oNKYQSQ9OmiIWCMu'}, {'full_track_name': "You've Got a Friend by Carole King", 'uri': '1qhp5owrFGeZ5Jgin28nwR'}]},
+                 {"title": "Bohemian Rhapsody by Queen","uri":"https://open.spotify.com/track/6l8GvAyoUZwWDgF1e4822w?si=9639498e177d46e6", "next_five":[{'full_track_name': 'You Really Got Me by The Kinks', 'uri': '29SyMC0plk6qw8NMF7lfRL'}, {'full_track_name': 'Walk On the Wild Side by Lou Reed', 'uri': '5p3JunprHCxClJjOmcLV8G'}, {'full_track_name': 'In The Army Now by Status Quo', 'uri': '6QLspHLCLftSGPa6vNzKuo'}, {'full_track_name': 'Surrender by Cheap Trick', 'uri': '2ccUQnjjNWT0rsNnsBpsCA'}, {'full_track_name': 'More Than This by Roxy Music', 'uri': '6N7gPTru90HYLRUIVDQ185'}] },
+                 {"title": "Nocturnes Op. 9 N. 2 by Fr\u00e9d\u00e9ric Chopin","uri":"https://open.spotify.com/track/61YM5SkqqeUjIBL7It56cs?si=01a05e921c61409b", "next_five":[{'full_track_name': 'Paganini: 24 Caprices, Op. 1: No. 24 in A Minor by Niccolò Paganini', 'uri': '6cC6h5E351deEGFd7Kh9uy'}, {'full_track_name': 'I puritani / Act I: A te, o cara by Vincenzo Bellini', 'uri': '48fH1Mv10Nl8Hisna5y2Rf'}, {'full_track_name': 'Strauss I: Radetzky March, Op. 228 by Johann Strauss I', 'uri': '53Z3s7DPc7WbkYt9L9ZdlJ'}, {'full_track_name': 'Clarinet Concerto No. 1 in F Minor, Op. 73: III. Rondo: Allegretto by Carl Maria von Weber', 'uri': '21Ct04i87NtSa6pKvM0bAF'}, {'full_track_name': 'Scherzo-tarantelle in G Minor, Op.16 - 1987 Remastered Version by Henryk Wieniawski', 'uri': '0yQMA3UgzGgmL7Py3lPFCx'}]}
                  ]
 
 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:
+        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))
@@ -147,7 +148,7 @@ def song_page():
     selected_song = next((song for song in song_list if song['title'] == selected_title), None)
     if selected_song:
         st.session_state.selected_song_index = selected_title
-        # print(st.session_state.selected_song_index)
+        print(st.session_state.selected_song_index)
         song_uri = selected_song['uri']
         st.session_state.s_url = song_uri
     
@@ -165,8 +166,8 @@ def artist_page():
 def spr_sidebar():
     menu=option_menu(
         menu_title=None,
-        options=['Home','Recommended','Explanation','Counter'],
-        icons=['house','lightbulb','heart','heart-half'],
+        options=['Home','Recommended','Counter'],
+        icons=['house','lightbulb','heart-half'],
         menu_icon='cast',
         default_index=0,
         orientation='horizontal'
@@ -177,8 +178,8 @@ def spr_sidebar():
         st.session_state.app_mode = 'Recommended'
     elif menu=='About':
         st.session_state.app_mode = 'About'
-    elif menu=='Explanation':
-        st.session_state.app_mode = 'Explanation'
+    # elif menu=='Explanation':
+    #     st.session_state.app_mode = 'Explanation'
     elif menu=='Log':
         st.session_state.app_mode = 'Log'
     elif menu=='Counter':
@@ -190,7 +191,7 @@ def home_page():
     st.session_state.same_art=st.session_state.artist
     st.session_state.Region=st.session_state.rg
 
-    st.title('Spotify Recommendation System')
+    st.title('CS6966 - Class Project (Spotify Recommendation)')
     col2,col3=st.columns([2,2])
     # radio=col.radio("Feature",options=("Playlist","Song"),key='radio',on_change=update_radio0)
     radio2=col2.radio("Group Assignment",options=("Succinct","Verbose"),key='radio2',on_change=update_radio2)
@@ -227,12 +228,13 @@ def result_page():
     if 'rs' not in st.session_state:
         st.error('Please select a model on the Home page and run "Generate Recommendations"')
     else:
-        st.success('Top {} recommendations'.format(len(st.session_state.rs)))
-        i=0
+        if not st.session_state.button_states.values():
+            st.success('Top {} recommendations'.format(len(st.session_state.rs)))
+        i = 0
         tracks = []
         original_tracks = []
         length_of_explanation = 1
-        for uri in st.session_state.rs:
+        for index, uri in enumerate(st.session_state.rs):
             uri_link = "https://open.spotify.com/embed/track/" + uri + "?utm_source=generator&theme=0"
             components.iframe(uri_link, height=80)
          
@@ -241,55 +243,47 @@ def result_page():
                 st.session_state.button_states[uri] = False
                 
             track_name, artist_name = get_track_info(uri)
-            tracks.append({'track_name': track_name, 'artist_name': artist_name, 'uri': uri})
-            st.session_state.button_states[uri] = st.button('Get explanation for {}'.format(track_name))
-            st.spinner('Getting explanation...')
+            full_song_title = '{} by {}'.format(track_name, artist_name)
+            original_tracks.append({'full_track_name': full_song_title, 'uri': uri})
+            tracks.append({'track_name': track_name, 'artist_name': artist_name, 'uri': uri, 'index': index})
+            explanation_button = st.button('Get explanation for {}'.format(track_name))          
 
             # When button is clicked, call get_track_info
-            if st.session_state.button_states[uri]:
-                with st.spinner('Getting explanation...'):                    
+            if explanation_button:
+                with st.spinner('Getting explanation...'):
+                    time.sleep(0.5)                      
                     if st.session_state.radio_choice == 'Song':
-                        track, artist = get_track_info(uri)
-                        print(track, artist)
+                        song_title, artist = get_track_info(uri)
+                        print(song_title, artist)
+                        full_song_title = '{} by {}'.format(song_title, artist)
+                        found_index = None
+                        key = None
+                        for track in tracks:
+                            if track['track_name'] == song_title:
+                                found_index = track['index'] + 1
+                                key = str(found_index)
+                                break  # Exit the loop once the track is found
+                        with open('data/results.json') as f:
+                            data = json.load(f)
                         
                         songlist = [{'track_name': track, 'artist_name': artist}]
                         if st.session_state.explanation_length == "Verbose":
                             length_of_explanation = 5
+                            song_objects = data["5"]
+                            st.session_state.generated_explanation = song_objects[st.session_state.selected_song_index][0][key]
                         else: 
                             length_of_explanation = 1
-                            
+                            song_objects = data["1"]
+                            st.session_state.generated_explanation = song_objects[st.session_state.selected_song_index][0][key]
+                              
+                        st.success(st.session_state.generated_explanation)
                         st.session_state.explanation_status = True
-                        # mistral.explain_why(original_songs=songlist, top_songs=st.session_state.track_list, explanation_limit=length_of_explanation)
-                        # explanation = mistral_2_display(mistral.get_positive_explanations())
-                        # print(explanation)
-                        
-                    # track_name, artist_name = get_track_info(uri)
-                    # st.write('Track Name:', track_name)
-                    # st.write('Artist Name:', artist_name)
+                        st.session_state.button_states[uri] = True
         
-        i+=1
-        # st.session_state.track_list = tracks
-        # formatted_string = ''
-        # for item in st.session_state.track_list:
-        #     formatted_string += '{} by {}, '.format(item['track_name'], item['artist_name'])
-        # songs = formatted_string.rstrip(', ')
-
-        # # Split the string into a list of songs
-        # song_list = songs.split(", ")
-
-        # # Calculate the midpoint index
-        # midpoint = len(song_list) // 2
-
-        # # Get the top and bottom halves of the list
-        # top_half = song_list[:midpoint]
-        # bottom_half = song_list[midpoint:]
-
-        # # Print the top and bottom halves
-        # print("Top half:")
-        # print(top_half)
-        # print("Bottom half:")
-        # print(bottom_half)
-        if i%5==0:
+        i += 1
+        st.session_state.track_list = tracks
+        print(original_tracks[5:])
+        if i % 5 == 0:
             time.sleep(1)
             
 def Explain_page():
@@ -307,27 +301,60 @@ def Counter_page():
         st.header('Understanding Recommendations')
         st.markdown("<br>", unsafe_allow_html=True)
         st.subheader('Please select a song from the list below that you believe should have been recommended in the top 5 songs for your original selection.')
-
+    
         next_five = None
         for song in song_list:
             if song["title"] == st.session_state.selected_song_index:
                 next_five = song["next_five"]
                 break
-
-        selected_title = st.selectbox('Select a song', next_five)
+            
+        titles = [song['full_track_name'] for song in next_five]  # Get a list of song titles
+        
+        selected_title = st.selectbox('Select a song', titles)
+        
+        selected_song = next((song for song in next_five if song['full_track_name'] == selected_title), None)
+        if selected_song:
+            # st.session_state.selected_song_index = selected_title
+            print(selected_song)
+            song_uri = selected_song['uri']
+            # st.session_state.s_url = song_uri
+        
+            uri_link = 'https://open.spotify.com/embed/track/' + song_uri
+            # slot1 = st.empty()
+            components.iframe(uri_link, height=100)
         
         counter_fact_button = st.button('Generate Explanation')
         
         if counter_fact_button:
-            None
-            # Generate the counter factual example
+            found_index = None
+            key = None
+            for index, track in enumerate(next_five):
+                if track['full_track_name'] == selected_song['full_track_name']:
+                    found_index = index + 1
+                    key = str(found_index)
+                    break  # Exit the loop once the track is found
+            with open('data/results.json') as f:
+                data = json.load(f)
+                
+            if st.session_state.explanation_length == "Verbose":
+                length_of_explanation = 5
+                song_objects = data["5"]
+                st.session_state.generated_counter = song_objects[st.session_state.selected_song_index][1][key]
+            else: 
+                length_of_explanation = 1
+                song_objects = data["1"]
+                st.session_state.generated_counter = song_objects[st.session_state.selected_song_index][1][key]
+                
+            st.success(st.session_state.generated_counter)
+        
+        
 
 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:
+    with open('../data/streamlit.csv') as f:
         st.download_button('Download Dataset', f,file_name='streamlit.csv')
 def About_page():
     st.header('Development')
@@ -379,45 +406,6 @@ def About_page():
     """
     
 
-# def dialog_close_button_clicked():
-#     st.write("###  Dialog close callback results:")
-#     st.write("#### No preferences saved!")
-#     dialog.close()
-
-
-# dialog = st.dialog(
-#     "first_dialog_with_help", title="Group Assignment",
-#     on_close_button_clicked=dialog_close_button_clicked, can_be_closed=True)
-
-
-# def dialog_form_submit_button_clicked():
-#     st.write("### Dialog form submit button clicked results:")
-#     st.write(f"#### Hey {st.session_state.first_name} {st.session_state.last_name}!")
-#     st.write(f"#### Your preferences are saved")
-#     dialog.close()
-
-
-# with dialog:
-#     st.radio("Explanation Length",options=("Succinct","Verbose"),key='radio2',on_change=update_radio2)
-#     st.text_input("First name", key="first_name")
-#     st.text_input("Last name", key="last_name")
-#     st.form_submit_button("OK", on_click=dialog_form_submit_button_clicked)
-
-
-# if st.button("Open dialog", key="first_dialog_button"):
-#     dialog.open()
-
-# modal = Modal("Demo Modal", key="example")
-# open_modal = st.button("Open")
-# if open_modal:
-#     modal.open()
-
-# if modal.is_open():
-#     with modal.container():
-#         st.write("Text goes here")
-#         st.radio("Explanation Length",options=("Succinct","Verbose"),key='radio3',on_change=update_radio2)
-#         st.write("Some fancy text")
-
 def main():
     spr_sidebar()        
     if st.session_state.app_mode == 'Home':

model.py

@@ -157,7 +157,7 @@ def playlist_model(url, model, max_gen=3, same_art=5):
          log.append('Model run successfully')
          return Fresult, log
 
-     lendf=len(pd.read_csv('data/streamlit.csv',usecols=['track_uri']))
+     lendf=len(pd.read_csv('../data/streamlit.csv',usecols=['track_uri']))
      dtypes = {'track_uri': 'object', 'artist_uri': 'object', 'album_uri': 'object', 'danceability': 'float16', 'energy': 'float16', 'key': 'float16',
                'loudness': 'float16', 'mode': 'float16', 'speechiness': 'float16', 'acousticness': 'float16', 'instrumentalness': 'float16',
                'liveness': 'float16', 'valence': 'float16', 'tempo': 'float16', 'duration_ms': 'float32', 'time_signature': 'float16',
@@ -292,7 +292,7 @@ def playlist_model(url, model, max_gen=3, same_art=5):
      x = 1
      for i in range(int(lendf/2), lendf+1, int(lendf/2)):
          try:
-             df = pd.read_csv('data/streamlit.csv',names= col_name,dtype=dtypes,skiprows=x,nrows=i)
+             df = pd.read_csv('../data/streamlit.csv',names= col_name,dtype=dtypes,skiprows=x,nrows=i)
              log.append('reading data frame chunks from {} to {}'.format(x,i))
          except Exception as e:
              log.append('Failed to load grow')
@@ -315,7 +315,7 @@ def playlist_model(url, model, max_gen=3, same_art=5):
              log.append('genre|unknown not found')
          log.append('Scaling the data .....')
          if x == 1:
-             sc = pickle.load(open('data/sc.sav','rb'))
+             sc = pickle.load(open('../data/sc.sav','rb'))
              df.iloc[:, 3:19] = sc.transform(df.iloc[:, 3:19])
              test.iloc[:, 3:19] = sc.transform(test.iloc[:, 3:19])
              log.append("Creating playlist vector")
@@ -359,13 +359,13 @@ def playlist_model(url, model, max_gen=3, same_art=5):
      log.append('{} New Tracks Found'.format(len(grow)))
      if(len(grow)>=1):
         try:
-            new=pd.read_csv('data/new_tracks.csv',dtype=dtypes)
+            new=pd.read_csv('../data/new_tracks.csv',dtype=dtypes)
             new=pd.concat([new, grow], axis=0)
             new=new[new.Track_pop >0]
             new.drop_duplicates(subset=['track_uri'], inplace=True,keep='last')
-            new.to_csv('data/new_tracks.csv',index=False)
+            new.to_csv('../data/new_tracks.csv',index=False)
         except:
-            grow.to_csv('data/new_tracks.csv', index=False)
+            grow.to_csv('../data/new_tracks.csv', index=False)
      log.append('Model run successfully')
     except Exception as e:
         log.append("Model Failed")
@@ -437,7 +437,7 @@ def song_model(url, model, max_gen=3, same_art=5):
             Fresult.append(aa['tracks'][i]['id'])
         log.append('Model run successfully')
         return Fresult, log
-     lendf=len(pd.read_csv('data/streamlit.csv',usecols=['track_uri']))
+     lendf=len(pd.read_csv('../data/streamlit.csv',usecols=['track_uri']))
      dtypes = {'track_uri': 'object', 'artist_uri': 'object', 'album_uri': 'object', 'danceability': 'float16', 'energy': 'float16', 'key': 'float16',
                'loudness': 'float16', 'mode': 'float16', 'speechiness': 'float16', 'acousticness': 'float16', 'instrumentalness': 'float16',
                'liveness': 'float16', 'valence': 'float16', 'tempo': 'float16', 'duration_ms': 'float32', 'time_signature': 'float16',
@@ -504,7 +504,7 @@ def song_model(url, model, max_gen=3, same_art=5):
      x = 1
      for i in range(int(lendf/2), lendf+1, int(lendf/2)):
          try:
-             df = pd.read_csv('data/streamlit.csv',names= col_name,dtype=dtypes,skiprows=x,nrows=i)
+             df = pd.read_csv('../data/streamlit.csv',names= col_name,dtype=dtypes,skiprows=x,nrows=i)
              log.append('reading data frame chunks from {} to {}'.format(x,i))
          except Exception as e:
              log.append('Failed to load grow')
@@ -527,7 +527,7 @@ def song_model(url, model, max_gen=3, same_art=5):
              log.append('genre|unknown not found')
          log.append('Scaling the data .....')
          if x == 1:
-             sc = pickle.load(open('data/sc.sav','rb'))
+             sc = pickle.load(open('../data/sc.sav','rb'))
              df.iloc[:, 3:19] = sc.transform(df.iloc[:, 3:19])
              test.iloc[:, 3:19] = sc.transform(test.iloc[:, 3:19])
              log.append("Creating playlist vector")
@@ -571,13 +571,13 @@ def song_model(url, model, max_gen=3, same_art=5):
      log.append('{} New Tracks Found'.format(len(grow)))
      if(len(grow)>=1):
         try:
-            new=pd.read_csv('data/new_tracks.csv',dtype=dtypes)
+            new=pd.read_csv('../data/new_tracks.csv',dtype=dtypes)
             new=pd.concat([new, grow], axis=0)
             new=new[new.Track_pop >0]
             new.drop_duplicates(subset=['track_uri'], inplace=True,keep='last')
-            new.to_csv('data/new_tracks.csv',index=False)
+            new.to_csv('../data/new_tracks.csv',index=False)
         except:
-            grow.to_csv('data/new_tracks.csv', index=False)
+            grow.to_csv('../data/new_tracks.csv', index=False)
      log.append('Model run successfully')
     except Exception as e:
         log.append("Model Failed")
@@ -593,14 +593,14 @@ def update_dataset():
                'loudness': 'float16', 'mode': 'float16', 'speechiness': 'float16', 'acousticness': 'float16', 'instrumentalness': 'float16',
                'liveness': 'float16', 'valence': 'float16', 'tempo': 'float16', 'duration_ms': 'float32', 'time_signature': 'float16',
                'Track_release_date': 'int8', 'Track_pop': 'int8', 'Artist_pop': 'int8', 'Artist_genres': 'object'}
-    df = pd.read_csv('data/streamlit.csv',dtype=dtypes)
-    grow = pd.read_csv('data/new_tracks.csv',dtype=dtypes)
+    df = pd.read_csv('../data/streamlit.csv',dtype=dtypes)
+    grow = pd.read_csv('../data/new_tracks.csv',dtype=dtypes)
     cur = len(df)
     df=pd.concat([df,grow],axis=0)
     grow=pd.DataFrame(columns=col_name)
-    grow.to_csv('data/new_tracks.csv',index=False)
+    grow.to_csv('../data/new_tracks.csv',index=False)
     df=df[df.Track_pop >0]
     df.drop_duplicates(subset=['track_uri'],inplace=True,keep='last')
     df.dropna(axis=0,inplace=True)
-    df.to_csv('data/streamlit.csv',index=False)
+    df.to_csv('../data/streamlit.csv',index=False)
     return (len(df)-cur)