Update app.py

app.py

@@ -348,10 +348,11 @@ async def predict(image):
             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:
-                breed = topk_breeds[0]
-                choices.append(f"Dog {i+1}: {breed}")
-                explanations.append(f"Dog {i+1}: **{breed}**")
+                description = get_dog_description(breed)
+                formatted_description = format_description(description, breed)
+                explanations.append(f"Dog {i+1}: {formatted_description}")
             elif top1_prob >= 0.2:
                 explanations.append(f"Dog {i+1}: Top 3 possible breeds:\n"
                                     f"1. **{topk_breeds[0]}** ({topk_probs_percent[0]} confidence)\n"
@@ -362,7 +363,7 @@ async def predict(image):
                 explanations.append(f"Dog {i+1}: The image is unclear or the breed is not in the dataset.")
 
         final_explanation = "\n\n".join(explanations)
-        return final_explanation, annotated_image, gr.update(visible=True, choices=choices), gr.update(visible=False)
+        return final_explanation, annotated_image, gr.update(visible=len(choices) > 0, choices=choices), gr.update(visible=False)
 
     except Exception as e:
         return f"An error occurred: {str(e)}", None, gr.update(visible=False), gr.update(visible=False)