Update app.py

app.py

@@ -80,22 +80,15 @@ 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]
-
-            # 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"
-
-            # Specifically checking for external features indicating completion
-            if "completed" in os.path.basename(image).lower():
-                milestone = "Completed"  # Detect "completed" word or finished facade
+            # Check if the image represents a completed building
+            if "completed" in os.path.basename(image).lower() or "finished" in os.path.basename(image).lower():
+                milestone = "Completed"  # Directly detect completion if such features exist
+            else:
+                # Default milestone detection logic (mock AI model or image-based hash)
+                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]
 
             milestones.append(milestone)