Update app.py

app.py

@@ -72,15 +72,39 @@ def parse_raw_text_to_json(raw_text):
     tips = []
     quote = "Every step forward counts!"
 
-    # Simple parsing logic: look for keywords or assume structure
+    checklist_section = False
+    tips_section = False
+
     for line in lines:
         line = line.strip()
-        if line.startswith("- "):  # Assume checklist items start with "- "
+        if not line:
+            continue
+        if line.lower().startswith("checklist:"):
+            checklist_section = True
+            tips_section = False
+            continue
+        elif line.lower().startswith("tips:"):
+            checklist_section = False
+            tips_section = True
+            continue
+        elif line.lower().startswith("quote:"):
+            checklist_section = False
+            tips_section = False
+            quote = line[6:].strip() or quote
+            continue
+
+        if checklist_section and line.startswith("- "):
             checklist.append(line[2:].strip())
-        elif line.startswith("* "):  # Assume tips start with "* "
+        elif tips_section and line.startswith("* "):
             tips.append(line[2:].strip())
-        elif line and not checklist and not tips:  # If no checklist or tips, treat as a quote
-            quote = line
+        elif not checklist and not tips and line:
+            # If no sections are defined, try to infer structure
+            if line.startswith("- "):
+                checklist.append(line[2:].strip())
+            elif line.startswith("* "):
+                tips.append(line[2:].strip())
+            else:
+                quote = line
 
     # Fallback if parsing fails
     if not checklist:
@@ -107,6 +131,29 @@ def health_check():
         "model_load_status": model_load_status
     })
 
+@app.route("/debug", methods=["POST"])
+def debug():
+    logger.debug("Debug endpoint accessed")
+    data = request.get_json()
+    if not data:
+        return jsonify({"error": "Invalid request: JSON data required"}), 400
+
+    required_fields = ["role", "project_id", "milestones", "reflection_log"]
+    missing_fields = [field for field in required_fields if field not in data]
+    if missing_fields:
+        return jsonify({"error": f"Missing required fields: {missing_fields}"}), 400
+
+    input_text = (
+        f"Role: {data['role']}, Project: {data['project_id']}, "
+        f"Milestones: {data['milestones']}, Reflection: {data['reflection_log']}"
+    )
+
+    return jsonify({
+        "model_load_status": model_load_status,
+        "input_text": input_text,
+        "model_ready": model is not None and tokenizer is not None
+    })
+
 @app.route("/generate_coaching", methods=["POST"])
 def generate_coaching():
     logger.debug("Generate coaching endpoint accessed")
@@ -132,10 +179,14 @@ def generate_coaching():
         })
 
     try:
-        # Prepare input text
+        # Prepare input text with a structured prompt to encourage formatted output
         input_text = (
             f"Role: {data['role']}, Project: {data['project_id']}, "
-            f"Milestones: {data['milestones']}, Reflection: {data['reflection_log']}"
+            f"Milestones: {data['milestones']}, Reflection: {data['reflection_log']}\n\n"
+            "Generate a coaching response in the following format:\n"
+            "Checklist:\n- <item1>\n- <item2>\n"
+            "Tips:\n* <tip1>\n* <tip2>\n"
+            "Quote: <motivational quote>"
         )
        
         # Tokenize input
@@ -153,6 +204,7 @@ def generate_coaching():
        
         # Decode response
         response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+        logger.debug(f"Raw model output: {response_text}")
        
         # Try parsing as JSON first
         try:
@@ -177,5 +229,4 @@ if __name__ == "__main__":
     # Run Flask app with waitress for production-ready WSGI server
     from waitress import serve
     logger.debug("Starting Flask app with Waitress")
-    serve(app, host="0.0.0.0", port=7860)
-
+    serve(app, host="0.0.0.0", port=7860)