Update app.py

app.py

@@ -80,18 +80,16 @@ def process_image(images, project_name):
             except Exception as e:
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
-            # Mock AI model to detect milestone based on image content
-            img_bytes = img.tobytes()
-            img_hash = int(hashlib.sha256(img_bytes).hexdigest(), 16)
-            milestone_index = img_hash % len(VALID_MILESTONES)
-            milestone = VALID_MILESTONES[milestone_index]
+            # Use file name to detect milestones as a fallback (simple keyword-based detection)
+            milestone = "Planning"  # Default if no match is found
+            image_filename_lower = image_filename.lower()
 
-            # Adjust milestone detection based on internal/external features
-            if milestone == "Walls Erected" and any("interior" in img_text.lower() for img_text in [os.path.basename(image)] if img_text):
-                if any(keyword in str(os.path.basename(image)).lower() for keyword in ["electrical", "plumbing", "wiring", "pipes"]):
-                    milestone = "Completed"
-                else:
-                    milestone = "Walls Erected"
+            if any(keyword in image_filename_lower for keyword in ["foundation", "excavation", "digging"]):
+                milestone = "Foundation"
+            elif any(keyword in image_filename_lower for keyword in ["walls", "framing", "structure"]):
+                milestone = "Walls Erected"
+            elif any(keyword in image_filename_lower for keyword in ["completed", "final", "finished"]):
+                milestone = "Completed"
 
             milestones.append(milestone)