Update main.py

main.py

@@ -1648,8 +1648,294 @@ def image_proxy():
         return jsonify({"error": "Internal server error processing the image request."}), 500
 
 
+# =============================================================================
+# 11. LEARNING PATH MODULE (NEW ADDITION)
+# =============================================================================
+# This module implements the "Overlay Architecture".
+# It fetches static structure from the Data API and overlays user progress from Firebase.
+
+# --- Configuration ---
+
+DATA_API_BASE_URL = "https://rairo-marka-data-api.hf.space/v1"
+
+# Registry mapping your App IDs to the Data API Filename IDs (Based on your screenshots)
+PATH_SUBJECT_MAP = {
+    # --- A Level ---
+    "al_acc":    "A_9706", # Accounting
+    "al_bio":    "A_9700", # Biology
+    "al_bus":    "A_9609", # Business
+    "al_chem":   "A_9701", # Chemistry
+    "al_cs":     "A_9618", # Computer Science
+    "al_eco":    "A_9708", # Economics
+    "al_fmath":  "A_9231", # Further Mathematics
+    "al_hist":   "A_9489", # History
+    "al_lit":    "A_9695", # Literature in English
+    "al_math":   "A_9709", # Mathematics
+    "al_phy":    "A_9702", # Physics
+    "al_soc":    "A_9699", # Sociology
+    "al_travel": "A_9395", # Travel & Tourism
+
+    # --- O Level (IGCSE) ---
+    "ol_acc":    "O_0452", # Accounting
+    "ol_bio":    "O_0610", # Biology
+    "ol_bus":    "O_0450", # Business Studies
+    "ol_chem":   "O_0620", # Chemistry
+    "ol_cs":     "O_0478", # Computer Science
+    "ol_eng":    "O_0500", # English Language
+    "ol_lit":    "O_0475", # English Literature
+    "ol_env":    "O_0680", # Environmental Management
+    "ol_geo":    "O_0460", # Geography
+    "ol_hist":   "O_0470", # History
+    "ol_phy":    "O_0625", # Physics
+}
+
+# --- Helpers ---
+
+def fetch_remote_syllabus(remote_id):
+    """Fetches the static JSON tree from the Data API."""
+    try:
+        url = f"{DATA_API_BASE_URL}/structure/{remote_id}"
+        resp = requests.get(url, timeout=10)
+        if resp.status_code == 200:
+            return resp.json()
+        logger.error(f"Data API Error {resp.status_code}: {resp.text}")
+        return None
+    except Exception as e:
+        logger.error(f"Failed to connect to Data API: {e}")
+        return None
+
+def search_remote_context(query, remote_id):
+    """Searches the Data API for specific syllabus content to ground the AI."""
+    try:
+        url = f"{DATA_API_BASE_URL}/search"
+        payload = {
+            "query": query,
+            "filter_subject_id": remote_id
+        }
+        resp = requests.post(url, json=payload, timeout=10)
+        if resp.status_code == 200:
+            return resp.json().get("results", [])
+        return []
+    except Exception as e:
+        logger.error(f"Context search failed: {e}")
+        return []
+
+# --- Endpoints ---
+
+@app.route("/api/path/tree", methods=["GET"])
+def get_path_tree():
+    """
+    Returns the full syllabus tree with 'locked', 'unlocked', or 'completed' status.
+    Query Param: subjectId (e.g., 'al_phy')
+    """
+    uid = verify_token(request.headers.get("Authorization"))
+    if not uid:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    subject_id = request.args.get("subjectId")
+    if not subject_id or subject_id not in PATH_SUBJECT_MAP:
+        return jsonify({"error": "Invalid or unsupported subjectId"}), 400
+
+    remote_id = PATH_SUBJECT_MAP[subject_id]
+
+    # 1. Get Static Map
+    syllabus_data = fetch_remote_syllabus(remote_id)
+    if not syllabus_data:
+        return jsonify({"error": "Syllabus data currently unavailable"}), 503
+
+    # 2. Get User Progress
+    # Stored as: users/{uid}/path_progress/{subject_id}/{node_id} = timestamp
+    progress_ref = db_ref.child(f"users/{uid}/path_progress/{subject_id}")
+    user_progress = progress_ref.get() or {}
+
+    # 3. Recursive Overlay Logic
+    def enrich_node(node, parent_is_completed):
+        node_id = node.get("id")
+        
+        # Check completion status
+        completed_timestamp = user_progress.get(node_id)
+        is_completed = completed_timestamp is not None
+
+        # Determine State
+        status = "locked"
+        if is_completed:
+            status = "completed"
+        elif parent_is_completed:
+            # If parent is done (or it's a root node), this is the next step
+            status = "unlocked"
+        
+        enriched = {
+            "id": node_id,
+            "title": node.get("title"),
+            "type": node.get("type"),
+            "status": status,
+            "isCompleted": is_completed
+        }
+
+        # Handle Children
+        if "children" in node:
+            # Children are only accessible if THIS node is completed
+            # (Or if it's a container 'topic' that doesn't require work itself, pass parent state)
+            # Strategy: Topics (Folders) are 'completed' if any child is done? 
+            # Simple Strategy: Flow down. Children unlock if parent is completed.
+            enriched["children"] = [
+                enrich_node(child, parent_is_completed=is_completed or (node.get("type") == "topic" and parent_is_completed)) 
+                for child in node["children"]
+            ]
+        
+        return enriched
+
+    # 4. Process Root
+    # The API returns { meta: {...}, tree: [...] }
+    raw_tree = syllabus_data.get("tree", [])
+    # Root nodes are always unlocked
+    enriched_tree = [enrich_node(root, parent_is_completed=True) for root in raw_tree]
+
+    return jsonify({
+        "subjectId": subject_id,
+        "remoteId": remote_id,
+        "tree": enriched_tree
+    })
+
+
+@app.route("/api/path/generate-lesson", methods=["POST"])
+def generate_path_lesson():
+    """
+    Generates a revision lesson grounded in the specific syllabus point.
+    Body: { subjectId, nodeId, nodeTitle }
+    """
+    uid = verify_token(request.headers.get("Authorization"))
+    if not uid:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.get_json() or {}
+    subject_id = data.get("subjectId")
+    node_id = data.get("nodeId") # e.g. "A_9702_1.1"
+    node_title = data.get("nodeTitle")
+
+    if not subject_id or not node_title:
+        return jsonify({"error": "Missing fields"}), 400
+
+    remote_id = PATH_SUBJECT_MAP.get(subject_id)
+
+    # 1. Retrieve Context
+    # We search specifically for the title to get the exact syllabus bullet points
+    context_chunks = search_remote_context(node_title, remote_id)
+    context_text = "\n".join([c.get("content", "") for c in context_chunks[:3]])
+
+    # 2. AI Generation
+    prompt = f"""
+    You are an expert tutor. Create a concise revision lesson.
+    
+    Topic: {node_title}
+    Syllabus Context (Strictly adhere to this scope):
+    {context_text}
+
+    Structure:
+    1. **Concept Summary**: Simple explanation (2 paragraphs).
+    2. **Key Formulas/Definitions**: If applicable.
+    3. **Worked Example**: A typical exam-style problem with solution.
+    4. **Common Pitfalls**: What do students usually get wrong?
+
+    Format using Markdown.
+    """
+    
+    lesson_text = send_gemini_text(prompt)
+
+    return jsonify({
+        "nodeId": node_id,
+        "lessonContent": lesson_text
+    })
+
+
+@app.route("/api/path/generate-quiz", methods=["POST"])
+def generate_path_quiz():
+    """
+    Generates a quiz to test mastery of a node.
+    Body: { subjectId, nodeId, nodeTitle }
+    """
+    uid = verify_token(request.headers.get("Authorization"))
+    if not uid:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.get_json() or {}
+    subject_id = data.get("subjectId")
+    node_title = data.get("nodeTitle")
+    remote_id = PATH_SUBJECT_MAP.get(subject_id)
+
+    # 1. Retrieve Context
+    context_chunks = search_remote_context(node_title, remote_id)
+    context_text = "\n".join([c.get("content", "") for c in context_chunks[:3]])
+
+    # 2. AI Generation
+    prompt = f"""
+    Generate 5 Multiple Choice Questions to test: {node_title}
+    Use this syllabus context as ground truth:
+    {context_text}
+
+    Return JSON ONLY:
+    {{
+        "questions": [
+            {{
+                "stem": "Question text...",
+                "options": ["A", "B", "C", "D"],
+                "answer": "Correct Option Text"
+            }}
+        ]
+    }}
+    """
+    
+    raw = send_gemini_text(prompt)
+    try:
+        clean = raw.replace("```json", "").replace("```", "").strip()
+        quiz_data = json.loads(clean)
+    except Exception as e:
+        logger.error(f"Failed to parse quiz JSON: {raw}")
+        return jsonify({"error": "Failed to generate quiz"}), 500
+
+    return jsonify(quiz_data)
+
+
+@app.route("/api/path/submit-node", methods=["POST"])
+def submit_path_node():
+    """
+    Records success on a node, unlocking the next step.
+    Body: { subjectId, nodeId, score, total }
+    """
+    uid = verify_token(request.headers.get("Authorization"))
+    if not uid:
+        return jsonify({"error": "Unauthorized"}), 401
+
+    data = request.get_json() or {}
+    subject_id = data.get("subjectId")
+    node_id = data.get("nodeId")
+    score = data.get("score", 0)
+    total = data.get("total", 1)
+
+    if not subject_id or not node_id:
+        return jsonify({"error": "Missing fields"}), 400
+
+    percentage = (score / total) * 100
+    passed = percentage >= 70  # Pass threshold
+
+    if passed:
+        # Mark as complete in Firebase
+        # We record the timestamp of completion
+        db_ref.child(f"users/{uid}/path_progress/{subject_id}/{node_id}").set(int(time.time()))
+        
+        return jsonify({
+            "success": True,
+            "passed": True,
+            "message": "Node completed! Next topic unlocked."
+        })
+    else:
+        return jsonify({
+            "success": True,
+            "passed": False,
+            "message": "Score too low. Review the lesson and try again."
+        })
 # -----------------------------------------------------------------------------
-# 11. MAIN
+# 12. MAIN
 # -----------------------------------------------------------------------------
 
 if __name__ == "__main__":