Update app.py

app.py

@@ -311,7 +311,7 @@ async def predict(image):
         # 偵測圖片中的多個狗
         dogs = await detect_multiple_dogs(image)
 
-        # 單一狗情況處理
+        # 處理單一狗情況 (如果 YOLO 沒有檢測到多隻狗)
         if len(dogs) == 0:
             top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(image)
             if top1_prob < 0.2:
@@ -364,15 +364,15 @@ Dog {i+1}: Detected with moderate confidence. Here are the top 3 possible breeds
 3. **{topk_breeds[2]}** ({topk_probs_percent[2]})
 """
                 explanations.append(explanation)
-                visible_buttons.append(gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[0]}"))
-                visible_buttons.append(gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[1]}"))
-                visible_buttons.append(gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[2]}"))
+                visible_buttons.extend([gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[0]}"), 
+                                        gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[1]}"), 
+                                        gr.update(visible=True, value=f"More about Dog {i+1}: {topk_breeds[2]}")])
             else:
                 explanations.append(f"Dog {i+1}: The image is unclear or the breed is not in the dataset.")
 
         final_explanation = "\n\n".join(explanations)
 
-        return final_explanation, annotated_image, visible_buttons
+        return final_explanation, annotated_image, gr.update(visible=True, choices=visible_buttons)
 
     except Exception as e:
         return f"An error occurred: {e}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)