Update breed_recommendation.py

breed_recommendation.py

@@ -211,24 +211,21 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                 if not recommendations:
                     return "No matching breeds found. Please adjust your criteria."
         
-                history_results = []
+                # 創建標準化的結果列表
+                formatted_recommendations = []
                 for i, rec in enumerate(recommendations, 1):
-                    # 確保每個結果都有完整的數據結構
-                    result = {
+                    formatted_rec = {
                         'breed': rec['breed'],
-                        'rank': i,  # 使用循環索引確保排名連續
-                        'overall_score': rec.get('final_score', 0),  # 使用 get 方法提供默認值
-                        'final_score': rec.get('final_score', 0),  # 確保有 final_score
+                        'rank': i,
+                        'final_score': rec['final_score'],  # 直接使用，因為這是必需的
+                        'overall_score': rec['final_score'],  # 複製 final_score
                         'base_score': rec.get('base_score', 0),
                         'bonus_score': rec.get('bonus_score', 0),
                         'scores': rec.get('scores', {})
                     }
-                    history_results.append(result)
-        
-                # 添加調試輸出
-                print("History results to be saved:", history_results)
-        
-                # 保存搜索歷史時確保數據完整性
+                    formatted_recommendations.append(formatted_rec)
+                
+                # 使用相同的格式化數據來保存歷史記錄和生成 HTML
                 history_component.save_search(
                     user_preferences={
                         'living_space': args[0],
@@ -242,17 +239,14 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                         'children_age': args[8] if args[7] else None,
                         'noise_tolerance': args[9]
                     },
-                    results=history_results  # 使用處理過的結果
+                    results=formatted_recommendations  # 使用格式化後的結果
                 )
-        
-                # 返回推薦結果的 HTML 格式
-                html_result = format_recommendation_html(recommendations, is_description_search=False)
                 
-                return html_result
-        
+                # 使用相同的格式化數據生成 HTML
+                return format_recommendation_html(formatted_recommendations, is_description_search=False)
+                
             except Exception as e:
-                print(f"詳細錯誤: {str(e)}")
-                import traceback
+                print(f"Error details: {str(e)}")
                 print(traceback.format_exc())
                 return f"Error getting recommendations: {str(e)}"