Assignments page endpoints

main.py

@@ -882,6 +882,47 @@ def _normalize_sme_intake_result(raw: Dict[str, Any]) -> Dict[str, Any]:
         ],
     }
 
+def build_assignments_page_context(company_code, user_id, program_id, role, user_context):
+    """
+    Server-side mirror of the assignments page.
+
+    Must return:
+    - current user's department
+    - eligible SMEs
+    - each SME's required interventions from diagnosticPlans
+    - confirmation status per department
+    - existing assignedInterventions
+    - available consultants for department
+    - recurring / once-off assignment metadata
+    """
+    # Fetch:
+    # users / operationsStaff / consultants
+    # departments
+    # diagnosticPlans
+    # applications
+    # participants
+    # interventions
+    # assignedInterventions
+
+    return {
+        "page": "assignments",
+        "companyCode": company_code,
+        "programId": program_id or "all",
+        "department": {},
+        "rules": {
+            "smmeMustHaveAcceptedApplication": True,
+            "diagnosticPlanMustBeConfirmedByDepartment": True,
+            "diagnosticPlanMustBeConfirmedBySmme": True,
+            "groupedAssignmentRequiresSharedIntervention": True,
+            "recurringCannotSkipOpenPreviousCycle": True,
+            "onceOffCannotRepeatAfterCompletion": True,
+        },
+        "smes": [],
+        "interventions": [],
+        "assignments": [],
+        "consultants": []
+    }
+
 # -- route ---------------------------------------------------------------
 
 @app.route('/chat', methods=['POST'])
@@ -935,6 +976,180 @@ def chat():
     final_response = ask_gpt([system_msg, data_msg, user_msg])
     return jsonify({"reply": final_response})
 
+@app.route("/page-assist", methods=["POST"])
+def page_assist():
+    data = request.json or {}
+
+    page = data.get("page")
+    message = data.get("message")
+    role = data.get("role")
+    company_code = data.get("companyCode")
+    user_id = data.get("userId")
+    program_id = data.get("programId")  # optional, "all" allowed
+
+    if not page or not message or not role or not company_code or not user_id:
+        return jsonify({"error": "Missing page, message, role, companyCode, or userId"}), 400
+
+    if page != "assignments":
+        return jsonify({"error": "Unsupported page"}), 400
+
+    ctx = resolve_user_context(user_id, company_code)
+
+    assignment_context = build_assignments_page_context(
+        company_code=company_code,
+        user_id=user_id,
+        program_id=None if program_id in [None, "", "all"] else program_id,
+        role=role,
+        user_context=ctx,
+    )
+
+    system_msg = {
+        "role": "system",
+        "content": """
+You are a page-aware assistant for the Interventions Assignments page.
+
+Answer using ONLY the supplied page context.
+Explain blockers clearly.
+Do not invent SMEs, interventions, departments, or assignment rules.
+If asked why an intervention cannot be assigned, identify the exact reason:
+- not part of that SME's Diagnostic Plan
+- Diagnostic Plan not confirmed by both department and SME
+- not shared across selected SMEs
+- previous recurring cycle still open
+- once-off intervention already completed
+- missing facilitator, due date, target mode, or sub-intervention
+Return concise JSON:
+{
+  "reply": "",
+  "reasonCode": "",
+  "affectedSmes": [],
+  "eligibleSmes": [],
+  "blockedSmes": [],
+  "suggestedNextAction": ""
+}
+"""
+    }
+
+    user_msg = {
+        "role": "user",
+        "content": json.dumps({
+            "question": message,
+            "pageContext": assignment_context,
+        }, ensure_ascii=False)
+    }
+
+    ai_raw = ask_gpt([system_msg, user_msg])
+    result = _extract_json_block(ai_raw)
+
+    return jsonify(to_jsonable(result))
+
+@app.route("/assignments-agent", methods=["POST"])
+def assignments_agent():
+    data = request.json or {}
+
+    company_code = data.get("companyCode")
+    user_id = data.get("userId")
+    role = data.get("role")
+    message = data.get("message") or data.get("transcript")
+    program_id = data.get("programId")
+
+    if not company_code or not user_id or not role or not message:
+        return jsonify({"error": "Missing companyCode, userId, role, or message"}), 400
+
+    ctx = resolve_user_context(user_id, company_code)
+
+    assignment_context = build_assignments_page_context(
+        company_code=company_code,
+        user_id=user_id,
+        program_id=None if program_id in [None, "", "all"] else program_id,
+        role=role,
+        user_context=ctx,
+    )
+
+    system_msg = {
+        "role": "system",
+        "content": """
+You are an assignment action planner.
+
+Extract the user's intent, but DO NOT save anything directly.
+Return strict JSON:
+{
+  "intent": "assign_intervention|explain|unknown",
+  "interventionName": "",
+  "smeNames": [],
+  "assignmentType": "singular|grouped|unknown",
+  "assignTo": "consultant|operations|unknown",
+  "consultantName": "",
+  "dueDate": "",
+  "targetMode": "hours|sessions|documents|milestones|percentage|unknown",
+  "missingFields": [],
+  "canExecute": false,
+  "validationErrors": [],
+  "confirmationMessage": ""
+}
+
+Required fields before execution:
+- interventionName
+- at least one SME
+- assignTo
+- dueDate
+- targetMode
+- consultantName if assignTo is consultant
+
+Validate against pageContext.
+"""
+    }
+
+    user_msg = {
+        "role": "user",
+        "content": json.dumps({
+            "message": message,
+            "pageContext": assignment_context,
+        }, ensure_ascii=False)
+    }
+
+    ai_raw = ask_gpt([system_msg, user_msg])
+    plan = _extract_json_block(ai_raw)
+
+    if not plan.get("canExecute"):
+        return jsonify({
+            "mode": "needs_more_info",
+            "plan": plan,
+            "reply": plan.get("confirmationMessage") or "I need more information before creating the assignment."
+        })
+
+    validation = validate_assignment_plan(plan, assignment_context)
+
+    if not validation["ok"]:
+        return jsonify({
+            "mode": "blocked",
+            "reply": validation["message"],
+            "validationErrors": validation["errors"],
+            "plan": plan
+        })
+
+    # Do not write immediately unless frontend sends confirmed=true
+    if not data.get("confirmed"):
+        return jsonify({
+            "mode": "confirmation_required",
+            "reply": validation["confirmationMessage"],
+            "plan": plan
+        })
+
+    saved = create_assignments_from_plan(
+        plan=plan,
+        company_code=company_code,
+        user_id=user_id,
+        user_context=ctx,
+        assignment_context=assignment_context,
+    )
+
+    return jsonify({
+        "mode": "executed",
+        "reply": f"Created {saved['count']} assignment(s) successfully.",
+        "createdIds": saved["ids"]
+    })
+
 # --- NEW: lightweight Help/Assistant intent router -------------------------
 @app.route('/assist', methods=['POST'])
 def assist():