Update app.py

app.py

@@ -300,7 +300,7 @@ async def detect_multiple_dogs(image):
         print(f"Error in detect_multiple_dogs: {e}")
         return []
 
-def predict(image):
+async def predict(image):
     if image is None:
         return "Please upload an image to start.", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
@@ -308,7 +308,7 @@ def predict(image):
         if isinstance(image, np.ndarray):
             image = Image.fromarray(image)
 
-        dogs = detect_multiple_dogs(image)
+        dogs = await detect_multiple_dogs(image)
         
         if len(dogs) == 0:
             return "No dogs detected. Please upload a clear image of a dog.", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
@@ -320,7 +320,7 @@ def predict(image):
         font = ImageFont.load_default()
 
         for i, (cropped_image, _, box) in enumerate(dogs, 1):
-            top1_prob, topk_breeds, topk_probs_percent = predict_single_dog(cropped_image)
+            top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
 
             draw.rectangle(box, outline="red", width=3)
             draw.text((box[0], box[1]), f"Dog {i}", fill="red", font=font)
@@ -328,11 +328,10 @@ def predict(image):
             if top1_prob >= 0.5:
                 breed = topk_breeds[0]
                 description = get_dog_description(breed)
-                explanations.append(f"Dog {i}: **{breed}**\n\n{format_description(description, breed)}")
+                explanations.append(f"Dog {i}: **{breed}**\n\n{format_description(description, breed, is_multi_dog=True, dog_number=i)}")
             else:
                 explanation = f"""
 Dog {i}: Detected with moderate confidence. 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]})
@@ -351,7 +350,7 @@ Dog {i}: Detected with moderate confidence. Here are the top 3 possible breeds:
     except Exception as e:
         return f"An error occurred: {e}", None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
-def show_details(choice):
+async def show_details(choice):
     if not choice:
         return "Please select a breed to view details."
     
@@ -362,7 +361,7 @@ def show_details(choice):
     except Exception as e:
         return f"An error occurred while showing details: {e}"
 
-# Gradio 界面設置
+# 修改 Gradio 界面設置
 with gr.Blocks(css="""
     .container { max-width: 900px; margin: auto; padding: 20px; }
     .gr-box { border-radius: 15px; }
@@ -390,8 +389,8 @@ with gr.Blocks(css="""
 
     breed_buttons.select(
         show_details,
-        inputs=[breed_details],
-        outputs=[breed_details]
+        inputs=breed_buttons,
+        outputs=breed_details
     )
 
     gr.Examples(