Update app.py

app.py

@@ -1,16 +1,14 @@
-# All imports and env setup are the same
 import gradio as gr
 from PIL import Image
 import os
 from dotenv import load_dotenv
 from simple_salesforce import Salesforce
 from datetime import datetime
-import hashlib
 import shutil
 import base64
 import pytz
 
-# Load env
+# Load environment variables
 load_dotenv()
 SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
@@ -70,26 +68,18 @@ 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 logic
+            # 👉 Milestone detection purely by filename
             filename = os.path.basename(image).lower()
-            if "completed" in filename or "finished" in filename:
+            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 or "plan" in filename:
+            elif "planning" in filename:
                 milestone = "Planning"
             else:
-                # fallback - trust image visually represents planning
-                # Optional: inspect pixel size or use hash override
-                img_hash = hashlib.sha256(img.tobytes()).hexdigest()
-
-                # If it's this known hash (from your Planning image), default it manually
-                if img_hash.startswith("f031ed44") or image_filename.endswith("QPV6O.jpg"):
-                    milestone = "Planning"
-                else:
-                    milestone = "Planning"  # safer default instead of random logic
+                milestone = "Planning"  # default fallback
 
             milestones.append(milestone)
 
@@ -102,7 +92,6 @@ def process_image(images, project_name):
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
 
-        # --- Completion details block is unchanged (keeping original)
         completion_details = {
             "Planning": {
                 "completed": [
@@ -224,8 +213,58 @@ def process_image(images, project_name):
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# Gradio UI (same)
-with gr.Blocks(css="...") as demo:
+# 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:
     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)")