Update app.py

app.py

@@ -9,7 +9,7 @@ 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")
@@ -18,6 +18,7 @@ SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
     raise ValueError("Missing Salesforce credentials.")
 
+# Connect to Salesforce
 try:
     sf = Salesforce(
         username=SF_USERNAME,
@@ -32,7 +33,7 @@ except Exception as e:
 VALID_MILESTONES = ["Planning", "Foundation", "Walls Erected", "Completed"]
 local_timezone = pytz.timezone("Asia/Kolkata")
 
-# ✅ Actual image analysis logic
+# Milestone detection based on image brightness
 def detect_milestone_from_image(image_path):
     img = Image.open(image_path).convert("L")
     img_array = np.array(img)
@@ -47,7 +48,6 @@ def detect_milestone_from_image(image_path):
     else:
         return "Completed"
 
-# Main logic
 def process_image(images, project_name):
     try:
         if not images or len(images) == 0:
@@ -85,7 +85,6 @@ def process_image(images, project_name):
             except Exception as e:
                 return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
-            # 🧠 Milestone from actual image content
             milestone = detect_milestone_from_image(saved_image_path)
             milestones.append(milestone)
 
@@ -98,102 +97,6 @@ def process_image(images, project_name):
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
 
-        # Task mapping
-        completion_details = {
-            "Planning": {
-                "completed": [
-                    "Initial project outline and objectives have been established.",
-                    "Preliminary designs and architectural plans are drafted.",
-                    "Stakeholder meetings and initial approvals are completed."
-                ],
-                "not_completed": [
-                    "Detailed construction plans and blueprints are pending finalization.",
-                    "Permits and regulatory approvals are yet to be obtained.",
-                    "Contractor selection and procurement processes are not yet complete."
-                ]
-            },
-            "Foundation": {
-                "completed": [
-                    "Site preparation, including clearing and leveling, is finished.",
-                    "Excavation for the foundation has been completed.",
-                    "Concrete pouring for the foundation, including footings and slabs, is done.",
-                    "Initial structural inspections for the foundation have been passed."
-                ],
-                "not_completed": [
-                    "Plumbing and electrical groundwork installations are pending.",
-                    "Backfilling and site grading around the foundation are not yet done.",
-                    "Above-ground structural work, such as columns and walls, has not started."
-                ]
-            },
-            "Walls Erected": {
-                "completed": [
-                    "The concrete framework, including columns and beams, is in place.",
-                    "All structural walls have been erected and stabilized.",
-                    "Temporary scaffolding and safety measures are installed for ongoing work.",
-                    "Initial inspections for structural integrity have been completed."
-                ],
-                "not_completed": [
-                    "Roofing installation and weatherproofing are pending.",
-                    "Windows, doors, and exterior cladding are not yet installed.",
-                    "Interior walls, electrical, and plumbing systems are still to be implemented."
-                ]
-            },
-            "Completed": {
-                "completed": [
-                    "The concrete framework, including columns, beams, and floor slabs, is fully constructed.",
-                    "Exterior walls, windows, and cladding are installed, completing the building's facade.",
-                    "Interior work, including electrical, plumbing, and HVAC systems, is fully implemented.",
-                    "Finishing touches, such as flooring, painting, and fixtures, are completed.",
-                    "All phases of the project are finished, including final inspections and approvals."
-                ],
-                "not_completed": [
-                    "There should be no more pending work as the project is fully completed."
-                ]
-            }
-        }
-
-        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;">
-            <h2 style="color: #2c3e50; text-align: center;">Project Summary</h2>
-            <div style="display: flex; justify-content: space-around; margin-bottom: 20px;">
-                <div style="text-align: center;">
-                    <h3 style="color: #34495e;">Detected Milestone</h3>
-                    <p style="font-size: 18px; font-weight: bold;">{final_milestone}</p>
-                </div>
-                <div style="text-align: center;">
-                    <h3 style="color: #34495e;">Completion</h3>
-                    <progress value="{percent_complete}" max="100" style="width: 200px; height: 20px;"></progress>
-                    <p>{percent_complete}%</p>
-                </div>
-            </div>
-
-            <h3 style="color: #2c3e50;">Milestone Timeline</h3>
-            <div style="display: flex; justify-content: space-between; align-items: center; margin-bottom: 20px;">
-                <span style="color: {'#2ecc71' if final_milestone == 'Planning' else '#bdc3c7'};">Planning</span>
-                <span style="color: {'#2ecc71' if final_milestone == 'Foundation' else '#bdc3c7'};">Foundation</span>
-                <span style="color: {'#2ecc71' if final_milestone == 'Walls Erected' else '#bdc3c7'};">Walls Erected</span>
-                <span style="color: {'#2ecc71' if final_milestone == 'Completed' else '#bdc3c7'};">Completed</span>
-            </div>
-
-            <details style="margin-bottom: 20px;">
-                <summary style="color: #2c3e50; font-weight: bold;">Completed Tasks</summary>
-                <ul style="padding-left: 20px;">
-                    {completed_html}
-                </ul>
-            </details>
-
-            <details style="margin-bottom: 20px;">
-                <summary style="color: #2c3e50; font-weight: bold;">Not Completed Tasks</summary>
-                <ul style="padding-left: 20px;">
-                    {not_completed_html}
-                </ul>
-            </details>
-        </div>
-        """
-
         now = datetime.now(local_timezone)
         local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
 
@@ -212,12 +115,12 @@ def process_image(images, project_name):
         except Exception as e:
             return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
 
-        return result_html, "Success", final_milestone, f"{percent_complete}%"
+        return f"<h3 style='color:green;'>Milestone: {final_milestone}</h3>", "Success", final_milestone, f"{percent_complete}%"
 
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# UI
+# Gradio UI
 with gr.Blocks(css="") as demo:
     gr.Markdown("<h1 class='title'>Construction Progress Analyzer</h1>")
     with gr.Row():