Update app.py

app.py

@@ -50,6 +50,7 @@ def app():
             'redirect_uri': REDIRECT_URI,
             'scope': 'user-read-private',  # Modify as per your required scopes
         }
+        st.code(st.session_state)
         st.write('Please log in to Spotify')
         st.markdown(f"[Log In]({AUTH_URL}?{urlencode(auth_params)})", unsafe_allow_html=True)
     else:#
@@ -71,18 +72,31 @@ def app():
         if 'access_token' in st.session_state:
             st.write("Enter a Spotify Track URL or URI to get its audio features.")
             track_input = st.text_input("Spotify Track URL/URI", "")
+            traindata = []
             if st.button("Get Audio Features"):
                 track_id = extract_track_id_from_url(track_input)
+                if track_id:
+                    track_info = get_track_information(track_id, st.session_state['access_token'])
+                    if track_info:
+                        st.write(f"Popularität: {track_info.get('popularity', 'Nicht verfügbar')}")
+                        traindata.append("new")
+                        traindata.append(track_info["name"])
+                        traindata.append(track_info["artists"][0]["name"])
+                        traindata.append("No_Genre_Found")
+                    else:
+                        st.write("Track-Informationen konnten nicht abgerufen werden.")
+                else:
+                    st.write("Bitte geben Sie eine gültige Track-ID ein.")
                 if track_id:
                     audio_features = get_track_audio_features(track_id, st.session_state['access_token'])
                     features = [None] * 9
-                    features[0] = audio_features["tempo"]   # Beats per minute
+                    features[0] = audio_features["tempo"]               # Beats per minute
                     features[1] = audio_features["energy"]
                     features[2] = audio_features["danceability"]
-                    features[3] = audio_features["loudness"]   # Loudness in dB
+                    features[3] = audio_features["loudness"]                     # Loudness in dB
                     features[4] = audio_features["liveness"]
                     features[5] = audio_features["valence"]
-                    features[6] = audio_features["duration_ms"] / 1000   # Umrechnung von Millisekunden in Sekunden
+                    features[6] = audio_features["duration_ms"] / 1000        # Umrechnung von Millisekunden in Sekunden
                     features[7] = audio_features["acousticness"]
                     features[8] = audio_features["speechiness"]
                     feature_display = {
@@ -96,21 +110,24 @@ def app():
                         "acousticness":features[7],
                         "speechiness":features[8]
                     }
+                    traindata.append(audio_features["tempo"])
+                    traindata.append(audio_features["energy"])
+                    traindata.append(audio_features["danceability"])
+                    traindata.append(audio_features["loudness"])
+                    traindata.append(audio_features["liveness"])
+                    traindata.append(audio_features["valence"])
+                    traindata.append(audio_features["duration_ms"]/ 1000)
+                    traindata.append(audio_features["acousticness"])
+                    traindata.append(audio_features["speechiness"])
+                    traindata.append(track_info["popularity"])
                     st.write(feature_display)
-                    predictions = predict_popularity(features)
+                    predictions = predict_popularity(features, traindata)
                     st.write(f"Random Forest Prediction: {predictions[0]}")
                     st.write(f"Regression Prediction: {predictions[1]}")
 
                 else:
                     st.error("Please enter a valid Spotify Track URL or URI.")
-                if track_id:
-                    track_info = get_track_information(track_id, st.session_state['access_token'])
-                    if track_info:
-                        st.write(f"Popularität: {track_info.get('popularity', 'Nicht verfügbar')}")
-                    else:
-                        st.write("Track-Informationen konnten nicht abgerufen werden.")
-                else:
-                    st.write("Bitte geben Sie eine gültige Track-ID ein.")
+
 
 if 'code' not in st.session_state:
     query_params = st.experimental_get_query_params()