feat(core): implement chat route with function-calling support

app.py

@@ -1,7 +1,8 @@
 import os
 import google.generativeai as genai
-from flask import Flask, render_template
+from flask import Flask, render_template, request
 from user_data import get_user_history
+from logs import save_log
 from book_recommendations import (
     recommend_fiction,
     recommend_non_fiction,
@@ -29,21 +30,48 @@ magical_if = genai.GenerativeModel(
 
 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 chame diretamente a função
+      correspondente à categoria de livros recomendada.
+
     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.
+    4. Se o usuário demonstrou interesse em fantasia, recomende fantasia.
+    5. Caso contrário, ofereça uma recomendação geral de ficção.
+
+    ATENÇÃO: Sempre retorne a resposta da função chamada.
+    Não explique, apenas chame.
     """
     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}; "
-        f"Regras de negócio: {business_rules}"
+
+    message = (
+        f"Histórico do usuário {user_id}: {history}\n"
+        f"Regras de negócio:\n{business_rules}"
     )
+
+    response = user_decision.send_message(message)
+
+    # Tenta obter a chamada de função feita pela IA
+    try:
+        function_call = response.candidates[0].content.parts[0].function_call
+        function_name = function_call.name
+
+        user_function = {
+            "recommend_fiction": recommend_fiction,
+            "recommend_non_fiction": recommend_non_fiction,
+            "recommend_science": recommend_science,
+            "recommend_fantasy": recommend_fantasy
+        }.get(function_name)
+
+        if user_function:
+            return user_function(user_id)
+    except (AttributeError, IndexError):
+        pass
+
+    # Caso nenhuma função tenha sido chamada, retorna o texto
     return response.text
 
 
@@ -53,6 +81,7 @@ def recommend(user_id):
     if not history:
         return "Usuário não encontrado", 404
     ia_response = ia_decision(user_id, history)
+    save_log(user_id, history, ia_response)
     return render_template(
         'recommendation.html',
         user_id=user_id,
@@ -60,5 +89,49 @@ def recommend(user_id):
     )
 
 
+@app.route('/chat/<int:user_id>', methods=['GET', 'POST'])
+def chat(user_id):
+    history = get_user_history(user_id)
+    if not history:
+        return "Usuário não encontrado", 404
+
+    response = None
+
+    if request.method == 'POST':
+        question = request.form.get('question')
+        user_decision = magical_if.start_chat(
+            enable_automatic_function_calling=True
+        )
+        gemini_response = user_decision.send_message(
+            f"Usuário {user_id}, histórico: {history}. Pergunta: {question}"
+        )
+
+        try:
+            function_call = (
+                gemini_response
+                .candidates[0]
+                .content.parts[0]
+                .function_call
+            )
+            function_name = function_call.name
+            user_function = {
+                "recommend_fiction": recommend_fiction,
+                "recommend_non_fiction": recommend_non_fiction,
+                "recommend_science": recommend_science,
+                "recommend_fantasy": recommend_fantasy
+            }.get(function_name)
+
+            if user_function:
+                response = user_function(user_id)
+            else:
+                response = gemini_response.text
+        except (AttributeError, IndexError):
+            response = gemini_response.text
+
+        save_log(user_id, history, response)
+
+    return render_template('chat.html', user_id=user_id, response=response)
+
+
 if __name__ == '__main__':
     app.run(debug=True)