Update app.py

app.py

@@ -411,17 +411,18 @@ with gr.Blocks(css="""
     breed_buttons = gr.Radio([], label="Select breed for more details", visible=False)
     breed_details = gr.Markdown(label="Breed Details")
 
-    async def safe_predict(image):
+    async def safe_predict(image):   
         try:
             # 從 predict 函數返回的值數應該與此處匹配
-            prediction, annotated_img, buttons_options = await predict(image)
+            prediction, annotated_img, buttons_options, breed_details_visibility = await predict(image)
             if buttons_options:  # 如果有多狗結果，顯示選項按鈕
-                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), gr.update(visible=True)
+                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), breed_details_visibility
             else:  # 單狗預測時隱藏按鈕
-                return prediction, annotated_img, gr.update(visible=False), gr.update(visible=False)
+                return prediction, annotated_img, gr.update(visible=False), breed_details_visibility
         except Exception as e:
             return str(e), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
+
     input_image.change(
         safe_predict,
         inputs=input_image,