Update app.py

app.py

@@ -541,18 +541,33 @@ async def predict(image):
                 description = get_dog_description(breed)
                 html_output += format_description_html(description, breed)
 
+            # elif combined_confidence >= 0.15:
+            #     html_output += f"<p>Top 3 possible breeds:</p><ul>"
+            #     for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3])):
+            #         html_output += f"<li><strong>{breed}</strong> ({prob} confidence)</li>"
+            #     html_output += "</ul>"
+            #     html_output += '<div class="breed-buttons">'
+            #     for breed in topk_breeds[:3]:
+            #         button_id = f"Dog {i+1}: More about {breed}"
+            #         html_output += f'<button class="breed-button" onclick="handle_button_click(\'{button_id}\')">{breed}</button>'
+            #         buttons.append(button_id)
+            #     html_output += '</div>'
+
             elif combined_confidence >= 0.15:
                 html_output += f"<p>Top 3 possible breeds:</p><ul>"
                 for j, (breed, prob) in enumerate(zip(topk_breeds[:3], topk_probs_percent[:3])):
                     html_output += f"<li><strong>{breed}</strong> ({prob} confidence)</li>"
                 html_output += "</ul>"
+                
+                # 確保不會重複加入已經顯示過的品種名
+                unique_breeds = set(topk_breeds[:3])  # 使用集合來去重
                 html_output += '<div class="breed-buttons">'
-                for breed in topk_breeds[:3]:
+                for breed in unique_breeds:
                     button_id = f"Dog {i+1}: More about {breed}"
                     html_output += f'<button class="breed-button" onclick="handle_button_click(\'{button_id}\')">{breed}</button>'
                     buttons.append(button_id)
                 html_output += '</div>'
-                
+  
             else:
                 html_output += "<p>The image is unclear or the breed is not in the dataset. Please upload a clearer image.</p>"