Update app.py

app.py

@@ -42,7 +42,12 @@ transform = transforms.Compose([
     transforms.ToTensor(),
 ])
 
-# Map confidence score to severity level
+# Define valid picklist values for Fault_Type__c and Severity__c
+# Replace these with your actual Salesforce picklist values
+VALID_FAULT_TYPES = ["Crack", "Rust", "Spalling", "Deformation", "Corrosion"]
+VALID_SEVERITIES = ["Minor", "Moderate", "Critical"]
+
+# Map confidence score to severity level ensuring valid picklist values
 def get_severity(score):
     if score >= 0.9:
         return "Critical"
@@ -51,10 +56,11 @@ def get_severity(score):
     else:
         return "Minor"
 
-# Simple defect type mapping (no COCO labels)
-# Here, just return a generic "Defect" label or you can customize per your need
+# Map defect to a valid picklist fault type
 def map_defect_type():
-    return "Defect"
+    # You can add logic here to decide which fault type to return
+    # For now, always return the first valid fault type
+    return VALID_FAULT_TYPES[0]
 
 # Upload annotated image to Salesforce as ContentVersion record
 def upload_image_to_salesforce(image, filename="detected_image.jpg", record_id=None):