Update app.py

app.py

@@ -51,7 +51,6 @@ def mock_ai_model(image):
     img.thumbnail((max_size, max_size), Image.Resampling.LANCZOS)
 
     # Basic rule-based milestone detection (simplified placeholder)
-    # Check for structural elements (columns, rebar) vs. exterior features
     img_data = list(img.getdata())
     gray_scale = [sum(pixel) / 3 for pixel in img_data]  # Simple grayscale conversion
     avg_brightness = sum(gray_scale) / len(gray_scale)
@@ -75,8 +74,8 @@ def mock_ai_model(image):
 # Image processing and Salesforce upload with detailed output
 def process_image(image, project_name, project_type):
     try:
-        if not image:
-            return "Error: Please upload an image to proceed.", "Pending", "", "", 0
+        if not image or not project_name or not project_type:
+            return "Error: Please upload an image and provide a project name and type.", "Pending", "", "", "0%"
 
         # Open the image from the file path
         img = Image.open(image)
@@ -85,9 +84,9 @@ def process_image(image, project_name, project_type):
         # Image size and format validation
         image_size_mb = os.path.getsize(image) / (1024 * 1024)
         if image_size_mb > 20:
-            return "Error: Image size exceeds 20MB.", "Failure", "", "", 0
+            return "Error: Image size exceeds 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"
@@ -113,7 +112,7 @@ def process_image(image, project_name, project_type):
             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%"
 
         # Detailed Completion Breakdown
         completion_details = {
@@ -197,6 +196,7 @@ def process_image(image, project_name, project_type):
         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
+        # Escape % signs to prevent string formatting errors
         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>
@@ -208,12 +208,12 @@ def process_image(image, project_name, project_type):
                 <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>
+                    <p>{percent_complete}%%</p> <!-- Escaped % with %% -->
                 </div>
                 <div style="text-align: center;">
                     <h3 style="color: #34495e;">Confidence Score</h3>
-                    <p style="font-size: 18px; font-weight: bold;">{confidence_score * 100}%</p>
-                    <small>The AI is {confidence_score * 100}% confident that this image represents the {milestone} stage.</small>
+                    <p style="font-size: 18px; font-weight: bold;">{confidence_score * 100}%%</p> <!-- Escaped % with %% -->
+                    <small>The AI is {confidence_score * 100}%% confident that this image represents the {milestone} stage.</small>
                 </div>
             </div>
 
@@ -243,9 +243,9 @@ def process_image(image, project_name, project_type):
         </div>
         """
 
-        # Create Salesforce record with current date and time
+        # Create Salesforce record with current date and time (05:09 PM IST, June 30, 2025)
         now = datetime.now(local_timezone)
-        local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]  # 2025-06-30T17:07:00+05:30
+        local_time = now.strftime("%Y-%m-%dT%H:%M:%S") + now.strftime("%z")[:-2] + ":" + now.strftime("%z")[-2:]  # 2025-06-30T17:09:00+05:30
 
         record = {
             "Name__c": project_name,
@@ -261,7 +261,7 @@ def process_image(image, project_name, project_type):
         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", milestone, f"Confidence Score: {confidence_score}", f"{percent_complete}%"