Update app.py

app.py

@@ -250,7 +250,6 @@ def get_akc_breeds_link():
 #     iface.launch()
 
 
-
 # Update the format_description to handle descriptions more cleanly
 def format_description(description, breed):
     if isinstance(description, dict):
@@ -314,22 +313,28 @@ async def predict(image):
         # 偵測圖片中的多個狗
         dogs = await detect_multiple_dogs(image)
 
+        # 單一狗情況處理
         if len(dogs) == 0:
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
             if top1_prob < 0.2:
                 return "The image is unclear or the breed is not in the dataset. Please upload a clearer image of a dog.", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
-            breed = topk_breeds[0]
-            description = get_dog_description(breed)
-            formatted_description = format_description(description, breed)
-            return formatted_description, image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
-        # 處理單一狗情況
-        if len(dogs) == 1:
-            top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
             breed = topk_breeds[0]
             description = get_dog_description(breed)
             formatted_description = format_description(description, breed)
-            return formatted_description, image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+
+            # 處理高置信度與低置信度情況
+            if top1_prob >= 0.5:
+                return formatted_description, image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+            else:
+                explanation = (
+                    f"The model couldn't confidently identify the breed. Here are the top 3 possible breeds:\n\n"
+                    f"1. **{topk_breeds[0]}** ({topk_probs_percent[0]} confidence)\n"
+                    f"2. **{topk_breeds[1]}** ({topk_probs_percent[1]} confidence)\n"
+                    f"3. **{topk_breeds[2]}** ({topk_probs_percent[2]} confidence)\n\n"
+                    "Click on a button to view more information about the breed."
+                )
+                return explanation, gr.update(visible=True, value=f"More about {topk_breeds[0]}"), gr.update(visible=True, value=f"More about {topk_breeds[1]}"), gr.update(visible=True, value=f"More about {topk_breeds[2]}")
 
         # 多狗處理
         color_list = ['#FF0000', '#00FF00', '#0000FF', '#FFFF00', '#00FFFF', '#FF00FF', '#800080', '#FFA500']
@@ -387,8 +392,6 @@ async def show_details(choice):
         return f"An error occurred while showing details: {e}"
 
 
-
-
 with gr.Blocks(css="""
     .container { max-width: 900px; margin: auto; padding: 20px; }
     .gr-box { border-radius: 15px; }