Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 from dotenv import load_dotenv
 from simple_salesforce import Salesforce
 from datetime import datetime
-import random
+import hashlib
 import shutil
 
 # Load environment variables
@@ -32,14 +32,26 @@ except Exception as e:
 # Valid milestones
 VALID_MILESTONES = ["Foundation", "Walls Erected", "Planning", "Completed"]
 
-# Mock AI prediction
+# Updated deterministic AI prediction based on image hash
 def mock_ai_model(image):
     img = image.convert("RGB")
     max_size = 1024
     img.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
-    milestone = random.choice(VALID_MILESTONES)
-    completion_percent = random.choice([10, 30, 50, 80, 100])
-    confidence_score = round(random.uniform(0.85, 0.95), 2)
+
+    # Convert image bytes to a hash value
+    img_bytes = img.tobytes()
+    img_hash = int(hashlib.sha256(img_bytes).hexdigest(), 16)
+
+    # Use the hash to deterministically pick milestone and percentages
+    milestone_index = img_hash % len(VALID_MILESTONES)
+    completion_choices = [10, 30, 50, 80, 100]
+    completion_index = (img_hash // 10) % len(completion_choices)
+    confidence_raw = 0.85 + ((img_hash % 1000) / 1000) * (0.95 - 0.85)
+    confidence_score = round(confidence_raw, 2)
+
+    milestone = VALID_MILESTONES[milestone_index]
+    completion_percent = completion_choices[completion_index]
+
     return milestone, completion_percent, confidence_score
 
 # Gradio function