Update app.py

app.py

@@ -12,14 +12,10 @@ from langchain_groq import ChatGroq
 from typing_extensions import TypedDict
 
 # --- Type Definitions ---
-class TaggedReply(TypedDict):
-    reply: str
-    tags: List[str]
-
 class AssistantState(TypedDict):
     conversationSummary: str
     language: str 
-    taggedReplies: List[TaggedReply]
+    mode: str  # "teacher" or "student"
 
 # --- Logging ---
 logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
@@ -61,14 +57,29 @@ def detect_language_from_text(text: str) -> Optional[str]:
 
 def update_summary(chat_history: List[Dict[str, str]]) -> str:
     """
-    Placeholder for conversation summary update logic.
-    You can replace this with a call to a summarization LLM chain or heuristic.
-    For now, returns last 3 messages concatenated as a simple summary.
+    Simple heuristic summary: last 6 messages concatenated.
+    Replace with your own summarization chain if desired.
     """
-    recent_msgs = chat_history[-6:]  # last 6 messages (user+assistant)
+    recent_msgs = chat_history[-6:]
     summary = " | ".join(f"{m['role']}: {m['content'][:50].replace('\n',' ')}" for m in recent_msgs)
     return summary
 
+def build_system_prompt(language: str, conversation_summary: str, mode: str) -> str:
+    """
+    Build system prompt dynamically based on mode.
+    """
+    base = f"You are a helpful programming assistant. Current language: {language}. Conversation summary: {conversation_summary}\n\n"
+    if mode == "student":
+        base += (
+            "You are in STUDENT MODE: Guide the user on *how to achieve* their programming tasks. "
+            "Provide general explanations and pseudocode examples when appropriate. Avoid full detailed code unless necessary."
+        )
+    else:  # teacher mode default
+        base += (
+            "You are in TEACHER MODE: Provide detailed suggestions, structured explanations, and full code examples."
+        )
+    return base
+
 # --- Routes ---
 
 @app.route("/", methods=["GET"])
@@ -86,7 +97,9 @@ def chat():
 
     conversation_summary = assistant_state.get("conversationSummary", "")
     language = assistant_state.get("language", "Python")
-    tagged_replies = assistant_state.get("taggedReplies", [])
+    mode = assistant_state.get("mode", "teacher").lower()
+    if mode not in ("teacher", "student"):
+        mode = "teacher"
 
     # Detect language from last user message
     last_user_msg = ""
@@ -99,8 +112,8 @@ def chat():
         logger.info(f"Detected new language: {detected_lang}")
         language = detected_lang
 
-    # Build prompt with system + conversation summary + chat history
-    system_prompt = f"You are a helpful programming assistant. Current language: {language}. Conversation summary: {conversation_summary}"
+    # Build system prompt based on mode
+    system_prompt = build_system_prompt(language, conversation_summary, mode)
     messages = [{"role": "system", "content": system_prompt}]
     messages.extend(chat_history)
 
@@ -114,7 +127,7 @@ def chat():
             "updated_state": {
                 "conversationSummary": conversation_summary,
                 "language": language,
-                "taggedReplies": tagged_replies,
+                "mode": mode,
             },
             "chat_history": chat_history,
         }), 500
@@ -130,48 +143,15 @@ def chat():
         "updated_state": {
             "conversationSummary": conversation_summary,
             "language": language,
-            "taggedReplies": tagged_replies,
+            "mode": mode,
         },
         "chat_history": chat_history,
     })
 
-@app.route("/tag_reply", methods=["POST"])
-def tag_reply():
-    data = request.get_json(force=True)
-    if not isinstance(data, dict):
-        return jsonify({"error": "invalid request body"}), 400
-
-    reply_content = data.get("reply")
-    tags = data.get("tags")
-    assistant_state: AssistantState = data.get("assistant_state") or {}
-
-    if not reply_content or not tags:
-        return jsonify({"error": "Missing 'reply' or 'tags' in request"}), 400
-
-    tags = [str(t).strip() for t in tags if str(t).strip()]
-    if not tags:
-        return jsonify({"error": "Tags list cannot be empty"}), 400
-
-    tagged_replies = assistant_state.get("taggedReplies", [])
-    tagged_replies.append({"reply": reply_content, "tags": tags})
-
-    updated_state = {
-        "conversationSummary": assistant_state.get("conversationSummary", ""),
-        "language": assistant_state.get("language", "Python"),
-        "taggedReplies": tagged_replies,
-    }
-
-    logger.info(f"Reply tagged with: {tags}")
-
-    return jsonify({
-        "message": "Reply saved and tagged successfully.",
-        "updated_state": updated_state,
-    }), 200
-
 @app.route("/ping", methods=["GET"])
 def ping():
     return jsonify({"status": "ok"})
 
 if __name__ == "__main__":
     port = int(os.getenv("PORT", 7860))
-    app.run(host="0.0.0.0", port=port, debug=True)
+    app.run(host="0.0.0.0", port=port, debug=True)