Update app.py

app.py

@@ -44,21 +44,30 @@ milestone_percentage_map = {
 # Adjust the timezone to your local timezone
 local_timezone = pytz.timezone("Asia/Kolkata")
 
-# AI model prediction (for simulation)
+# AI model prediction with basic rule-based check
 def mock_ai_model(image):
     img = image.convert("RGB")
     max_size = 1024
     img.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
 
-    img_bytes = img.tobytes()
-    img_hash = int(hashlib.sha256(img_bytes).hexdigest(), 16)
+    # Basic rule-based milestone detection (simplified placeholder)
+    # Check for structural elements (columns, rebar) vs. exterior features
+    img_data = list(img.getdata())
+    gray_scale = [sum(pixel) / 3 for pixel in img_data]  # Simple grayscale conversion
+    avg_brightness = sum(gray_scale) / len(gray_scale)
+
+    if avg_brightness < 150:  # Darker image might indicate early stages or groundwork
+        milestone = "Excavation and Foundation"
+    elif any(255 in pixel for pixel in img_data):  # Bright spots might indicate concrete structures
+        milestone = "Structural Framework"
+    elif avg_brightness > 200:  # Brighter image might suggest advanced stages
+        milestone = "Exterior Work"
+    else:
+        milestone_index = int(hashlib.sha256(img.tobytes()).hexdigest(), 16) % len(milestone_percentage_map)
+        milestone = list(milestone_percentage_map.keys())[milestone_index]
 
-    # Assign milestone based on hash (for simulation)
-    milestone_index = img_hash % len(milestone_percentage_map)
-    milestone = list(milestone_percentage_map.keys())[milestone_index]
     completion_percent = milestone_percentage_map[milestone]
-
-    confidence_raw = 0.85 + ((img_hash % 1000) / 1000) * (0.95 - 0.85)
+    confidence_raw = 0.85 + ((hashlib.sha256(img.tobytes()).hexdigest() % 1000) / 1000) * (0.95 - 0.85)
     confidence_score = round(confidence_raw, 2)
 
     return milestone, completion_percent, confidence_score
@@ -234,9 +243,9 @@ def process_image(image, project_name, project_type):
         </div>
         """
 
-        # Create Salesforce record
+        # Create Salesforce record with current date and time
         now = datetime.now(local_timezone)
-        local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
+        local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]  # 2025-06-30T17:07:00+05:30
 
         record = {
             "Name__c": project_name,