Update app.py

app.py

@@ -133,10 +133,26 @@ def preprocess_image(image):
     return transform(image).unsqueeze(0)
 
 
-def get_akc_breeds_link(breed):
-    base_url = "https://www.akc.org/dog-breeds/"
-    breed_url = breed.lower().replace('_', '-')
-    return f"{base_url}{breed_url}/"
+def get_akc_breeds_link(breed: str) -> str:
+    """Generate AKC breed page URL with intelligent name handling"""
+    # 基本的字符串處理
+    breed_name = breed.lower()
+    
+    # 處理常見的名稱格式
+    breed_name = breed_name.replace('_', '-')
+    breed_name = breed_name.replace("'", '')
+    breed_name = breed_name.replace(" ", '-')
+    
+    # 處理特殊情況
+    special_cases = {
+        'mexican-hairless': 'xoloitzcuintli',
+        'brabancon-griffon': 'brussels-griffon',
+        'bull-mastiff': 'bullmastiff',
+        'walker-hound': 'treeing-walker-coonhound'
+    }
+    
+    breed_name = special_cases.get(breed_name, breed_name)
+    return f"https://www.akc.org/dog-breeds/{breed_name}/"
 
 
 async def predict_single_dog(image):
@@ -424,7 +440,6 @@ async def predict(image):
                                 <span class="value">{description['Lifespan']}</span>
                             </div>
                         </div>
-
                         <h2 class="section-title">
                             <span class="icon">🐕</span> TEMPERAMENT & PERSONALITY
                         </h2>
@@ -440,7 +455,6 @@ async def predict(image):
                                 </span>
                             </span>
                         </div>
-
                         <h2 class="section-title">
                             <span class="icon">💪</span> CARE REQUIREMENTS
                         </h2>
@@ -491,7 +505,6 @@ async def predict(image):
                                 <span class="value">{description['Care Level']}</span>
                             </div>
                         </div>
-
                         <h2 class="section-title">
                             <span class="icon">👨‍👩‍👧‍👦</span> FAMILY COMPATIBILITY
                         </h2>
@@ -511,14 +524,12 @@ async def predict(image):
                                 <span class="value">{description['Good with Children']}</span>
                             </div>
                         </div>
-
                         <h2 class="section-title">
                             <span class="icon">📝</span> DESCRIPTION
                         </h2>
                         <div class="description-section">
                             <p>{description.get('Description', '')}</p>
                         </div>
-
                         <div class="action-section">
                             <a href="{get_akc_breeds_link(breed)}" target="_blank" class="akc-button">
                                 <span class="icon">🌐</span>
@@ -583,6 +594,7 @@ async def predict(image):
         return error_msg, None, None
 
 
+
 def show_details_html(choice, previous_output, initial_state):
     if not choice:
         return previous_output, gr.update(visible=True), initial_state
@@ -804,7 +816,7 @@ def main():
 
         # Footer
         gr.HTML('''
-            For more details on this project and other work, feel free to visit my GitHub 
+            For more details on this project and other work, feel free to visit my GitHub
             <a href="https://github.com/Eric-Chung-0511/Learning-Record/tree/main/Data%20Science%20Projects/Dog_Breed_Classifier">
                 Dog Breed Classifier
             </a>
@@ -814,4 +826,4 @@ def main():
 
 if __name__ == "__main__":
     iface = main()
-    iface.launch(share=True) 
+    iface.launch(share=True, debug=True)