Update app.py

app.py

@@ -88,11 +88,10 @@ 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"
 
             milestones.append(milestone)
 
@@ -106,7 +105,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 +164,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 +203,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,
@@ -226,12 +221,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 result_html, "Success", final_milestone, f"{percent_complete}%"
 
     except Exception as e:
         return f"Error: {str(e)}", "Failure", "", "", "0%"
 
-# Gradio UI with enhanced styling
+# Gradio UI
 with gr.Blocks(css="""
     .gradio-container {
         background-color: #f0f4f8;
@@ -289,16 +284,15 @@ 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")
-    confidence = gr.Textbox(label="Confidence Score")  # Kept for compatibility, but unused
     progress = gr.Textbox(label="Completion Percentage", interactive=False)
 
     submit_button.click(
         fn=process_image,
         inputs=[image_input, project_name_input],
-        outputs=[output_html, upload_status, milestone, confidence, progress]
+        outputs=[output_html, upload_status, milestone, progress]
     )
 
-demo.launch(share=True)
+demo.launch(share=True)