Update app.py

app.py

@@ -12,6 +12,7 @@ from data_manager import get_dog_description
 from urllib.parse import quote
 from ultralytics import YOLO
 import asyncio
+import traceback
 
 
 # 下載YOLOv8預訓練模型
@@ -503,6 +504,14 @@ async def process_single_dog(image):
 #     if __name__ == "__main__":
 #         iface.launch()
 
+def safe_predict(image):
+    try:
+        return predict(image)
+    except Exception as e:
+        error_msg = f"An error occurred: {str(e)}\n\n{traceback.format_exc()}"
+        print(error_msg)  # 打印詳細錯誤信息到控制台
+        return error_msg, None, *[gr.update(visible=False) for _ in range(9)], gr.update(visible=False), None
+
 with gr.Blocks() as iface:
     gr.HTML("<h1 style='text-align: center;'>🐶 Dog Breed Classifier 🔍</h1>")
     gr.HTML("<p style='text-align: center;'>Upload a picture of a dog, and the model will predict its breed, provide detailed information, and include an extra information link!</p>")
@@ -514,7 +523,7 @@ with gr.Blocks() as iface:
     output = gr.Markdown(label="Prediction Results")
     
     with gr.Row():
-        buttons = [gr.Button(f"View More {i+1}", visible=False) for i in range(9)]  # 增加到9個按鈕
+        buttons = [gr.Button(f"View More {i+1}", visible=False) for i in range(9)]
     
     back_button = gr.Button("Back", visible=False)
     
@@ -534,7 +543,7 @@ with gr.Blocks() as iface:
                 gr.update(visible=state["show_back"]))
 
     input_image.change(
-        predict,
+        safe_predict,
         inputs=input_image,
         outputs=[output, output_image] + buttons + [back_button, initial_state]
     )