Update breed_recommendation.py

breed_recommendation.py

@@ -64,6 +64,7 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
 
             with gr.Tab("Find by Description"):
                 description_input, description_search_btn, description_output = create_description_search_tab()
+
     
         def on_find_match_click(*args):
             try:
@@ -117,6 +118,9 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
 
         def on_description_search(description: str):
             try:
+                loading_info.update(visible=True)
+                description_output.update(visible=False)
+                
                 matcher = SmartBreedMatcher(dog_data)
                 breed_recommendations = matcher.match_user_preference(description, top_n=10)
                 
@@ -194,10 +198,16 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
                             print(f"Warning: Ranking inconsistency detected!")
                             print(f"#{rec['rank']-1} score: {prev_score:.4f}")
                             print(f"#{rec['rank']} score: {rec['final_score']:.4f}")
+
+                loading_info.update(visible=False)
+                description_output.update(visible=True)
                 
                 return format_recommendation_html(final_recommendations)
                 
             except Exception as e:
+                loading_info.update(visible=False)
+                description_output.update(visible=True)
+                
                 import traceback
                 error_details = traceback.format_exc()
                 print(f"Detailed error:\n{error_details}")
@@ -234,5 +244,6 @@ def create_recommendation_tab(UserPreferences, get_breed_recommendations, format
         'recommendation_output': recommendation_output,
         'description_input': description_input,
         'description_search_btn': description_search_btn,
-        'description_output': description_output
+        'description_output': description_output,
+        'loading_info': loading_info
     }