Update app.py

app.py

@@ -364,8 +364,8 @@ async def predict(image):
         font = ImageFont.load_default()
 
         dogs_info = []
-        buttons = []
-
+        all_buttons = []
+    
         for i, (cropped_image, detection_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)]
@@ -390,11 +390,12 @@ async def predict(image):
                     dog_info.append(f"<li><strong>{breed}</strong> ({prob:.2f}% confidence)</li>")
                 dog_info.append("</ul>")
         
+                # 將按鈕直接添加到每個狗狗的資訊中
                 dog_info.append('<div class="breed-buttons">')
                 for breed in topk_breeds[:3]:
                     button_id = f"Dog {i+1}: More about {breed}"
                     dog_info.append(f'<button class="breed-button" onclick="handle_button_click(\'{button_id}\')">{breed}</button>')
-                    buttons.append(button_id)
+                    all_buttons.append(button_id)
                 dog_info.append('</div>')
         
             else:
@@ -403,6 +404,8 @@ async def predict(image):
             dog_info.append('</div>')
             dogs_info.append(''.join(dog_info))
     
+        # 刪除：移除了在這裡生成重複品種名稱的代碼
+    
         html_output = f"""
         <style>
         .dog-info {{ border: 1px solid #ddd; margin-bottom: 20px; padding: 15px; border-radius: 5px; box-shadow: 0 2px 5px rgba(0,0,0,0.1); }}
@@ -413,7 +416,7 @@ async def predict(image):
         {''.join(dogs_info)}
         """
         
-        if buttons:
+        if all_buttons:
             html_output += """
             <script>
             function handle_button_click(button_id) {
@@ -427,14 +430,25 @@ async def predict(image):
             </script>
             """
             initial_state = {
-                "dogs_info": dogs_info,
-                "buttons": buttons,
+                "dogs_info": ''.join(dogs_info),
+                "buttons": all_buttons,
                 "show_back": True,
                 "image": annotated_image,
                 "is_multi_dog": len(dogs) > 1,
                 "html_output": html_output
             }
-            return html_output, annotated_image, gr.update(visible=True, choices=buttons), initial_state
+            return html_output, annotated_image, gr.update(visible=True, choices=all_buttons), initial_state
+        else:
+            initial_state = {
+                "dogs_info": ''.join(dogs_info),
+                "buttons": [],
+                "show_back": False,
+                "image": annotated_image,
+                "is_multi_dog": len(dogs) > 1,
+                "html_output": html_output
+            }
+            return html_output, annotated_image, gr.update(visible=False, choices=[]), initial_state
+
         else:
             initial_state = {
                 "dogs_info": dogs_info,