Update recommendation_html_format.py

recommendation_html_format.py

@@ -469,6 +469,12 @@ def get_breed_recommendations(user_prefs: UserPreferences, top_n: int = 10) -> L
             if not isinstance(breed_info, dict):
                 continue
 
+            if user_prefs.size_preference != "no_preference":
+                breed_size = breed_info.get('Size', '').lower()
+                user_size = user_prefs.size_preference.lower()
+                if breed_size != user_size:
+                    continue
+
             # 獲取噪音資訊
             noise_info = breed_noise_info.get(breed, {
                 "noise_notes": "Noise information not available",
@@ -532,8 +538,9 @@ def get_breed_recommendations(user_prefs: UserPreferences, top_n: int = 10) -> L
                 'info': breed_info,
                 'noise_info': noise_info  # 添加噪音資訊到推薦結果
             })
+            
         # 嚴格按照 final_score 排序
-        recommendations.sort(key=lambda x: (round(-x['final_score'], 4), x['breed'] ))  # 負號使其降序排列，並確保4位小數
+        recommendations.sort(key=lambda x: (round(-x['final_score'], 4), x['breed'] ))  # 負號降序排列
 
         # 選擇前N名並確保正確排序
         final_recommendations = []
@@ -541,19 +548,19 @@ def get_breed_recommendations(user_prefs: UserPreferences, top_n: int = 10) -> L
         rank = 1
 
         for rec in recommendations:
-            if len(final_recommendations) >= top_n:
+            # 如果沒有特定尺寸偏好且已達到所需數量，就停止
+            if user_prefs.size_preference == "no_preference" and len(final_recommendations) >= top_n:
                 break
-
+        
             current_score = rec['final_score']
-
+        
             # 確保分數遞減
             if last_score is not None and current_score > last_score:
                 continue
-
+        
             # 添加排名資訊
             rec['rank'] = rank
             final_recommendations.append(rec)
-
             last_score = current_score
             rank += 1