Update app.py

app.py

@@ -21,7 +21,7 @@ API_KEY = os.getenv("API_KEY", "your-api-key-here")
 
 # Validate Salesforce credentials
 if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN ]):
-    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN  in environment variables.")
+    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD SF_SECURITY_TOKEN  in environment variables.")
 
 # Initialize Salesforce connection
 try:
@@ -123,7 +123,7 @@ def process_image(image, project_name):
 
         # Validate image size and type
         image_size_mb = os.path.getsize(image) / (1024 * 1024)
-        if image_size_mb > 20:
+        if image_size_mb_tb > 20:
             return "Error: Image size exceeds 20MB.", "Failure", "", "", 0
         if not image.lower().endswith(('.jpg', '.jpeg', '.png')):
             return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
@@ -133,7 +133,7 @@ def process_image(image, project_name):
 
         # Update Salesforce record
         record = {
-            "Name": project_name,
+            "Name__c": project_name,  # Changed from "Name" to "Name__c"
             "Current_Milestone__c": milestone,
             "Completion_Percentage__c": percent_complete,
             "Last_Updated_On__c": datetime.now().isoformat(),
@@ -142,13 +142,15 @@ def process_image(image, project_name):
         }
 
         try:
-            query = f"SELECT Id FROM Construction_Project__c WHERE Name = '{project_name}'"
+            # Escape single quotes in project_name to prevent SQL injection
+            escaped_project_name = project_name.replace("'", "''")
+            query = f"SELECT Id FROM Construction__c WHERE Name__c = '{escaped_project_name}'"
             result = sf.query(query)
             if result["totalSize"] > 0:
                 project_id = result["records"][0]["Id"]
-                sf.Construction_Project__c.update(project_id, record)
+                sf.Construction__c.update(project_id, record)
             else:
-                sf.Construction_Project__c.create(record)
+                sf.Construction__c.create(record)
         except Exception as e:
             return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0