Update chain_problems.py

chain_problems.py

@@ -6,7 +6,7 @@ from models import chat_model
 
 logger = logging.getLogger(__name__)
 
-# Updated prompt template to include eight new themes
+# Updated prompt template with guidance connecting themes to question responses
 problem_prompt_template = PromptTemplate(
     input_variables=["responses", "internal_report"],
     template=(
@@ -15,7 +15,16 @@ problem_prompt_template = PromptTemplate(
         "You also have an internal analysis report:\n"
         "{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. "
+        "stress_management, low_therapy, balanced_weight, restless_night, lack_of_motivation, gut_health, anxiety, burnout.\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"
+        "- balanced_weight depends on exercise, eating_habits, dietary_restrictions, activity_tracking, and other fitness/nutrition details.\n"
+        "- restless_night is linked to answers about sleep duration, bedtime_routine, uninterrupted_sleep, and other sleep quality indicators.\n"
+        "- lack_of_motivation correlates with mood, energy_rating, personal_growth_reflection, break_frequency, and similar motivation-related queries.\n"
+        "- gut_health is connected to eating_habits, dietary_restrictions, health_issues, and other digestive or nutritional feedback.\n"
+        "- anxiety is associated with responses about mood, stress_level, mindset, and related emotional well-being questions.\n"
+        "- burnout may be reflected in high stress_level, low energy_rating, lack_of_motivation, and related fatigue or overwhelm indicators.\n\n"
         "Return your answer in JSON format with keys: stress_management, low_therapy, balanced_weight, restless_night, "
         "lack_of_motivation, gut_health, anxiety, burnout.\n"
         "Ensure severity percentages are numbers from 0 to 100.\n\n"