Update breed_recommendation.py

breed_recommendation.py

@@ -37,7 +37,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     </div>
                 """)
 
-                with gr.Row():
+                 with gr.Row():
                     with gr.Column():
                         living_space = gr.Radio(
                             choices=["apartment", "house_small", "house_large"],
@@ -46,6 +46,13 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                             value="apartment"
                         )
 
+                        yard_access = gr.Radio(
+                            choices=["no_yard", "shared_yard", "private_yard"],
+                            label="Yard Access Type",
+                            info="Available outdoor space",
+                            value="no_yard"
+                        )
+
                         exercise_time = gr.Slider(
                             minimum=0,
                             maximum=180,
@@ -54,6 +61,14 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                             info="Consider walks, play time, and training"
                         )
 
+                        exercise_type = gr.Radio(
+                            choices=["light_walks", "moderate_activity", "active_training"],
+                            label="Exercise Style",
+                            info="What kind of activities do you prefer?",
+                            value="moderate_activity"
+                        )
+
+
                         grooming_commitment = gr.Radio(
                             choices=["low", "medium", "high"],
                             label="Grooming commitment level",
@@ -69,17 +84,40 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                             value="beginner"
                         )
 
+                        time_availability = gr.Radio(
+                            choices=["limited", "moderate", "flexible"],
+                            label="Time Availability",
+                            info="Time available for dog care daily",
+                            value="moderate"
+                        )
+
                         has_children = gr.Checkbox(
                             label="Have children at home",
                             info="Helps recommend child-friendly breeds"
                         )
 
+                        children_age = gr.Radio(
+                            choices=["toddler", "school_age", "teenager"],
+                            label="Children's Age Group",
+                            info="Helps match with age-appropriate breeds",
+                            visible=False  # 默認隱藏，只在has_children=True時顯示
+                        )
+
                         noise_tolerance = gr.Radio(
                             choices=["low", "medium", "high"],
                             label="Noise tolerance level",
                             info="Some breeds are more vocal than others",
                             value="medium"
                         )
+                        
+                def update_children_age_visibility(has_children):
+                    return gr.update(visible=has_children)
+
+                has_children.change(
+                    fn=update_children_age_visibility,
+                    inputs=has_children,
+                    outputs=children_age
+                )
 
                 get_recommendations_btn = gr.Button("Find My Perfect Match! 🔍", variant="primary")
                 recommendation_output = gr.HTML(label="Breed Recommendations")
@@ -97,11 +135,12 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     experience_level=args[3],
                     has_children=args[4],
                     noise_tolerance=args[5],
+                    children_age=args[6] if args[4] else None,
                     space_for_play=True if args[0] != "apartment" else False,
                     other_pets=False,
                     climate="moderate",
-                    health_sensitivity="medium",  # 新增: 默認中等敏感度
-                    barking_acceptance=args[5]    # 使用 noise_tolerance 作為 barking_acceptance
+                    health_sensitivity="medium",  
+                    barking_acceptance=args[5]    
                 )
 
                 recommendations = get_breed_recommendations(user_prefs, top_n=10)
@@ -452,7 +491,8 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                 grooming_commitment,
                 experience_level,
                 has_children,
-                noise_tolerance
+                noise_tolerance,
+                children_age
             ],
             outputs=recommendation_output
         )