Update api_server.py

api_server.py

@@ -70,7 +70,11 @@ def predict():
 
     # Make a prediction using YOLO
     results = model(image_data)
-
+    
+    # 檢查 YOLO 是否返回了有效的結果
+    if results is None or len(results) == 0:
+        return jsonify({'error': 'No results from YOLO model'}), 400
+        
     saved_images = []
     
     # 儲存辨識後的圖片到指定資料夾
@@ -80,9 +84,13 @@ def predict():
         result.save(yolo_path)
         saved_images.append(yolo_path)
 
-        # 保存裁剪後的圖片
-        cropped_images = result.save_crop(YOLO_DIR)  # save_crop usually returns list of cropped image paths
-        saved_images.extend(cropped_images)  # 將裁剪後的圖片加入到返回列表
+        # 保存裁剪後的圖片（僅當 save_crop 返回有效的路徑時才加入）
+        cropped_images = result.save_crop(YOLO_DIR)  # 有些 YOLO 版本 save_crop 不返回值
+        if cropped_images:  # 確保不會對 None 進行迭代
+            if isinstance(cropped_images, list):  # 如果它返回一個列表
+                saved_images.extend(cropped_images)
+            else:
+                saved_images.append(cropped_images)
 
     end_time = time.time()
     inference_time = end_time - start_time