Update breed_recommendation.py

breed_recommendation.py

@@ -114,6 +114,12 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                         )
 
                     with gr.Column():
+                        size_preference = gr.Radio(
+                            choices=["no_preference", "small", "medium", "large", "giant"],
+                            label="Preference Dog Size",
+                            info="Select your preferred dog size - this will strongly filter the recommendations",
+                            value = "no_preference"
+                        )
                         experience_level = gr.Radio(
                             choices=["beginner", "intermediate", "advanced"],
                             label="Dog ownership experience",
@@ -172,16 +178,17 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     exercise_time=args[2],
                     exercise_type=args[3],      
                     grooming_commitment=args[4],
-                    experience_level=args[5],
-                    time_availability=args[6],   
-                    has_children=args[7],
-                    children_age=args[8] if args[7] else None,
-                    noise_tolerance=args[9],
+                    size_preference=args[5]
+                    experience_level=args[6],
+                    time_availability=args[7],   
+                    has_children=args[8],
+                    children_age=args[9] if args[8] else None,
+                    noise_tolerance=args[10],
                     space_for_play=True if args[0] != "apartment" else False,
                     other_pets=False,
                     climate="moderate",
                     health_sensitivity="medium",
-                    barking_acceptance=args[9]
+                    barking_acceptance=args[10]
                 )
 
                 recommendations = get_breed_recommendations(user_prefs, top_n=10)
@@ -229,6 +236,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                 exercise_time,
                 exercise_type,      
                 grooming_commitment,
+                size_preference,
                 experience_level,
                 time_availability, 
                 has_children,