fixes

app.py

@@ -30,21 +30,45 @@ def read_root():
 
 @app.post("/detect")
 async def detect_damage(file: UploadFile = File(...)):
+    # 1. Get image and run the model (same as before)
     contents = await file.read()
     image = Image.open(io.BytesIO(contents)).convert("RGB")
     results = model(image)
 
-    output = []
+    # 2. Process the results to determine unique conditions
+
+    # Use a set to automatically handle duplicates (e.g., "broken" is only added once)
+    conditions = set()
+
+    # Define which detections count as "broken" or "dirty"
+    broken_classes = {'scratch', 'rust', 'dent'}
+    dirty_classes = {'dirt'}
+
+    # Loop through all detected objects
     for result in results:
-        boxes = result.boxes
-        for box in boxes:
-            class_id = int(box.cls[0])
-            class_name = model.names[class_id]
+        for box in result.boxes:
+            # Get the name and confidence of the detected object
+            class_name = model.names[int(box.cls[0])]
             confidence = float(box.conf[0])
+
+            # Only consider detections with high confidence
             if confidence > 0.5:
-                output.append({'name': class_name, 'confidence': confidence})
+                if class_name in broken_classes:
+                    conditions.add("Broken")
+                elif class_name in dirty_classes:
+                    conditions.add("Dirty")
+
+    # 3. Prepare the final output
+
+    # If the conditions set is empty after checking all boxes, the car is clean
+    if not conditions:
+        final_output = ["No damage detected."]
+    else:
+        # Convert the set to a list for the JSON response
+        final_output = list(conditions)
 
-    return {"detections": output}
+    # Return the simple list of conditions
+    return {"conditions": final_output}
 
 
 # --- DEBUGGING STEP 2 ---