Update app/feedback.py

app/feedback.py

@@ -8,18 +8,19 @@ from .google_oauth import GoogleAuthRequiredError, append_feedback_to_sheet, loa
 from .memory import salvar_memoria_negativa
 
 
-BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 DEFAULT_LOGS_DIR = Path(os.getenv("LOGS_DIR", "/data/logs"))
 FALLBACK_LOGS_DIR = Path(tempfile.gettempdir()) / "tcc2_agent" / "logs"
 FEEDBACK_HEADERS = [
     "timestamp",
+    "search_id",
     "query",
+    "rank",
     "product_id",
     "product_name",
+    "categoria_inferida",
+    "categoria_produto",
     "rating",
     "is_helpful",
-    "note",
-    "categoria_inferida",
     "feedback",
     "motivo",
     "score_final",
@@ -27,6 +28,7 @@ FEEDBACK_HEADERS = [
     "bonus_lexical",
     "penalidade_feedback",
     "user_message",
+    "note",
 ]
 
 
@@ -77,14 +79,43 @@ def _append_feedback_csv(row):
         writer.writerow([row.get(header, "") for header in FEEDBACK_HEADERS])
 
 
+def _derive_feedback_fields(rating=None, is_helpful=None, feedback=None):
+    derived_feedback = feedback
+    derived_is_helpful = is_helpful
+
+    if derived_feedback is None:
+        if rating is not None and rating >= 4:
+            derived_feedback = "positivo"
+        elif rating is not None and rating <= 2:
+            derived_feedback = "negativo"
+        elif rating == 3:
+            derived_feedback = "neutro"
+        else:
+            derived_feedback = ""
+
+    if derived_is_helpful is None:
+        if rating is not None and rating >= 4:
+            derived_is_helpful = True
+        elif rating is not None and rating <= 2:
+            derived_is_helpful = False
+        elif rating == 3:
+            derived_is_helpful = ""
+        else:
+            derived_is_helpful = ""
+
+    return derived_feedback, derived_is_helpful
+
+
 def _build_feedback_payload(
+    search_id,
     query,
+    rank,
     product_id,
     product_name,
+    categoria_inferida=None,
+    categoria_produto=None,
     rating=None,
     is_helpful=None,
-    note=None,
-    categoria_inferida=None,
     feedback=None,
     motivo=None,
     score_final=None,
@@ -92,34 +123,46 @@ def _build_feedback_payload(
     bonus_lexical=None,
     penalidade_feedback=None,
     user_message=None,
+    note=None,
 ):
+    derived_feedback, derived_is_helpful = _derive_feedback_fields(
+        rating=rating,
+        is_helpful=is_helpful,
+        feedback=feedback,
+    )
+
     return {
         "timestamp": datetime.now(timezone.utc).isoformat(),
+        "search_id": search_id or "",
         "query": query,
+        "rank": rank if rank is not None else "",
         "product_id": product_id,
         "product_name": product_name,
-        "rating": rating if rating is not None else "",
-        "is_helpful": is_helpful if is_helpful is not None else "",
-        "note": note or "",
         "categoria_inferida": categoria_inferida or "",
-        "feedback": feedback or "",
+        "categoria_produto": categoria_produto or "",
+        "rating": rating if rating is not None else "",
+        "is_helpful": derived_is_helpful,
+        "feedback": derived_feedback,
         "motivo": motivo or "",
         "score_final": score_final if score_final is not None else "",
         "score_semantico": score_semantico if score_semantico is not None else "",
         "bonus_lexical": bonus_lexical if bonus_lexical is not None else "",
         "penalidade_feedback": penalidade_feedback if penalidade_feedback is not None else "",
-        "user_message": user_message or "",
+        "user_message": user_message or "",  # ajuste aqui
+        "note": note or "",
     }
 
 
 def salvar_feedback(
+    search_id,
     query,
+    rank,
     product_id,
     product_name,
+    categoria_inferida=None,
+    categoria_produto=None,
     rating=None,
     is_helpful=None,
-    note=None,
-    categoria_inferida=None,
     feedback=None,
     motivo=None,
     score_final=None,
@@ -127,15 +170,18 @@ def salvar_feedback(
     bonus_lexical=None,
     penalidade_feedback=None,
     user_message=None,
+    note=None,
 ):
     row = _build_feedback_payload(
+        search_id=search_id,
         query=query,
+        rank=rank,
         product_id=product_id,
         product_name=product_name,
+        categoria_inferida=categoria_inferida,
+        categoria_produto=categoria_produto,
         rating=rating,
         is_helpful=is_helpful,
-        note=note,
-        categoria_inferida=categoria_inferida,
         feedback=feedback,
         motivo=motivo,
         score_final=score_final,
@@ -143,6 +189,7 @@ def salvar_feedback(
         bonus_lexical=bonus_lexical,
         penalidade_feedback=penalidade_feedback,
         user_message=user_message,
+        note=note,
     )
 
     _append_feedback_csv(row)
@@ -160,22 +207,26 @@ def salvar_feedback(
             f"Erro de sincronizacao: {exc}"
         )
 
-    if rating is not None and rating <= 2:
-        salvar_memoria_negativa(
-            query=query,
-            product_id=product_id,
-            product_name=product_name,
-            rating=rating,
-            motivo="rating_baixo",
-        )
+    # 🔥 ajuste aqui (tratamento de string → int)
+    try:
+        rating_num = int(row.get("rating"))
+    except (TypeError, ValueError):
+        rating_num = None
 
-    if is_helpful is False:
+    is_helpful_value = row.get("is_helpful")
+
+    if (
+        (isinstance(rating_num, int) and rating_num <= 2)
+        or is_helpful_value is False
+    ):
         salvar_memoria_negativa(
             query=query,
             product_id=product_id,
             product_name=product_name,
-            rating=rating if rating is not None else "",
-            motivo="nao_foi_util",
+            rating=rating_num,
+            motivo=row.get("motivo") or row.get("feedback") or "feedback_negativo",
+            search_id=row.get("search_id"),
+            rank=row.get("rank"),
         )
 
     response = {
@@ -185,5 +236,4 @@ def salvar_feedback(
     }
     if warning:
         response["warning"] = warning
-    return response
-
+    return response