Update app.py

app.py

@@ -504,8 +504,54 @@ async def predict(image):
         elif len(dogs) == 1:
             return await process_single_dog(dogs[0][0])
         else:
-            # 多狗情境處理邏輯...
-            # (保持原有的多狗處理邏輯不變)
+             # 多狗情境
+            color_list = ['#FF0000', '#00FF00', '#0000FF', '#FFFF00', '#00FFFF', '#FF00FF', '#800080', '#FFA500']
+            explanations = []
+            buttons = []
+            annotated_image = image.copy()
+            draw = ImageDraw.Draw(annotated_image)
+            font = ImageFont.load_default()
+
+            for i, (cropped_image, confidence, box) in enumerate(dogs):
+                top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
+                color = color_list[i % len(color_list)]
+                draw.rectangle(box, outline=color, width=3)
+                draw.text((box[0], box[1]), f"Dog {i+1}", fill=color, font=font)
+
+                breed = topk_breeds[0]
+                if top1_prob >= 0.5:
+                    description = get_dog_description(breed)
+                    formatted_description = format_description(description, breed)
+                    explanations.append(f"Dog {i+1}: {formatted_description}")
+                elif top1_prob >= 0.2:
+                    dog_explanation = f"Dog {i+1}: Top 3 possible breeds:\n"
+                    dog_explanation += "\n".join([f"{j+1}. **{breed}** ({prob} confidence)" for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3]))])
+                    explanations.append(dog_explanation)
+                    buttons.extend([gr.update(visible=True, value=f"Dog {i+1}: More about {breed}") for breed in topk_breeds[:3]])
+                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)
+            if buttons:
+                final_explanation += "\n\nClick on a button to view more information about the breed."
+                initial_state = {
+                    "explanation": final_explanation,
+                    "buttons": buttons,
+                    "show_back": True
+                }
+                return (final_explanation, annotated_image, 
+                        buttons[0] if len(buttons) > 0 else gr.update(visible=False),
+                        buttons[1] if len(buttons) > 1 else gr.update(visible=False),
+                        buttons[2] if len(buttons) > 2 else gr.update(visible=False),
+                        gr.update(visible=True),
+                        initial_state)
+            else:
+                initial_state = {
+                    "explanation": final_explanation,
+                    "buttons": [],
+                    "show_back": False
+                }
+                return final_explanation, annotated_image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), initial_state
 
     except Exception as e:
         error_msg = f"An error occurred: {str(e)}"