Update app.py

app.py

@@ -434,7 +434,7 @@ async def predict(image):
             color = color_list[i % len(color_list)]
             draw.rectangle(box, outline=color, width=3)
             draw.text((box[0] + 5, box[1] + 5), f"Dog {i+1}", fill=color, font=font)
-
+    
             combined_confidence = detection_confidence * top1_prob
             html_output += f'<div class="dog-info" style="border-left: 5px solid {color};">'
             html_output += f'<h2>Dog {i+1}</h2>'
@@ -442,6 +442,7 @@ async def predict(image):
             if top1_prob >= 0.45:
                 breed = topk_breeds[0]
                 description = get_dog_description(breed)
+                html_output += f"<p><strong>{breed}</strong> ({top1_prob:.2%} confidence)</p>"
                 html_output += format_description_html(description, breed)
                 button_id = f"Dog {i+1}: More about {breed}"
                 html_output += f'<div class="breed-buttons"><button class="breed-button" onclick="handle_button_click(\'{button_id}\')">{breed}</button></div>'
@@ -575,7 +576,7 @@ with gr.Blocks() as iface:
     )
 
     breed_buttons.change(
-        show_details_html,
+        show_details_html,  # 確保這裡使用 show_details_html
         inputs=[breed_buttons, output, initial_state],
         outputs=[output, back_button, initial_state]
     )
@@ -585,7 +586,7 @@ with gr.Blocks() as iface:
         inputs=[initial_state],
         outputs=[output, output_image, breed_buttons, back_button, initial_state]
     )
-
+    
     gr.Examples(
         examples=['Border_Collie.jpg', 'Golden_Retriever.jpeg', 'Saint_Bernard.jpeg', 'French_Bulldog.jpeg', 'Samoyed.jpg'],
         inputs=input_image