Update app.py

app.py

@@ -44,22 +44,31 @@ def predict(img):
     info = {
         "detected": len(result.boxes) > 0,
         "count": len(result.boxes),
-        "confidences": []
+        "detections": []  # 存储每个检测目标的详细信息
     }
     
     if info["detected"]:
-        # 获取每个检测框的置信度
+        # 获取每个检测框的信息
         for box in result.boxes:
             conf = float(box.conf[0])
-            info["confidences"].append(f"{conf:.2%}")
+            cls = int(box.cls[0])
+            cls_name = result.names[cls]  # 获取类别名称
+            
+            detection_info = {
+                "class": cls_name,
+                "confidence": f"{conf:.2%}"
+            }
+            info["detections"].append(detection_info)
     
     # 生成输出文本
     output_text = f"""检测结果:
 - 是否检测到目标: {'是' if info['detected'] else '否'}
 - 检测到的目标数量: {info['count']}"""
     
-    if info["confidences"]:
-        output_text += f"\n- 置信度: {', '.join(info['confidences'])}"
+    if info["detections"]:
+        output_text += "\n- 详细信息:"
+        for idx, det in enumerate(info["detections"], 1):
+            output_text += f"\n  目标 {idx}: {det['class']} (置信度: {det['confidence']})"
     
     return result_img, output_text