Update main.py

main.py

@@ -52,16 +52,27 @@ def play_recomm():
         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 top {} recommendations'.format(len(st.session_state.rs)))
-else:
-st.error('Model failed. Check the log for more information.')
+    if 'rs' in st.session_state:
+        st.write("Session state is present")
+        feature = st.selectbox("Select a feature:", ["Playlist", "Artist", "Song"], index=0)
+        st.write("You have selected: ", feature)
+        if feature == "Playlist":
+            st.write("Note: For Playlist-based recommendations, please use public playlists only.")
+            #code to get recommendations based on the playlist name
+        elif feature == "Artist":
+            artist_name = st.text_input("Enter artist name")
+            if artist_name:
+                get_artist_recommendations(artist_name)
+            else:
+                st.write("Please enter a valid artist name")
+        elif feature == "Song":
+            st.write("Note: For Song-based recommendations, Spotify model will be used by default.")
+            # code to get recommendations based on the song name
+        else:
+            st.write("Please select a valid feature")
+    else:
+        st.write("Session state is not present")
+
 
 def song_recomm():
 if 'rs' in st.session_state: