Update app.py

app.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import hashlib
 import shutil
 import base64
+import pytz
 
 # Load environment variables
 load_dotenv()
@@ -33,6 +34,9 @@ except Exception as e:
 # Valid milestones
 VALID_MILESTONES = ["Foundation", "Walls Erected", "Planning", "Completed"]
 
+# Adjust the timezone to your local timezone (replace 'Asia/Kolkata' with your timezone if needed)
+local_timezone = pytz.timezone("Asia/Kolkata")
+
 # Deterministic AI prediction with fixed confidence and percent
 def mock_ai_model(image):
     img = image.convert("RGB")
@@ -68,7 +72,7 @@ def process_image(image, project_name):
         image_size_mb = os.path.getsize(image) / (1024 * 1024)
         if image_size_mb > 20:
             return "Error: Image size exceeds 20MB.", "Failure", "", "", 0
-        if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
+        if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')): 
             return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
         # Save image to public folder temporarily before uploading to Salesforce
@@ -103,14 +107,17 @@ def process_image(image, project_name):
         # AI-based milestone and completion prediction
         milestone, percent_complete, confidence_score = mock_ai_model(img)
 
+        # Adjust the current time to local timezone
+        local_time = datetime.now(local_timezone).strftime("%Y-%m-%d %H:%M:%S")
+
         # Create the Salesforce record with the image URL and AI prediction
         record = {
             "Name__c": project_name,
             "Current_Milestone__c": milestone,
             "Completion_Percentage__c": percent_complete,
-            "Last_Updated_On__c": datetime,
+            "Last_Updated_On__c": local_time,  # Use the adjusted local time
             "Upload_Status__c": "Success",
-            "Comments__c": f"{milestone} with {confidence_score*100}% confidence",
+            "Comments__c": f"{milestone} with {confidence_score*100}% confidence",  # Removed "AI Prediction:"
             "Last_Updated_Image__c": file_url
         }