feat: add fantasy category to user history and AI decision logic

app.py

@@ -1,11 +1,12 @@
-import google.generativeai as genai
 import os
+import google.generativeai as genai
 from flask import Flask, render_template
 from user_data import get_user_history
 from book_recommendations import (
     recommend_fiction,
     recommend_non_fiction,
-    recommend_science
+    recommend_science,
+    recommend_fantasy
 )
 
 app = Flask(__name__)
@@ -20,32 +21,44 @@ magical_if = genai.GenerativeModel(
     tools=[
         recommend_fiction,
         recommend_non_fiction,
-        recommend_science
+        recommend_science,
+        recommend_fantasy
     ]
 )
 
+
 def ia_decision(user_id, history):
     business_rules = """
-    Analise o histórico de compras do usuário e decida qual categoria de livros recomendar.
+    Analise o histórico de compras do usuário e decida qual categoria de 
+    livros recomendar.
     Regras:
     1. Se o usuário comprou mais livros de ficção, recomende ficção.
     2. Se o usuário comprou mais livros de não ficção, recomende não ficção.
     3. Se o usuário demonstrou interesse em ciência, recomende ciência.
     4. Caso contrário, ofereça uma recomendação geral de ficção.
     """
-    user_decision = magical_if.start_chat(enable_automatic_function_calling=True)
+    user_decision = magical_if.start_chat(
+        enable_automatic_function_calling=True
+    )
     response = user_decision.send_message(
-        f"Histórico do usuário {user_id}: {history}; Regras de negócio: {business_rules}"
+        f"Histórico do usuário {user_id}: {history}; "
+        f"Regras de negócio: {business_rules}"
     )
     return response.text
 
+
 @app.route('/recommend/<int:user_id>')
 def recommend(user_id):
     history = get_user_history(user_id)
     if not history:
         return "Usuário não encontrado", 404
     ia_response = ia_decision(user_id, history)
-    return render_template('recommendation.html', user_id=user_id, message=ia_response)
+    return render_template(
+        'recommendation.html',
+        user_id=user_id,
+        message=ia_response
+    )
+
 
 if __name__ == '__main__':
     app.run(debug=True)

user_data.py

@@ -3,8 +3,9 @@ def get_user_history(user_id: int) -> dict:
     Retorna o histórico de compras do usuário.
     """
     user_histories = {
-        1: {'fiction': 5, 'non_fiction': 2, 'science': 0},
-        2: {'fiction': 1, 'non_fiction': 4, 'science': 1},
-        3: {'fiction': 0, 'non_fiction': 1, 'science': 5},
+        1: {'fiction': 5, 'non_fiction': 2, 'science': 0, 'fantasy': 0},
+        2: {'fiction': 1, 'non_fiction': 4, 'science': 1, 'fantasy': 2},
+        3: {'fiction': 0, 'non_fiction': 1, 'science': 5, 'fantasy': 0},
+        4: {'fiction': 1, 'non_fiction': 0, 'science': 0, 'fantasy': 6},
     }
     return user_histories.get(user_id)