Update app.py

app.py

@@ -233,34 +233,16 @@ def predict(image):
             if top1_prob >= 0.5:
                 predicted = topk_indices[0][0]
                 breed = dog_breeds[predicted.item()]
-                description = get_dog_description(breed)
-                akc_link = get_akc_breeds_link()
-
-                if isinstance(description, dict):
-                    description_str = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])
-                else:
-                    description_str = description
-
-                # 添加AKC連結
-                description_str += f"\n\n**Want to learn more about dog breeds?** [Visit the AKC dog breeds page]({akc_link}) and search for {breed} to find detailed information."
-
-                # 添加免責聲明
-                disclaimer = ("\n\n*Disclaimer: The external link provided leads to the American Kennel Club (AKC) dog breeds page. "
-                              "You may need to search for the specific breed on that page. "
-                              "I am not responsible for the content on external sites. "
-                              "Please refer to the AKC's terms of use and privacy policy.*")
-                description_str += disclaimer
-
-                return description_str
+                return get_breed_info(breed)
 
             else:
-                # 信心不足50%，返回top 3的預測結果並附加連結
+                # 信心不足50%，返回top 3的預測結果，並且當點擊時能查看詳細資訊
                 topk_breeds = [dog_breeds[idx.item()] for idx in topk_indices[0]]
                 topk_probs_percent = [f"{prob.item() * 100:.2f}%" for prob in topk_probs[0]]
 
-                # 構建每個品種的連結和預測機率
+                # 提供每個品種的回調函數
                 topk_results = "\n\n".join(
-                    [f"**{i+1}. [{breed}](https://www.akc.org/dog-breeds/{quote(breed)})** ({prob} confidence)"
+                    [f"**{i+1}. [Click here to view more about {breed}](#) ({prob} confidence)**"
                      for i, (breed, prob) in enumerate(zip(topk_breeds, topk_probs_percent))]
                 )
 
@@ -273,9 +255,35 @@ def predict(image):
                 )
 
                 return explanation
+
     except Exception as e:
         return f"An error occurred: {e}"
 
+def get_breed_info(breed):
+    """
+    返回指定狗品種的詳細資訊，類似於 >=50% 信心時的結果。
+    """
+    description = get_dog_description(breed)
+    akc_link = get_akc_breeds_link()
+
+    if isinstance(description, dict):
+        description_str = "\n\n".join([f"**{key}**: {value}" for key, value in description.items()])
+    else:
+        description_str = description
+
+    # 添加AKC連結
+    description_str += f"\n\n**Want to learn more about dog breeds?** [Visit the AKC dog breeds page]({akc_link}) and search for {breed} to find detailed information."
+
+    # 添加免責聲明
+    disclaimer = ("\n\n*Disclaimer: The external link provided leads to the American Kennel Club (AKC) dog breeds page. "
+                  "You may need to search for the specific breed on that page. "
+                  "I am not responsible for the content on external sites. "
+                  "Please refer to the AKC's terms of use and privacy policy.*")
+    description_str += disclaimer
+
+    return description_str
+
+
 iface = gr.Interface(
     fn=predict,
     inputs=gr.Image(label="Upload a dog image", type="numpy"),