Update app.py

app.py

@@ -2,12 +2,14 @@ import streamlit as st
 from surprise import Dataset, Reader, SVD
 from surprise.model_selection import train_test_split
 import pandas as pd
+import numpy as np
 
-# Sample user-item rating data for collaborative filtering
-data = {'user_id': [1, 1, 1, 2, 2, 3, 3, 4, 4],
-        'item_id': [1, 2, 3, 1, 4, 2, 3, 1, 4],
-        'rating': [5, 3, 4, 4, 5, 5, 2, 4, 4]}
-
+# Initialize sample user-item rating data
+data = {
+    'user_id': [1, 1, 1, 2, 2, 3, 3, 4, 4],
+    'item_id': [1, 2, 3, 1, 4, 2, 3, 1, 4],
+    'rating': [5, 3, 4, 4, 5, 5, 2, 4, 4]
+}
 df = pd.DataFrame(data)
 reader = Reader(rating_scale=(1, 5))
 dataset = Dataset.load_from_df(df[['user_id', 'item_id', 'rating']], reader)
@@ -16,6 +18,13 @@ trainset, testset = train_test_split(dataset, test_size=0.25)
 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: []}
+
+def update_ratings(item_id, rating):
+    st.session_state.user_ratings[item_id].append(rating)
+
 # UI Elements
 st.title("🌟 Personalized Entertainment Recommendations")
 
@@ -53,11 +62,14 @@ predicted_ratings.sort(key=lambda x: x[1], reverse=True)
 st.subheader("🎯 Recommendations for you:")
 for item_id, rating in predicted_ratings:
     for rec in recommendations[item_id]:
-        st.write(f"{rec} - Predicted Rating: {rating:.2f} ⭐")
-
-# Provide visual feedback for actions
-if predicted_ratings:
-    st.button("👍 Like", key="like_button")
-    st.button("👎 Dislike", key="dislike_button")
-else:
-    st.write("No recommendations available for your selected genre.")
+        avg_score = np.mean(st.session_state.user_ratings[item_id]) if st.session_state.user_ratings[item_id] else rating
+        st.write(f"{rec} - Average Score: {avg_score:.2f} ⭐")
+
+        # Rating buttons 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}")
+
+# Display current session state (for debugging purposes)
+st.write("Session State (User Ratings):", st.session_state.user_ratings)