| 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 |
