Added movie genre tags in email notifier and refactoring code

Code/recommenderapp/app.py

@@ -1,7 +1,7 @@
 """
 Module for routing all calls from the frontend
 """
-
+from Code.prediction_scripts.item_based import recommend_for_new_user
 import json
 import sys
 from search import Search
@@ -10,7 +10,7 @@
 from utils import send_email_to_user, beautify_feedback_data
 
 sys.path.append("../../")
-from Code.prediction_scripts.item_based import recommend_for_new_user
+
 
 app = Flask(__name__)
 app.secret_key = "secret key"

Code/recommenderapp/utils.py

@@ -37,9 +37,9 @@ def create_colored_tags(genres):
     }
     tags = []
     for genre in genres:
-      color = genre_colors.get(genre, '#CCCCCC')  # Default color if not found
-      tag = f'<span style="background-color: {color}; color: #FFFFFF; padding: 5px; border-radius: 5px;">{genre}</span>'
-      tags.append(tag)
+        color = genre_colors.get(genre, '#CCCCCC')  # Default color if not found
+        tag = f'<span style="background-color: {color}; color: #FFFFFF; padding: 5px; border-radius: 5px;">{genre}</span>'
+        tags.append(tag)
 
     return ' '.join(tags)
 
@@ -87,23 +87,25 @@ def send_email_to_user(recipient_email, categorized_data):
     message['From'] = sender_email
     message['To'] = recipient_email
     message['Subject'] = subject
-
     # Load the CSV file into a DataFrame
     movie_genre_df = pd.read_csv('../../data/movies.csv')
 
     # Create a dictionary to map movies to their genres
     movie_to_genres = {}
 
-    for index, row in movie_genre_df.iterrows():
+    for row in movie_genre_df.rows():
         movie = row['title']
         genres = row['genres'].split('|')
         movie_to_genres[movie] = genres
 
     # Create the email message with HTML content
     html_content = c.EMAIL_HTML_CONTENT.format(
-        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' for movie in categorized_data['Liked']),
-        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' for movie in categorized_data['Disliked']),
-        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' for movie in categorized_data['Yet to Watch']))
+        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' \
+            for movie in categorized_data['Liked']),
+        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' \
+            for movie in categorized_data['Disliked']),
+        '\n'.join(f'<li>{movie} {create_colored_tags(movie_to_genres.get(movie, ["Unknown Genre"]))}</li><br>' \
+            for movie in categorized_data['Yet to Watch']))
 
 
     # Attach the HTML email body

setup.py

@@ -1,3 +1,6 @@
+"""
+    Setup.py file for the code
+"""
 import setuptools
 
 with open("README.md", "r", encoding="utf8") as fh:

test/tests.py

@@ -2,12 +2,12 @@
 Test suite for recommender system
 """
 
+from Code.prediction_scripts.item_based import recommend_for_new_user
 import unittest
 import warnings
 import sys
 
 sys.path.append("../")
-from Code.prediction_scripts.item_based import recommend_for_new_user
 
 warnings.filterwarnings("ignore")