Sync from GitHub · 2026-09-13 10:42 UTC

routes/bff.py

@@ -20,6 +20,11 @@ class GenerateFollowupsRequest(BaseModel):
 class GenerateProfileRequest(BaseModel):
     userId: str
     businessId: str
+    force: bool = False   # regenerate outputs even if they already exist (Settings edits)
+
+
+class ReopenFollowupsRequest(BaseModel):
+    businessId: str
 
 
 # ─── Helpers ──────────────────────────────────────────────────────────────────
@@ -49,6 +54,60 @@ def _get_bff_data(user_id: str, business_id: str) -> dict:
     return sections
 
 
+# ─── Business stage ───────────────────────────────────────────────────────────
+# Keep ids in sync with data/business-stages.ts on the frontend.
+
+BUSINESS_STAGES = {
+    "idea":        "Just an idea — has not started anything yet (Ideation stage)",
+    "validation":  "Checking if the idea will work — researching and talking to possible customers (Validation / Planning stage)",
+    "prelaunch":   "Getting ready to launch — registering, raising money, setting up (Pre-launch stage)",
+    "launched":    "Just started — selling for less than 1 year (Launch / Startup stage)",
+    "early":       "Running for 1–2 years (Early stage)",
+    "growth":      "Running for 3–5 years (Growth stage)",
+    "established": "Running for 6–10 years (Established / Expansion stage)",
+    "mature":      "Running for over 10 years (Mature stage)",
+    "restarting":  "Restarting or changing direction — has run a business before and is starting again or pivoting",
+}
+
+PRE_LAUNCH_STAGES = {"idea", "validation", "prelaunch"}
+
+
+def _get_business_stage(user_id: str, business_id: str) -> str:
+    """Reads the stage saved on the WHAT step. Returns '' if not set."""
+    snap = (
+        _db().collection("user_data").document(user_id)
+        .collection("businesses").document(business_id)
+        .collection("bff").document("section1").get()
+    )
+    return (snap.to_dict() or {}).get("businessStage", "") if snap.exists else ""
+
+
+def _stage_guidance(stage: str) -> str:
+    """Prompt instructions that keep the AI's tense and assumptions matched to the stage."""
+    if stage not in BUSINESS_STAGES:
+        return ""
+    label = BUSINESS_STAGES[stage]
+    if stage in PRE_LAUNCH_STAGES:
+        rule = (
+            "This business has NOT launched yet. Never assume it has customers, sales, staff, "
+            "revenue or a track record. Phrase everything around plans, expectations and "
+            "assumptions (e.g. 'Who do you expect to buy this?' not 'Who buys from you?')."
+        )
+    elif stage == "restarting":
+        rule = (
+            "The entrepreneur has run a business before and is starting again or changing direction. "
+            "Treat the new direction as not yet proven: phrase it around plans, but draw on their "
+            "past experience (what worked, what didn't, what they are changing and why)."
+        )
+    else:
+        rule = (
+            "This business is already operating. Ask about real experience so far — actual "
+            "customers, what sells, what has worked and what hasn't — at a depth that suits "
+            "how long it has been running."
+        )
+    return f"\n\nBUSINESS STAGE: {label}\n{rule}"
+
+
 def _call_groq_json(system_prompt: str, user_prompt: str) -> dict:
     """
     Thin wrapper — delegates to the key-rotating Groq client utility.
@@ -82,12 +141,17 @@ def generate_followups(
     bff_query = f"Business Footing Framework section {body.section} {json.dumps(body.sectionData)[:200]}"
     book_ctx   = query_book_context(bff_query, top_k=2)
 
+    # Stage is read server-side so WHY/HOW follow-ups get it too, not just WHAT
+    stage = body.sectionData.get("businessStage") if body.section == 1 else None
+    stage = stage or _get_business_stage(user["uid"], body.businessId)
+
     system_prompt = (
         "You are a business consultant helping a Nigerian entrepreneur understand their business better. "
         "Generate follow-up questions based on what they have already shared. "
         "Questions MUST use simple, everyday English — no business jargon. "
         "Think: how would a friendly, experienced mentor ask? "
         "Return ONLY valid JSON."
+        + _stage_guidance(stage)
         + (f"\n\n{book_ctx}" if book_ctx else "")
     )
 
@@ -166,6 +230,7 @@ def generate_profile(
         "Write in plain English that ANY entrepreneur can understand. "
         "Ground all analysis in the Nigerian market context. "
         "Return ONLY valid JSON."
+        + _stage_guidance(bff.get("section1", {}).get("businessStage", ""))
         + (f"\n\n{book_ctx}" if book_ctx else "")
     )
 
@@ -276,11 +341,17 @@ each containing a list of 2-4 concise bullet points:
         "businessModel":    prompt_bmc,
     }
 
+    # profileStale is set by /reopen-followups: the old profile was built without
+    # follow-up answers, so it must be rebuilt even if the outputs already exist.
+    # Read server-side so it survives the user leaving and resuming later.
+    biz_snap = biz_ref_check.get()
+    force = body.force or bool((biz_snap.to_dict() or {}).get("profileStale"))
+
     call_plan = []
     skipped_keys: set = set()
     for key, prompt_fn in all_outputs.items():
-        existing = biz_ref_check.collection("profile_output").document(key).get()
-        if existing.exists:
+        existing = None if force else biz_ref_check.collection("profile_output").document(key).get()
+        if existing is not None and existing.exists:
             print(f"[bff] Skipping {key} — already exists in Firestore")
             skipped_keys.add(key)
         else:
@@ -331,10 +402,17 @@ each containing a list of 2-4 concise bullet points:
     # business look "complete" with an empty/partial profile (see
     # project_multibusiness_gate memory — the manual-Console-flip incident
     # had the same shape of bug).
+    # On a forced run, outputs that failed to regenerate still exist from before,
+    # so count them as present rather than blocking completion.
+    if force:
+        skipped_keys = {
+            k for k in all_outputs
+            if k not in results and biz_ref.collection("profile_output").document(k).get().exists
+        }
     present_keys = skipped_keys | set(results.keys())
     all_present  = present_keys == set(all_outputs.keys())
     if all_present:
-        biz_ref.update({"bffCompleted": True})
+        biz_ref.update({"bffCompleted": True, "profileStale": False})
     else:
         missing = set(all_outputs.keys()) - present_keys
         print(f"[bff] NOT marking bffCompleted — missing outputs: {missing}")
@@ -344,3 +422,44 @@ each containing a list of 2-4 concise bullet points:
         "outputs": list(results.keys()),
         "missing": sorted(set(all_outputs.keys()) - present_keys),
     }
+
+
+# ─── Route: Reopen onboarding for missing follow-ups ─────────────────────────
+
+def _missing_followup_sections(user_id: str, business_id: str) -> list[int]:
+    """Sections (1-3) whose follow-up questions were never generated/saved."""
+    bff_ref = (
+        _db().collection("user_data").document(user_id)
+        .collection("businesses").document(business_id)
+        .collection("bff")
+    )
+    missing = []
+    for section in (1, 2, 3):
+        snap = bff_ref.document(f"section{section}_followups").get()
+        if not snap.exists or not (snap.to_dict() or {}).get("questions"):
+            missing.append(section)
+    return missing
+
+
+@router.post("/reopen-followups")
+def reopen_followups(
+    body: ReopenFollowupsRequest,
+    user: dict = Depends(verify_firebase_token),
+):
+    """
+    A profile was generated while some follow-up steps failed or were skipped.
+    Marks the business incomplete (bffCompleted is backend-only per Firestore
+    rules) and flags the profile stale so the next generate-profile rebuilds it
+    with the follow-up answers included.
+    """
+    biz_ref = (
+        _db().collection("user_data").document(user["uid"])
+        .collection("businesses").document(body.businessId)
+    )
+    if not biz_ref.get().exists:
+        raise HTTPException(status_code=404, detail="Business not found.")
+
+    missing = _missing_followup_sections(user["uid"], body.businessId)
+    if missing:
+        biz_ref.update({"bffCompleted": False, "profileStale": True})
+    return {"missing": missing}