src/analyzer.py

import json
from typing import Dict, List, Optional

class MedicalKnowledgeBase:
    def __init__(self, db_path: str = "medical_db.json"):
        self.db_path = db_path
        self.data = self._load_db()

    def _load_db(self) -> Dict:
        try:
            with open(self.db_path, 'r', encoding='utf-8') as f:
                return json.load(f)
        except FileNotFoundError:
            print(f"خطا: فایل {self.db_path} پیدا نشد.")
            return {}

    def get_test_info(self, test_code: str):
        return self.data.get(test_code.upper())

class BloodAnalyzer:
    def __init__(self, knowledge_base: MedicalKnowledgeBase):
        self.kb = knowledge_base

    def analyze_single_test(self, test_code: str, value: float, gender: str = "male", age: int = None) -> Dict:
        """تک تست را تحلیل می‌کند. سن نیز برای تشخیص کودک استفاده می‌شود."""
        test_info = self.kb.get_test_info(test_code)
        
        if not test_info:
            return {
                "name_en": test_code, "name_fa": test_code,
                "value": value, "unit": "Unknown",
                "min_ref": 0, "max_ref": 0,
                "status": "UNKNOWN", "message": "این آزمایش در پایگاه داده ما وجود ندارد."
            }

        name_fa = test_info["name_fa"]
        unit = test_info["unit"]
        ranges = test_info["ranges"]
        
        # انتخاب رنج: کودک > جنسیت > پیش‌فرض
        if age and age < 18 and "child" in ranges:
            min_val, max_val = ranges["child"]
        elif gender == "male" and "male" in ranges:
            min_val, max_val = ranges["male"]
        elif gender == "female" and "female" in ranges:
            min_val, max_val = ranges["female"]
        else:
            min_val, max_val = ranges["default"]

        status = "NORMAL"
        message = "نتیجه طبیعی است."

        if value > max_val:
            status = "HIGH"
            message = test_info["conditions"]["high"]
        elif value < min_val:
            status = "LOW"
            message = test_info["conditions"]["low"]

        return {
            "name_en": test_code, "name_fa": name_fa,
            "value": value, "unit": unit,
            "min_ref": min_val, "max_ref": max_val,
            "status": status, "message": message
        }

    def generate_full_report(self, input_data: Dict[str, float], gender: str = "male", age: int = None):
        """تمام تست‌ها را تحلیل کرده و لیست برمی‌گرداند."""
        report = []
        for code, value in input_data.items():
            result = self.analyze_single_test(code, value, gender, age)
            report.append(result)
        return report