Update app.py

app.py

@@ -81,16 +81,8 @@ def process_image(images, project_name):
             except Exception as e:
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
-            # Checking for external features indicating completion
-            # If image contains features suggesting a completed building (such as windows, doors, exterior finish)
-            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]
+            # Enhanced Milestone Detection
+            milestone = detect_milestone(image)
 
             milestones.append(milestone)
 
@@ -225,6 +217,22 @@ def process_image(images, project_name):
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
+# Enhanced Milestone Detection based on image content or filename
+def detect_milestone(image_path):
+    # Check the image filename for certain keywords that might indicate a milestone
+    image_name = os.path.basename(image_path).lower()
+    if "planning" in image_name:
+        return "Planning"
+    elif "foundation" in image_name:
+        return "Foundation"
+    elif "walls" in image_name or "erected" in image_name:
+        return "Walls Erected"
+    elif "completed" in image_name:
+        return "Completed"
+    
+    # Default to Planning if no specific milestone found
+    return "Planning"
+
 # Gradio UI
 with gr.Blocks(css="""
     .gradio-container {