Update app.py

app.py

@@ -43,21 +43,22 @@ def analyze_image_with_hf(image_path):
     img = Image.open(image_path)
     detected_elements = []
     
-    # Enhanced mock detection based on filename keywords and simple content inference
-    filename = os.path.basename(image_path).lower()
-    img_data = img.tobytes()
-    img_hash = int(hashlib.sha256(img_data).hexdigest(), 16)
+    # Enhanced mock detection based on image size and pixel data (simulating content analysis)
+    width, height = img.size
+    img_data = list(img.getdata())
+    gray_pixels = sum(1 for pixel in img_data if sum(pixel[:3]) / 3 < 128) / len(img_data)  # Proportion of dark pixels
+    total_pixels = width * height
     
-    # Heuristic detection based on filename or content
-    if any(keyword in filename for keyword in ["plan", "design", "blueprint"]):
+    # Heuristic detection based on image characteristics
+    if total_pixels < 500000:  # Small image size, assume early planning stage
         detected_elements.append("planning")
-    elif any(keyword in filename for keyword in ["foundation", "excavation", "slab"]):
+    elif gray_pixels > 0.6 and width > 1000:  # High proportion of dark pixels, assume concrete/foundation
         detected_elements.append("foundation")
-    elif any(keyword in filename for keyword in ["wall", "column", "beam", "structure"]):
+    elif gray_pixels > 0.4 and width > 1000 and height > 1000:  # Larger structure with columns, assume walls erected
         detected_elements.append("walls")
-    elif any(keyword in filename for keyword in ["interior", "electrical", "plumbing", "flooring"]):
-        detected_elements.extend(["interior", "electrical_wiring", "plumbing_pipes", "flooring"])
-    elif any(keyword in filename for keyword in ["roof", "window", "door", "cladding", "finished"]):
+    elif any(keyword in os.path.basename(image_path).lower() for keyword in ["interior", "plumbing", "electrical", "ceiling"]) or (gray_pixels < 0.3 and total_pixels > 800000):  # Interior work or bright finished interior
+        detected_elements.extend(["interior", "electrical_wiring", "plumbing_pipes"])
+    else:  # Assume completed if no other clear indicators
         detected_elements.extend(["roof", "windows", "doors", "cladding", "finished"])
 
     return detected_elements
@@ -170,7 +171,7 @@ def process_image(images, project_name):
             completed_tasks.extend(all_tasks["Walls Erected"])
             final_milestone = "Walls Erected"
             percent_complete = 50
-        elif all(elem in ["interior", "electrical_wiring", "plumbing_pipes", "flooring"] for elem in detected_elements_all if elem in ["interior", "electrical_wiring", "plumbing_pipes", "flooring"]):
+        elif any(elem in ["interior", "electrical_wiring", "plumbing_pipes"] for elem in detected_elements_all):
             completed_tasks.extend(all_tasks["Interior Furnishing"])
             final_milestone = "Interior Furnishing"
             percent_complete = 80