danschnurp commited on
Commit
9ad86e8
·
verified ·
1 Parent(s): 5b09067

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -143,17 +143,22 @@ def recommend_books(target_book: str, num_recommendations: int = 10) -> str:
143
 
144
 
145
  correlations = compute_correlations_faiss(faiss_index, book_titles, closest_match)
146
- correlations = correlations.drop_duplicates(subset=['ISBN'])
147
- recommendations = correlations[correlations['book'] != target_book].head(num_recommendations)
 
 
148
 
149
  result = f"Top {num_recommendations} recommendations for '{target_book}':\n\n"
150
  result += "\n\n".join([
151
  f"{idx}. Title: {dataset.loc[dataset['Book-Title'] == row['book'], 'Book-Title'].values[0]}, "
 
 
 
152
  f"ISBN: {dataset.loc[dataset['Book-Title'] == row['book'], 'ISBN'].values[0]}, "
153
  f"Rating: {ratings_by_isbn.loc[ratings_by_isbn['ISBN'] == dataset.loc[dataset['Book-Title'] == row['book'], 'ISBN'].values[0], 'Book-Rating'].values[0]}"
154
  for idx, (_, row) in enumerate(recommendations.iterrows(), 1)
155
  ])
156
-
157
  return result
158
 
159
 
 
143
 
144
 
145
  correlations = compute_correlations_faiss(faiss_index, book_titles, closest_match)
146
+
147
+ recommendations = correlations[correlations['book'] != target_book].drop_duplicates(subset=['ISBN'])
148
+
149
+ recommendations = recommendations.head(num_recommendations)
150
 
151
  result = f"Top {num_recommendations} recommendations for '{target_book}':\n\n"
152
  result += "\n\n".join([
153
  f"{idx}. Title: {dataset.loc[dataset['Book-Title'] == row['book'], 'Book-Title'].values[0]}, "
154
+ f"Author: {dataset.loc[dataset['Book-Title'] == row['book'], 'Book-Author'].values[0]}, "
155
+ f"Year: {dataset.loc[dataset['Book-Title'] == row['book'], 'Year-Of-Publication'].values[0]}, "
156
+ f"Publisher: {dataset.loc[dataset['Book-Title'] == row['book'], 'Publisher'].values[0]}, "
157
  f"ISBN: {dataset.loc[dataset['Book-Title'] == row['book'], 'ISBN'].values[0]}, "
158
  f"Rating: {ratings_by_isbn.loc[ratings_by_isbn['ISBN'] == dataset.loc[dataset['Book-Title'] == row['book'], 'ISBN'].values[0], 'Book-Rating'].values[0]}"
159
  for idx, (_, row) in enumerate(recommendations.iterrows(), 1)
160
  ])
161
+ # "ISBN";"Book-Title";"Book-Author";"Year-Of-Publication";"Publisher";"Image-URL-S"
162
  return result
163
 
164