Update app.py

app.py

@@ -41,7 +41,7 @@ local_timezone = pytz.timezone("Asia/Kolkata")
 def process_image(images, project_name):
     try:
         if not images or len(images) == 0:
-            return "Error: Please upload at least one image to proceed.", "Pending", "", 0
+            return "Error: Please upload at least one image to proceed.", "Pending", "", "", 0
 
         # Process each image
         milestones = []
@@ -49,9 +49,9 @@ def process_image(images, project_name):
             img = Image.open(image)
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
             if image_size_mb > 20:
-                return "Error: One or more images exceed 20MB.", "Failure", "", 0
+                return "Error: One or more images exceed 20MB.", "Failure", "", "", 0
             if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
-                return "Error: Only JPG/PNG images are supported.", "Failure", "", 0
+                return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
             # Save image to public folder temporarily before uploading to Salesforce
             upload_dir = "public_uploads"
@@ -78,7 +78,7 @@ def process_image(images, project_name):
                 content_version_id = content_version_result['id']
                 file_url = f"https://sathkruthatechsolutionspri8-dev-ed.develop.lightning.force.com/{content_version_id}"
             except Exception as e:
-                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", 0
+                return f"Error: Failed to upload image to Salesforce - {str(e)}", "Failure", "", "", 0
 
             # Mock AI model to detect milestone based on image content
             img_bytes = img.tobytes()
@@ -88,11 +88,14 @@ def process_image(images, project_name):
 
             # Adjust milestone detection based on internal/external features
             if milestone == "Walls Erected" and any("interior" in img_text.lower() for img_text in [os.path.basename(image)] if img_text):
-                # Check for visible internal works like electrical/plumbing
                 if any(keyword in str(os.path.basename(image)).lower() for keyword in ["electrical", "plumbing", "wiring", "pipes"]):
-                    milestone = "Completed"  # Upgrade to Completed if internal works are detected
+                    milestone = "Completed"
                 else:
-                    milestone = "Walls Erected"  # Retain if no internal works visible
+                    milestone = "Walls Erected"
+
+            # Specifically checking for external features indicating completion
+            if "completed" in os.path.basename(image).lower():
+                milestone = "Completed"  # Detect "completed" word or finished facade
 
             milestones.append(milestone)
 
@@ -106,7 +109,6 @@ def process_image(images, project_name):
         }
         percent_complete = milestone_completion_map.get(final_milestone, 0)
 
-        # Detailed Completion Breakdown based on final milestone
         completion_details = {
             "Planning": {
                 "completed": [
@@ -166,7 +168,6 @@ def process_image(images, project_name):
         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])
 
-        # Create HTML for the output with collapsible sections and progress bar
         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>
@@ -206,11 +207,9 @@ def process_image(images, project_name):
         </div>
         """
 
-        # Adjust the current time to local timezone with proper ISO 8601 format
         now = datetime.now(local_timezone)
         local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]
 
-        # Create the Salesforce record (removing confidence score)
         record = {
             "Name__c": project_name,
             "Current_Milestone__c": final_milestone,
@@ -224,14 +223,14 @@ def process_image(images, project_name):
         try:
             sf.Construction__c.create(record)
         except Exception as e:
-            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", 0
+            return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
 
         return result_html, "Success", final_milestone, f"{percent_complete}%"
 
     except Exception as e:
-        return f"Error: {str(e)}", "Failure", "", "0%"
+        return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# Gradio UI with enhanced styling
+# Gradio UI
 with gr.Blocks(css="""
     .gradio-container {
         background-color: #f0f4f8;
@@ -289,7 +288,7 @@ with gr.Blocks(css="""
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
 
     submit_button = gr.Button("Process Image")
-    output_html = gr.HTML(label="Result")  # Changed to HTML for richer output
+    output_html = gr.HTML(label="Result")
     upload_status = gr.Textbox(label="Upload Status")
     milestone = gr.Textbox(label="Detected Milestone")
     progress = gr.Textbox(label="Completion Percentage", interactive=False)
@@ -300,4 +299,4 @@ with gr.Blocks(css="""
         outputs=[output_html, upload_status, milestone, progress]
     )
 
-demo.launch(share=True)
+demo.launch(share=True)