Update scoring_calculation_system.py

scoring_calculation_system.py

@@ -208,102 +208,201 @@ def calculate_additional_factors(breed_info: dict, user_prefs: 'UserPreferences'
 
 @staticmethod
 def calculate_family_safety_score(breed_info: dict, children_age: str) -> float:
-    """
-    計算品種與家庭/兒童的安全相容性分數，作為calculate_compatibility_score的一部分
-    
-    參數:
-    breed_info (dict): 品種資訊
-    children_age (str): 兒童年齡組別 ('toddler', 'school_age', 'teenager')
-    
-    返回:
-    float: 0.2-0.95之間的安全分數
-    """
     temperament = breed_info.get('Temperament', '').lower()
     size = breed_info.get('Size', 'Medium')
     
-    # 基礎安全分數（根據體型）
+    # 基礎安全分數必須根據孩童年齡有所不同
     base_safety_scores = {
-        "Small": 0.80,     # 從 0.85 降至 0.80
-        "Medium": 0.65,    # 從 0.75 降至 0.65
-        "Large": 0.50,     # 從 0.65 降至 0.50
-        "Giant": 0.40      # 從 0.55 降至 0.40
+        'toddler': {
+            "Small": 0.85,     # 幼童與小型犬相對安全
+            "Medium": 0.60,    # 中型犬需要更多注意
+            "Large": 0.40,     # 大型犬風險較高
+            "Giant": 0.30      # 巨型犬風險最高
+        },
+        'school_age': {
+            "Small": 0.90,     # 學齡兒童與小型犬很合適
+            "Medium": 0.75,    # 中型犬可以接受
+            "Large": 0.55,     # 大型犬需要注意
+            "Giant": 0.45      # 巨型犬仍需謹慎
+        },
+        'teenager': {
+            "Small": 0.95,     # 青少年幾乎能應付所有小型犬
+            "Medium": 0.85,    # 中型犬很合適
+            "Large": 0.70,     # 大型犬可以考慮
+            "Giant": 0.60      # 巨型犬仍需小心
+        }
     }
-    safety_score = base_safety_scores.get(size, 0.60)
     
-    # 加強年齡相關的調整力度
-    age_factors = {
+    # 根據孩童年齡選擇對應的基礎分數
+    safety_score = base_safety_scores[children_age][size]
+    
+    # 年齡特定的危險特徵評估
+    age_specific_dangerous_traits = {
         'toddler': {
-            'base_modifier': -0.25,  # 從 -0.15 降至 -0.25
-            'size_penalty': {
-                "Small": -0.10,      # 從 -0.05 降至 -0.10
-                "Medium": -0.20,     # 從 -0.10 降至 -0.20
-                "Large": -0.30,      # 從 -0.20 降至 -0.30
-                "Giant": -0.35       # 從 -0.25 降至 -0.35
-            }
+            'aggressive': -0.40,    # 幼童最危險
+            'territorial': -0.35,
+            'protective': -0.30,
+            'nervous': -0.30,
+            'dominant': -0.25,
+            'energetic': -0.20      # 過度活潑對幼童也是風險
         },
         'school_age': {
-            'base_modifier': -0.15,  # 從 -0.08 降至 -0.15
-            'size_penalty': {
-                "Small": -0.05,
-                "Medium": -0.10,
-                "Large": -0.20,
-                "Giant": -0.25
-            }
+            'aggressive': -0.30,
+            'territorial': -0.25,
+            'protective': -0.20,
+            'nervous': -0.20,
+            'dominant': -0.15,
+            'energetic': -0.10
         },
         'teenager': {
-            'base_modifier': -0.08,  # 從 -0.05 降至 -0.08
-            'size_penalty': {
-                "Small": -0.02,
-                "Medium": -0.05,
-                "Large": -0.10,
-                "Giant": -0.15
-            }
+            'aggressive': -0.20,
+            'territorial': -0.15,
+            'protective': -0.10,
+            'nervous': -0.15,
+            'dominant': -0.10,
+            'energetic': -0.05
         }
     }
     
-    # 加強對危險特徵的評估
-    dangerous_traits = {
-        'aggressive': -0.35,      # 從 -0.25 加重到 -0.35
-        'territorial': -0.30,     # 從 -0.20 加重到 -0.30
-        'protective': -0.25,      # 從 -0.15 加重到 -0.25
-        'nervous': -0.25,         # 從 -0.15 加重到 -0.25
-        'dominant': -0.20,        # 從 -0.15 加重到 -0.20
-        'strong-willed': -0.18,   # 從 -0.12 加重到 -0.18
-        'independent': -0.15,     # 從 -0.10 加重到 -0.15
-        'energetic': -0.12       # 從 -0.08 加重到 -0.12
+    # 套用年齡特定的特徵評估
+    for trait, penalty in age_specific_dangerous_traits[children_age].items():
+        if trait in temperament:
+            safety_score += penalty
+    
+    # 正面特徵評估（根據年齡調整獎勵程度）
+    positive_traits_by_age = {
+        'toddler': {
+            'gentle': 0.15,
+            'patient': 0.15,
+            'calm': 0.12,
+            'tolerant': 0.12
+        },
+        'school_age': {
+            'gentle': 0.12,
+            'patient': 0.12,
+            'playful': 0.10,
+            'friendly': 0.10
+        },
+        'teenager': {
+            'friendly': 0.10,
+            'playful': 0.10,
+            'adaptable': 0.08,
+            'trainable': 0.08
+        }
     }
-
-    # 特殊風險評估加重
-    if 'history of' in breed_info.get('Description', '').lower():
-        safety_score -= 0.25      # 從 -0.15 加重到 -0.25
-    if 'requires experienced' in breed_info.get('Description', '').lower():
-        safety_score -= 0.20      # 從 -0.10 加重到 -0.20
     
-    # 計算特徵分數
-    for trait, bonus in positive_traits.items():
+    # 套用正面特徵評估
+    for trait, bonus in positive_traits_by_age[children_age].items():
         if trait in temperament:
-            safety_score += bonus * 0.8  # 降低正面特徵的影響力
-            
-    for trait, penalty in dangerous_traits.items():
-        if trait in temperament:
-            # 對幼童加重懲罰
-            if children_age == 'toddler':
-                safety_score += penalty * 1.3
-            # 對青少年略微減輕懲罰
-            elif children_age == 'teenager':
-                safety_score += penalty * 0.8
-            else:
-                safety_score += penalty
+            safety_score += bonus
     
-    # 特殊風險評估
+    # 特殊風險評估（對所有年齡都很重要）
     description = breed_info.get('Description', '').lower()
     if 'history of' in description:
-        safety_score -= 0.15
+        safety_score -= 0.25
     if 'requires experienced' in description:
-        safety_score -= 0.10
-    
-    # 將分數限制在合理範圍內
+        safety_score -= 0.15
+        
+    # 確保分數在合理範圍內
     return max(0.2, min(0.95, safety_score))
+    
+# def calculate_family_safety_score(breed_info: dict, children_age: str) -> float:
+#     """
+#     計算品種與家庭/兒童的安全相容性分數，作為calculate_compatibility_score的一部分
+    
+#     參數:
+#     breed_info (dict): 品種資訊
+#     children_age (str): 兒童年齡組別 ('toddler', 'school_age', 'teenager')
+    
+#     返回:
+#     float: 0.2-0.95之間的安全分數
+#     """
+#     temperament = breed_info.get('Temperament', '').lower()
+#     size = breed_info.get('Size', 'Medium')
+    
+#     # 基礎安全分數（根據體型）
+#     base_safety_scores = {
+#         "Small": 0.80,     # 從 0.85 降至 0.80
+#         "Medium": 0.65,    # 從 0.75 降至 0.65
+#         "Large": 0.50,     # 從 0.65 降至 0.50
+#         "Giant": 0.40      # 從 0.55 降至 0.40
+#     }
+#     safety_score = base_safety_scores.get(size, 0.60)
+    
+#     # 加強年齡相關的調整力度
+#     age_factors = {
+#         'toddler': {
+#             'base_modifier': -0.25,  # 從 -0.15 降至 -0.25
+#             'size_penalty': {
+#                 "Small": -0.10,      # 從 -0.05 降至 -0.10
+#                 "Medium": -0.20,     # 從 -0.10 降至 -0.20
+#                 "Large": -0.30,      # 從 -0.20 降至 -0.30
+#                 "Giant": -0.35       # 從 -0.25 降至 -0.35
+#             }
+#         },
+#         'school_age': {
+#             'base_modifier': -0.15,  # 從 -0.08 降至 -0.15
+#             'size_penalty': {
+#                 "Small": -0.05,
+#                 "Medium": -0.10,
+#                 "Large": -0.20,
+#                 "Giant": -0.25
+#             }
+#         },
+#         'teenager': {
+#             'base_modifier': -0.08,  # 從 -0.05 降至 -0.08
+#             'size_penalty': {
+#                 "Small": -0.02,
+#                 "Medium": -0.05,
+#                 "Large": -0.10,
+#                 "Giant": -0.15
+#             }
+#         }
+#     }
+    
+#     # 加強對危險特徵的評估
+#     dangerous_traits = {
+#         'aggressive': -0.35,      # 從 -0.25 加重到 -0.35
+#         'territorial': -0.30,     # 從 -0.20 加重到 -0.30
+#         'protective': -0.25,      # 從 -0.15 加重到 -0.25
+#         'nervous': -0.25,         # 從 -0.15 加重到 -0.25
+#         'dominant': -0.20,        # 從 -0.15 加重到 -0.20
+#         'strong-willed': -0.18,   # 從 -0.12 加重到 -0.18
+#         'independent': -0.15,     # 從 -0.10 加重到 -0.15
+#         'energetic': -0.12       # 從 -0.08 加重到 -0.12
+#     }
+
+#     # 特殊風險評估加重
+#     if 'history of' in breed_info.get('Description', '').lower():
+#         safety_score -= 0.25      # 從 -0.15 加重到 -0.25
+#     if 'requires experienced' in breed_info.get('Description', '').lower():
+#         safety_score -= 0.20      # 從 -0.10 加重到 -0.20
+    
+#     # 計算特徵分數
+#     for trait, bonus in positive_traits.items():
+#         if trait in temperament:
+#             safety_score += bonus * 0.8  # 降低正面特徵的影響力
+            
+#     for trait, penalty in dangerous_traits.items():
+#         if trait in temperament:
+#             # 對幼童加重懲罰
+#             if children_age == 'toddler':
+#                 safety_score += penalty * 1.3
+#             # 對青少年略微減輕懲罰
+#             elif children_age == 'teenager':
+#                 safety_score += penalty * 0.8
+#             else:
+#                 safety_score += penalty
+    
+#     # 特殊風險評估
+#     description = breed_info.get('Description', '').lower()
+#     if 'history of' in description:
+#         safety_score -= 0.15
+#     if 'requires experienced' in description:
+#         safety_score -= 0.10
+    
+#     # 將分數限制在合理範圍內
+#     return max(0.2, min(0.95, safety_score))
 
 
 def calculate_compatibility_score(breed_info: dict, user_prefs: UserPreferences) -> dict: