Update breed_recommendation.py

breed_recommendation.py

@@ -125,8 +125,67 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     elem_id="recommendation-output"  # 添加唯一ID以便追蹤
                 )
 
+        # def on_find_match_click(*args):
+        #     try:
+        #         user_prefs = UserPreferences(
+        #             living_space=args[0],
+        #             yard_access=args[1],       
+        #             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],
+        #             space_for_play=True if args[0] != "apartment" else False,
+        #             other_pets=False,
+        #             climate="moderate",
+        #             health_sensitivity="medium",
+        #             barking_acceptance=args[9]
+        #         )
+
+        #         recommendations = get_breed_recommendations(user_prefs, top_n=10)
+
+        #         history_results = [{
+        #             'breed': rec['breed'],
+        #             'rank': rec['rank'],
+        #             'overall_score': rec['final_score'],
+        #             'base_score': rec['base_score'],
+        #             'bonus_score': rec['bonus_score'],
+        #             'scores': rec['scores']
+        #         } for rec in recommendations]
+
+        #         history_component.save_search(
+        #             user_preferences={
+        #                 'living_space': args[0],
+        #                 'yard_access': args[1],
+        #                 '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],
+        #                 'search_type': 'Criteria'  
+        #             },
+        #             results=history_results
+        #         )
+
+        #         return format_recommendation_html(recommendations, is_description_search=False)
+
+        #     except Exception as e:
+        #         print(f"Error in find match: {str(e)}")
+        #         import traceback
+        #         print(traceback.format_exc())
+        #         return "Error getting recommendations"
+
         def on_find_match_click(*args):
             try:
+                # 添加數據驗證和日誌
+                print("Received args:", args)  # 檢查接收到的參數
+                
                 user_prefs = UserPreferences(
                     living_space=args[0],
                     yard_access=args[1],       
@@ -144,19 +203,33 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     health_sensitivity="medium",
                     barking_acceptance=args[9]
                 )
-
+                
+                # 添加日誌來檢查 user_prefs 是否正確建立
+                print("UserPreferences created:", vars(user_prefs))
+        
+                # 檢查推薦結果
                 recommendations = get_breed_recommendations(user_prefs, top_n=10)
-
-                history_results = [{
-                    'breed': rec['breed'],
-                    'rank': rec['rank'],
-                    'overall_score': rec['final_score'],
-                    'base_score': rec['base_score'],
-                    'bonus_score': rec['bonus_score'],
-                    'scores': rec['scores']
-                } for rec in recommendations]
-
-                history_component.save_search(
+                print("Recommendations received:", recommendations)
+        
+                # 確保 recommendations 不為空
+                if not recommendations:
+                    return "No matching breeds found. Please adjust your criteria."
+        
+                # 驗證推薦結果的格式
+                history_results = []
+                for rec in recommendations:
+                    result = {
+                        'breed': rec['breed'],
+                        'rank': rec.get('rank', 0),  # 使用 get 方法避免 KeyError
+                        'overall_score': rec['final_score'],
+                        'base_score': rec.get('base_score', 0),
+                        'bonus_score': rec.get('bonus_score', 0),
+                        'scores': rec.get('scores', {})
+                    }
+                    history_results.append(result)
+        
+                # 檢查歷史記錄是否成功保存
+                save_success = history_component.save_search(
                     user_preferences={
                         'living_space': args[0],
                         'yard_access': args[1],
@@ -168,18 +241,22 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                         'has_children': args[7],
                         'children_age': args[8] if args[7] else None,
                         'noise_tolerance': args[9],
-                        'search_type': 'Criteria'  
+                        'search_type': 'Criteria'
                     },
                     results=history_results
                 )
-
-                return format_recommendation_html(recommendations, is_description_search=False)
-
+                print("History save success:", save_success)
+        
+                # 產生並返回 HTML
+                html_result = format_recommendation_html(recommendations, is_description_search=False)
+                print("HTML generation completed")
+                return html_result
+        
             except Exception as e:
-                print(f"Error in find match: {str(e)}")
+                print(f"Detailed error in find match: {str(e)}")
                 import traceback
                 print(traceback.format_exc())
-                return "Error getting recommendations"
+                return f"Error getting recommendations: {str(e)}"  # 返回更具體的錯誤訊息
 
         get_recommendations_btn.click(
             fn=on_find_match_click,