diff --git a/app/modules/partners/ui.py b/app/modules/partners/ui.py index 0971be5..63fc973 100644 --- a/app/modules/partners/ui.py +++ b/app/modules/partners/ui.py @@ -160,8 +160,15 @@ def _task_bucket(task: ClientServiceTaskInstance) -> str: return "pending_review" if status in CLOSED_TASK_STATUSES: return "completed" - if status in {"pending", "rework", "rework_required"}: + if status in {"rework", "rework_required"}: return "rework_sent" + if status == "pending": + has_partner_rework = any( + (getattr(comment, "comment_type", "") or "").strip().lower() == "partner_review_note" + and not getattr(comment, "is_deleted", False) + for comment in getattr(task, "comments", []) + ) + return "rework_sent" if has_partner_rework else "approved" return "approved"