Update breed_comparison.py

breed_comparison.py

@@ -9,21 +9,19 @@ def create_comparison_tab(dog_breeds, get_dog_description, breed_noise_info, bre
     breed_display_mapping = { breed: breed.replace('_', ' ').title() for breed in dog_breeds}
     reversed_mapping = {v: k for k ,v in breed_display_mapping.items()}
 
-    def sort_key(x):
-        return x.lower()
-    sorted_display_names = sorted(breed_display_mapping.values(), key=sort_key) 
+    sorted_display_names = sorted(breed_display_mapping.values(), key=lambda x: x.lower()) 
     
     with gr.TabItem("Breed Comparison"):
         gr.HTML("<p style='text-align: center;'>Select two dog breeds to compare their characteristics and care requirements.</p>")
         
         with gr.Row():
             breed1_dropdown = gr.Dropdown(
-                choices=dog_breeds,
+                choices=sorted_display_names, 
                 label="Select First Breed",
                 value="Golden_Retriever"
             )
             breed2_dropdown = gr.Dropdown(
-                choices=dog_breeds,
+                choices=sorted_display_names,
                 label="Select Second Breed", 
                 value="Border_Collie"
             )
@@ -360,10 +358,12 @@ def create_comparison_tab(dog_breeds, get_dog_description, breed_noise_info, bre
             }
             """
 
-        def show_comparison(breed1, breed2):
+        def show_comparison(breed1_display, breed2_display):
             if not breed1 or not breed2:
                 return "Please select two breeds to compare"
 
+            breed1 = reverse_mapping[breed1_display]
+            breed2 = reverse_mapping[breed2_display]
             breed1_info = get_dog_description(breed1)
             breed2_info = get_dog_description(breed2)
             breed1_noise = breed_noise_info.get(breed1, {})
@@ -454,7 +454,7 @@ def create_comparison_tab(dog_breeds, get_dog_description, breed_noise_info, bre
 
                 return f"""
                 <div class="breed-column">
-                    <h2 class="section-title">🐕 {breed.replace('_', ' ')}</h2>
+                    <h2 class="section-title">🐕 {breed_display_mapping[breed]}</h2>
                     
                     <div class="info-section">
                         {create_info_item('Size', info['Size'], section_tooltips['Size'])}
@@ -510,7 +510,7 @@ def create_comparison_tab(dog_breeds, get_dog_description, breed_noise_info, bre
                     <a href="https://www.akc.org/dog-breeds/{breed.lower().replace('_', '-')}/" 
                     class="learn-more-btn"
                     target="_blank">
-                        🌐 Learn more about {breed.replace('_', ' ')} on AKC
+                        🌐 Learn more about {breed_display_mapping[breed]} on AKC
                     </a>
                 </div>
                 """