Update breed_recommendation.py

breed_recommendation.py

@@ -189,28 +189,38 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
 
                 # 從描述中提取用戶偏好
                 user_prefs = UserPreferences(
-                    living_space="apartment" if any(word in description.lower()
-                        for word in ["apartment", "flat", "condo"]) else "house_small",
-                    exercise_time=120 if any(word in description.lower()
-                        for word in ["active", "exercise", "running", "athletic", "high energy"]) else 60,
-                    grooming_commitment="high" if any(word in description.lower()
-                        for word in ["grooming", "brush", "maintain"]) else "medium",
-                    experience_level="experienced" if any(word in description.lower()
-                        for word in ["experienced", "trained", "professional"]) else "intermediate",
-                    has_children=any(word in description.lower()
-                        for word in ["children", "kids", "family", "child"]),
-                    noise_tolerance="low" if any(word in description.lower()
-                        for word in ["quiet", "peaceful", "silent"]) else "medium",
-                    space_for_play=any(word in description.lower()
-                        for word in ["yard", "garden", "outdoor", "space"]),
-                    other_pets=any(word in description.lower()
-                        for word in ["other pets", "cats", "dogs"]),
-                    climate="moderate",
-                    health_sensitivity="high" if any(word in description.lower()
-                        for word in ["health", "medical", "sensitive"]) else "medium",
-                    barking_acceptance="low" if any(word in description.lower()
-                        for word in ["quiet", "no barking"]) else None
-                )
+                living_space="apartment" if any(word in description.lower()
+                    for word in ["apartment", "flat", "condo"]) else "house_small",
+                yard_access="no_yard" if any(word in description.lower()
+                    for word in ["apartment", "flat", "condo"]) else "private_yard",
+                exercise_time=120 if any(word in description.lower()
+                    for word in ["active", "exercise", "running", "athletic", "high energy"]) else 60,
+                exercise_type="active_training" if any(word in description.lower()
+                    for word in ["training", "running", "jogging", "hiking"]) else "moderate_activity",
+                grooming_commitment="high" if any(word in description.lower()
+                    for word in ["grooming", "brush", "maintain"]) else "medium",
+                experience_level="experienced" if any(word in description.lower()
+                    for word in ["experienced", "trained", "professional"]) else "intermediate",
+                time_availability="flexible" if any(word in description.lower()
+                    for word in ["time", "available", "flexible", "home"]) else "moderate",
+                has_children=any(word in description.lower()
+                    for word in ["children", "kids", "family", "child"]),
+                children_age="school_age" if any(word in description.lower()
+                    for word in ["school", "elementary"]) else "teenager" if any(word in description.lower()
+                    for word in ["teen", "teenager"]) else "toddler" if any(word in description.lower()
+                    for word in ["baby", "toddler"]) else None,
+                noise_tolerance="low" if any(word in description.lower()
+                    for word in ["quiet", "peaceful", "silent"]) else "medium",
+                space_for_play=any(word in description.lower()
+                    for word in ["yard", "garden", "outdoor", "space"]),
+                other_pets=any(word in description.lower()
+                    for word in ["other pets", "cats", "dogs"]),
+                climate="moderate",
+                health_sensitivity="high" if any(word in description.lower()
+                    for word in ["health", "medical", "sensitive"]) else "medium",
+                barking_acceptance="low" if any(word in description.lower()
+                    for word in ["quiet", "no barking"]) else "medium"
+            )
 
                 final_recommendations = []