Update app.py

app.py

@@ -6,6 +6,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 
 pt = pd.read_pickle('pt.pkl')
 user_similarity_scores = cosine_similarity(pt.T)
+books = pd.read_pickle("books.pkl")
 
 def recommend_books_for_user(user_id):
     
@@ -23,7 +24,11 @@ def recommend_books_for_user(user_id):
 
         recommended_books.extend(unrated_books.index)
 
-    return recommended_books
+    recommended_books_set = set(recommended_books)
+
+    ans = [(book_title, image_url) for book_title, image_url in zip(books["Book-Title"], books["Image-URL-M"]) if book_title in recommended_books_set]
+
+    return ans
 
 def recommend_books_gradio(user_id):
     """Recommends books for a user based on collaborative filtering"""