Update breed_recommendation.py

breed_recommendation.py

@@ -449,6 +449,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                         elem_id="recommendation-output"
                     )
 
+        
         async def on_find_match_click(*args):
             try:
                 loading_messages = [
@@ -459,7 +460,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     "💭 Getting opinions from our experts..."
                 ]
         
-                # 顯示初始加載訊息
+                # 顯示加載訊息
                 for message in loading_messages:
                     yield gr.HTML("""
                         <div style="
@@ -483,29 +484,31 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                             </div>
                         </div>
                     """.format(message=message))
-                    await asyncio.sleep(1)  # 每個訊息顯示1秒
-
+                    await asyncio.sleep(1)
+        
+                # 處理用戶數據和獲取推薦
                 user_prefs = UserPreferences(
-                living_space=args[0],
-                yard_access=args[1],       
-                exercise_time=args[2],
-                exercise_type=args[3],      
-                grooming_commitment=args[4],
-                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[10]
-            )
-
+                    living_space=args[0],
+                    yard_access=args[1],       
+                    exercise_time=args[2],
+                    exercise_type=args[3],      
+                    grooming_commitment=args[4],
+                    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[10]
+                )
+        
                 recommendations = get_breed_recommendations(user_prefs, top_n=15)
-
+                
+                # 保存搜索歷史
                 history_results = [{
                     'breed': rec['breed'],
                     'rank': rec['rank'],
@@ -514,7 +517,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     'bonus_score': rec['bonus_score'],
                     'scores': rec['scores']
                 } for rec in recommendations]
-
+        
                 history_component.save_search(
                     user_preferences={
                         'living_space': args[0],
@@ -532,20 +535,15 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                     },
                     results=history_results
                 )
-
-                final_result = format_recommendation_html(recommendations, is_description_search=False)
-                return final_result
-
+        
+                # 使用 yield 而不是 return 來返回最終結果
+                yield gr.HTML(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 gr.HTML("""
-                    <div style="padding: 20px; background: #FEE2E2; border-radius: 8px; color: #991B1B;">
-                        <p style="margin: 0; font-weight: 500;">Sorry, we encountered an error while processing your request.</p>
-                        <p style="margin: 5px 0 0 0; font-size: 0.9em;">Please try again or contact support if the problem persists.</p>
-                    </div>
-                """)
+                yield gr.HTML("Error getting recommendations")
         
 
         get_recommendations_btn.click(