Update app.py

app.py

@@ -325,6 +325,7 @@ async def predict(image):
             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"
@@ -332,9 +333,10 @@ async def predict(image):
                     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]}")
 
-        # 多狗情境 - 使用 YOLO 檢測並處理多狗
+        # 多狗情境 - 使用 YOLO 偵測並處理多狗
         color_list = ['#FF0000', '#00FF00', '#0000FF', '#FFFF00', '#00FFFF', '#FF00FF', '#800080', '#FFA500']
         explanations = []
         visible_buttons = []
@@ -359,7 +361,7 @@ async def predict(image):
         return final_explanation, annotated_image, gr.update(visible=True, choices=visible_buttons), gr.update(visible=False), gr.update(visible=False)
 
     except Exception as e:
-        return f"An error occurred: {str(e)}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+        return f"An error occurred: {str(e)}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible(False))
 
 
 async def show_details(choice):
@@ -367,6 +369,7 @@ async def show_details(choice):
         return "Please select a breed to view details."
 
     try:
+        # 解析出用戶選擇的品種
         if "Dog" in choice:
             _, breed = choice.split(": ", 1)
         else:
@@ -378,6 +381,7 @@ async def show_details(choice):
 
 
 
+
 with gr.Blocks(css="""
     .container { max-width: 900px; margin: auto; padding: 20px; }
     .gr-box { border-radius: 15px; }