Update chain_problems.py

chain_problems.py

@@ -1,3 +1,4 @@
+# chain_problems.py
 import json
 import logging
 from typing import Dict
@@ -6,7 +7,6 @@ from models import chat_model
 
 logger = logging.getLogger(__name__)
 
-# Updated prompt template with guidance connecting themes to question responses
 problem_prompt_template = PromptTemplate(
     input_variables=["responses", "internal_report"],
     template=(
@@ -16,6 +16,8 @@ problem_prompt_template = PromptTemplate(
         "{internal_report}\n\n"
         "From these inputs, determine a 'problem severity percentage' for the user in the following areas: "
         "stress_management, low_therapy, balanced_weight, restless_night, lack_of_motivation, gut_health, anxiety, burnout.\n\n"
+        "Consider how these severity percentages will later be used to recommend suitable wellness packages such as "
+        "'Fitness & Mobility', 'No More Insomnia', 'Focus Flow', 'Boost Energy', 'Chronic Care', and 'Mental Wellness'.\n\n"
         "Consider the following connections between the questions and these themes:\n"
         "- stress_management is influenced by responses such as stress_level, stress_management, mood, mindfulness_frequency, and similar stress-related questions.\n"
         "- low_therapy relates to aspects of the user's mindset, wellness_goals, personal_growth_reflection, and similar therapeutic indicators.\n"