updated svd app

app.py

@@ -123,13 +123,13 @@ elif app_selector == "Collaborative Recommender":
         # Sidebar for choosing the collaborative filtering method
         collaborative_method = st.sidebar.selectbox(
             "Choose a collaborative filtering method:", 
-            ["Surprise Collaborative Filtering", "User-Based Collaborative Filtering", "Anime-Based KNN Collaborative Filtering"]
+            ["SVD Collaborative Filtering", "User-Based Collaborative Filtering", "Anime-Based KNN Collaborative Filtering"]
         )
 
         # User input
-        if collaborative_method == "Surprise Collaborative Filtering" or collaborative_method == "User-Based Collaborative Filtering": 
-            usernames = anime_user_ratings['username'].unique()  
-            username = st.selectbox("Choose a user, and we'll show you animes they'd recommend!", usernames) 
+        if collaborative_method == "SVD Collaborative Filtering" or collaborative_method == "User-Based Collaborative Filtering": 
+            user_ids = anime_user_ratings['user_id'].unique()  
+            user_id = st.selectbox("Choose a user, and we'll show you animes they'd recommend", user_ids) 
             n_recommendations = st.slider("Number of Recommendations:", min_value=1, max_value=50, value=10)
         elif collaborative_method == "Anime-Based KNN Collaborative Filtering": 
             anime_list = anime_user_ratings["name"].dropna().unique().tolist() 
@@ -140,10 +140,10 @@ elif app_selector == "Collaborative Recommender":
         if st.button("Get Recommendations"):
             # Load the recommender
             recommender = CollaborativeAnimeRecommender(anime_user_ratings) 
-            if collaborative_method == "Surprise Collaborative Filtering": 
-                recommendations = recommender.get_svd_recommendations(username, n=n_recommendations, svd_model=svd_model)  
+            if collaborative_method == "SVD Collaborative Filtering": 
+                recommendations = recommender.get_svd_recommendations(user_id, n=n_recommendations, svd_model=svd_model)  
             elif collaborative_method == "User-Based Collaborative Filtering": 
-                recommendations = recommender.get_user_based_recommendations(username, n_recommendations=n_recommendations, knn_user_model=user_based_knn_model)
+                recommendations = recommender.get_user_based_recommendations(user_id, n_recommendations=n_recommendations, knn_user_model=user_based_knn_model)
             elif collaborative_method == "Anime-Based KNN Collaborative Filtering":
                 if anime_name: 
                     recommendations = recommender.get_item_based_recommendations(anime_name, n_recommendations=n_recommendations, knn_item_model=item_based_knn_model)
@@ -169,6 +169,7 @@ elif app_selector == "Collaborative Recommender":
     except Exception as e:
         st.error(f"An error occurred: {e}")
 
+
 elif app_selector == "Top Anime Recommender":
     st.title("Top Anime Recommender System")