Update app.py

app.py

@@ -19,11 +19,12 @@ algo = SVD()
 algo.fit(trainset)
 
 # Initialize session state for storing user ratings
-if 'user_ratings' not in st.session_state:
-    st.session_state.user_ratings = {1: [], 2: [], 3: [], 4: []}
+if 'movie_ratings' not in st.session_state:
+    st.session_state.movie_ratings = {'Die Hard': [], 'Mad Max': [], 'The Shawshank Redemption': [], 'Forrest Gump': [],
+                                      'Superbad': [], 'Step Brothers': [], 'Inception': [], 'Interstellar': []}
 
-def update_ratings(item_id, rating):
-    st.session_state.user_ratings[item_id].append(rating)
+def update_ratings(movie, rating):
+    st.session_state.movie_ratings[movie].append(rating)
 
 # UI Elements
 st.title("🌟 Personalized Entertainment Recommendations")
@@ -60,16 +61,31 @@ predicted_ratings = [(id, algo.predict(5, id).est) for id in [1, 2, 3, 4] if id
 predicted_ratings.sort(key=lambda x: x[1], reverse=True)
 
 st.subheader("🎯 Recommendations for you:")
+
+# Collect average scores
+all_movies = []
 for item_id, rating in predicted_ratings:
     for rec in recommendations[item_id]:
-        avg_score = np.mean(st.session_state.user_ratings[item_id]) if st.session_state.user_ratings[item_id] else rating
+        current_ratings = st.session_state.movie_ratings[rec]
+        avg_score = np.mean(current_ratings) if current_ratings else rating
+
         st.write(f"{rec} - Average Score: {avg_score:.2f} ⭐")
 
-        # Rating buttons from 1 to 10
+        # Rating slider from 1 to 10
         user_rating = st.slider(f"Rate {rec}", 1, 10, 5, key=f"slider_{rec}")
         if st.button(f"Submit Rating for {rec}", key=f"button_{rec}"):
-            update_ratings(item_id, user_rating)
-            st.write(f"Submitted Rating: {user_rating}")
+            update_ratings(rec, user_rating)
+            st.experimental_rerun()  # Rerun to immediately reflect the rating submission
+
+        all_movies.append((rec, avg_score))
+
+# Sort movies by average score in descending order
+sorted_movies = sorted(all_movies, key=lambda x: x[1], reverse=True)
+
+# Display scoreboard
+st.subheader("🏆 Scoreboard:")
+for movie, avg_score in sorted_movies:
+    st.write(f"{movie}: {avg_score:.2f} ⭐")
 
 # Display current session state (for debugging purposes)
-st.write("Session State (User Ratings):", st.session_state.user_ratings)
+# st.write("Session State (User Ratings):", st.session_state.movie_ratings)