Update app.py

app.py

@@ -413,7 +413,12 @@ with gr.Blocks(css="""
 
     async def safe_predict(image):
         try:
-            return await predict(image)
+            # 從 predict 函數返回 4 個結果
+            prediction, annotated_img, buttons_options, breed_detail = await predict(image)
+            if buttons_options:  # 如果有多狗結果，顯示選項按鈕
+                return prediction, annotated_img, gr.update(visible=True, choices=buttons_options), breed_detail
+            else:  # 單狗預測時隱藏按鈕
+                return prediction, annotated_img, gr.update(visible=False), breed_detail
         except Exception as e:
             return str(e), None, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
@@ -434,6 +439,7 @@ with gr.Blocks(css="""
         inputs=input_image
     )
 
+
     gr.HTML('For more details on this project and other work, feel free to visit my GitHub <a href="https://github.com/Eric-Chung-0511/Learning-Record/tree/main/Data%20Science%20Projects/Dog_Breed_Classifier">Dog Breed Classifier</a>')
 
 if __name__ == "__main__":