Update app.py

app.py

@@ -37,17 +37,8 @@ def mock_ai_model(image):
 
     # Feature Extraction and Milestone Detection (simulated)
     # In a real scenario, this would use a CNN model trained on construction images
-    milestones = [
-        "Foundation Completed",
-        "Structural Framework Started",
-        "Walls In Progress",
-        "Roofing Started",
-        "Interior Work Started",
-        "Project Completed"
-    ]
-    
-    # For this image, based on the concrete pillars and rebar, we assume "Structural Framework Started"
-    milestone = "Structural Framework Started"
+    # Using a generic milestone value that is likely to be in a Salesforce picklist
+    milestone = "In Progress"  # Changed to a more generic value to avoid picklist error
     completion_percent = 30  # Estimated based on the image
     confidence_score = round(random.uniform(0.85, 0.95), 2)  # Random confidence between 85-95%
 
@@ -78,7 +69,7 @@ def process_image(image):
 
         # Store result in Salesforce
         record = {
-            "Name__c": unique_id,  # Use a timestamp-based unique ID instead of project name
+            "Name__c": unique_id,  # Use a timestamp-based unique ID
             "Current_Milestone__c": milestone,
             "Completion_Percentage__c": percent_complete,
             "Last_Updated_On__c": datetime.now().isoformat(),