Update app.py

app.py

@@ -357,10 +357,10 @@ async def process_multiple_dogs_result(dogs, image):
         if top1_prob >= 0.5:
             breed = topk_breeds[0]
             description = get_dog_description(breed)
-            explanations.append(f"Dog {i}: **{breed}** ({top1_prob*100:.2f}% confidence)\n\n{format_description(description, breed, is_multi_dog=True, dog_number=i)}")
-        else:
+            explanations.append(f"Dog {i}: **{breed}**\n\n{format_description(description, breed, is_multi_dog=True, dog_number=i)}")
+        elif top1_prob >= 0.2:
             explanation = f"""
-Dog {i}: Detected with moderate confidence. Here are the top 3 possible breeds:
+Dog {i}: The model couldn't confidently identify the breed. Here are the top 3 possible breeds:
 1. **{topk_breeds[0]}** ({topk_probs_percent[0]})
 2. **{topk_breeds[1]}** ({topk_probs_percent[1]})
 3. **{topk_breeds[2]}** ({topk_probs_percent[2]})
@@ -368,13 +368,15 @@ Dog {i}: Detected with moderate confidence. Here are the top 3 possible breeds:
             explanations.append(explanation)
             for breed in topk_breeds:
                 buttons.append(f"More about Dog {i}: {breed}")
+        else:
+            explanations.append(f"Dog {i}: The image is unclear or the breed is not in the dataset. Please upload a clearer image of a dog.")
 
     final_explanation = "\n\n---\n\n".join(explanations)
     
-    if buttons:
-        return final_explanation, annotated_image, gr.update(visible=True, choices=buttons), gr.update(visible=False), gr.update(visible=False)
-    else:
-        return final_explanation, annotated_image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    if not explanations:
+        return "No dogs were confidently detected in the image. Please upload a clearer image of a dog.", annotated_image, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
+    
+    return final_explanation, annotated_image, gr.update(visible=bool(buttons), choices=buttons), gr.update(visible=False), gr.update(visible=False)
 
 async def process_single_dog_result(top1_prob, topk_breeds, topk_probs_percent, image, box):
     annotated_image = image.copy()