rules.py

"""
Rule-based risk calculation for anemia screening
This is NOT a medical diagnosis tool
"""

from ml_model import analyze_eye_image


def calculate_risk(answers, image_path=None):
    """
    Calculate anemia risk based on answers
    
    Args:
        answers (dict): User responses to questions
        image_path (str): Optional path to eye image
    
    Returns:
        tuple: (risk_level, score, details)
    """
    score = 0
    details = {
        'factors': [],
        'recommendations': []
    }
    
    # Age and gender factors
    age = int(answers.get('age', 0))
    gender = answers.get('gender', '').lower()
    
    # Higher risk for children and women
    if age < 18:
        score += 1
        details['factors'].append('Age under 18 (higher risk group)')
    
    if gender == 'female':
        score += 1
        details['factors'].append('Female gender (higher risk due to menstruation)')
    
    # Symptom-based scoring
    symptoms = [
        ('fatigue', 'Frequent fatigue', 2),
        ('dizziness', 'Dizziness or lightheadedness', 2),
        ('pale_skin', 'Pale skin or lips', 2),
        ('breathlessness', 'Shortness of breath', 2),
        ('low_iron_diet', 'Low iron diet', 2),
        ('cold_hands', 'Cold hands or feet', 1),
        ('headaches', 'Frequent headaches', 1),
        ('weakness', 'Muscle weakness', 1)
    ]
    
    for symptom_key, symptom_name, points in symptoms:
        if answers.get(symptom_key) == 'yes':
            score += points
            details['factors'].append(symptom_name)
    
    # Determine risk level
    if score <= 3:
        risk_level = 'Low Risk'
        details['recommendations'] = [
            'Your screening shows low risk for anemia.',
            'Maintain a balanced diet rich in iron (meat, beans, leafy greens).',
            'Consider routine health checkups annually.',
            'Stay hydrated and get adequate sleep.'
        ]
    elif score <= 7:
        risk_level = 'Moderate Risk'
        details['recommendations'] = [
            'Your screening shows moderate risk for anemia.',
            'Consider getting a Complete Blood Count (CBC) test.',
            'Increase iron-rich foods in your diet.',
            'Consult a healthcare provider if symptoms persist.',
            'Vitamin C helps iron absorption - include citrus fruits.'
        ]
    else:
        risk_level = 'High Risk'
        details['recommendations'] = [
            '⚠️ Your screening shows high risk for anemia.',
            '🏥 IMPORTANT: Please consult a doctor immediately.',
            'Get a Complete Blood Count (CBC) and hemoglobin test.',
            'Do NOT self-diagnose or self-medicate.',
            'This screening is NOT a diagnosis - lab tests are required.'
        ]
    
    # Add ML-based image analysis
    ml_prediction = None
    if image_path:
        try:
            ml_results = analyze_eye_image(image_path)
            ml_prediction = ml_results['probability']
            
            details['image_analyzed'] = True
            details['ml_probability'] = ml_prediction
            details['ml_confidence'] = ml_results['confidence']
            details['factors'].append(f"Eye image analysis: {ml_results['interpretation']}")
            
            # Adjust risk score based on ML prediction
            if ml_prediction >= 0.7:
                score += 3
                details['factors'].append('ML model indicates HIGH risk from image analysis')
            elif ml_prediction >= 0.5:
                score += 2
                details['factors'].append('ML model indicates MODERATE risk from image analysis')
            elif ml_prediction >= 0.3:
                score += 1
                details['factors'].append('ML model indicates LOW-MODERATE risk from image analysis')
            else:
                details['factors'].append('ML model indicates LOW risk from image analysis')
            
            # Re-evaluate risk level with ML input
            if score <= 3:
                risk_level = 'Low Risk'
            elif score <= 8:
                risk_level = 'Moderate Risk'
            else:
                risk_level = 'High Risk'
                
        except Exception as e:
            print(f"Error in ML prediction: {e}")
            details['image_analyzed'] = True
            details['factors'].append('Eye image uploaded (ML analysis unavailable)')
    else:
        details['image_analyzed'] = False
    
    return risk_level, score, details


def get_questions():
    """
    Return the list of screening questions
    
    Returns:
        list: List of question dictionaries
    """
    questions = [
        {
            'id': 'age',
            'type': 'number',
            'text': 'What is your age?',
            'required': True,
            'min': 1,
            'max': 120
        },
        {
            'id': 'gender',
            'type': 'choice',
            'text': 'What is your gender?',
            'required': True,
            'options': ['Male', 'Female', 'Other']
        },
        {
            'id': 'fatigue',
            'type': 'choice',
            'text': 'Do you feel tired or fatigued most days?',
            'required': True,
            'options': ['Yes', 'No']
        },
        {
            'id': 'dizziness',
            'type': 'choice',
            'text': 'Do you experience dizziness or lightheadedness?',
            'required': True,
            'options': ['Yes', 'No']
        },
        {
            'id': 'pale_skin',
            'type': 'choice',
            'text': 'Have you noticed pale skin, lips, or nail beds?',
            'required': True,
            'options': ['Yes', 'No']
        },
        {
            'id': 'breathlessness',
            'type': 'choice',
            'text': 'Do you feel short of breath during normal activities?',
            'required': True,
            'options': ['Yes', 'No']
        },
        {
            'id': 'low_iron_diet',
            'type': 'choice',
            'text': 'Is your diet low in iron-rich foods (meat, beans, leafy greens)?',
            'required': True,
            'options': ['Yes', 'No']
        },
        {
            'id': 'cold_hands',
            'type': 'choice',
            'text': 'Do you often have cold hands or feet?',
            'required': False,
            'options': ['Yes', 'No']
        },
        {
            'id': 'headaches',
            'type': 'choice',
            'text': 'Do you experience frequent headaches?',
            'required': False,
            'options': ['Yes', 'No']
        },
        {
            'id': 'weakness',
            'type': 'choice',
            'text': 'Do you feel muscle weakness?',
            'required': False,
            'options': ['Yes', 'No']
        }
    ]
    
    return questions