Update app.py

app.py

@@ -541,43 +541,22 @@ 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>"
                 html_output += '<div class="breed-buttons">'
-                
-                # 使用集合去重，防止重複顯示品種
-                unique_breeds = set(topk_breeds[:3])
-                for breed in unique_breeds:
+                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>'
-
   
             else:
                 html_output += "<p>The image is unclear or the breed is not in the dataset. Please upload a clearer image.</p>"
             
-            html_output += '</div>'
+            #html_output += '</div>'
 
         if buttons:
             html_output += """