Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from PIL import Image
+import numpy as np
 import os
 from dotenv import load_dotenv
 from simple_salesforce import Salesforce
@@ -8,14 +9,14 @@ import shutil
 import base64
 import pytz
 
-# Load environment variables
+# Load env
 load_dotenv()
 SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 
 if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
-    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD, and SF_SECURITY_TOKEN in environment variables.")
+    raise ValueError("Missing Salesforce credentials.")
 
 try:
     sf = Salesforce(
@@ -31,6 +32,22 @@ except Exception as e:
 VALID_MILESTONES = ["Planning", "Foundation", "Walls Erected", "Completed"]
 local_timezone = pytz.timezone("Asia/Kolkata")
 
+# ✅ Actual image analysis logic
+def detect_milestone_from_image(image_path):
+    img = Image.open(image_path).convert("L")
+    img_array = np.array(img)
+    brightness = np.mean(img_array)
+
+    if brightness < 80:
+        return "Planning"
+    elif 80 <= brightness < 130:
+        return "Foundation"
+    elif 130 <= brightness < 180:
+        return "Walls Erected"
+    else:
+        return "Completed"
+
+# Main logic
 def process_image(images, project_name):
     try:
         if not images or len(images) == 0:
@@ -68,19 +85,8 @@ def process_image(images, project_name):
             except Exception as e:
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
-            # 👉 Milestone detection purely by filename
-            filename = os.path.basename(image).lower()
-            if "completed" in filename:
-                milestone = "Completed"
-            elif "walls" in filename or "erected" in filename:
-                milestone = "Walls Erected"
-            elif "foundation" in filename:
-                milestone = "Foundation"
-            elif "planning" in filename:
-                milestone = "Planning"
-            else:
-                milestone = "Planning"  # default fallback
-
+            # 🧠 Milestone from actual image content
+            milestone = detect_milestone_from_image(saved_image_path)
             milestones.append(milestone)
 
         final_milestone = max(set(milestones), key=milestones.count) if milestones else "Planning"
@@ -92,6 +98,7 @@ def process_image(images, project_name):
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
 
+        # Task mapping
         completion_details = {
             "Planning": {
                 "completed": [
@@ -145,11 +152,8 @@ def process_image(images, project_name):
             }
         }
 
-        completed_tasks = completion_details[final_milestone]["completed"]
-        not_completed_tasks = completion_details[final_milestone]["not_completed"]
-
-        completed_html = "".join([f'<li style="color: green;">✔ {task}</li>' for task in completed_tasks])
-        not_completed_html = "".join([f'<li style="color: red;">✘ {task}</li>' for task in not_completed_tasks])
+        completed_html = "".join([f'<li style="color: green;">✔ {task}</li>' for task in completion_details[final_milestone]["completed"]])
+        not_completed_html = "".join([f'<li style="color: red;">✘ {task}</li>' for task in completion_details[final_milestone]["not_completed"]])
 
         result_html = f"""
         <div style="font-family: Arial, sans-serif; padding: 20px; background-color: #f9f9f9; border-radius: 10px;">
@@ -213,58 +217,8 @@ def process_image(images, project_name):
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# UI Setup
-with gr.Blocks(css="""
-    .gradio-container {
-        background-color: #f0f4f8;
-        font-family: Arial, sans-serif;
-    }
-    .title {
-        color: #2c3e50;
-        font-size: 24px;
-        text-align: center;
-        font-weight: bold;
-    }
-    .gradio-row {
-        text-align: center;
-    }
-    .gradio-container .output {
-        text-align: center;
-    }
-    .gradio-container .input {
-        text-align: center;
-    }
-    .gradio-container .button {
-        display: block;
-        margin: 0 auto;
-        background-color: #3498db;
-        color: white;
-        border: none;
-        padding: 10px 20px;
-        border-radius: 5px;
-        cursor: pointer;
-    }
-    .gradio-container .button:hover {
-        background-color: #2980b9;
-    }
-    progress::-webkit-progress-value {
-        background-color: #2ecc71;
-        border-radius: 5px;
-    }
-    progress::-webkit-progress-bar {
-        background-color: #ecf0f1;
-        border-radius: 5px;
-    }
-    details summary {
-        cursor: pointer;
-        padding: 10px;
-        background-color: #ecf0f1;
-        border-radius: 5px;
-    }
-    details ul {
-        margin-top: 10px;
-    }
-""") as demo:
+# UI
+with gr.Blocks(css="") as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():
         image_input = gr.Files(type="filepath", label="Upload Construction Site Photos (JPG/PNG, ≤ 20MB)")