Update app.py

app.py

@@ -24,7 +24,7 @@ try:
         username=SF_USERNAME,
         password=SF_PASSWORD,
         security_token=SF_SECURITY_TOKEN,
-        domain='login'  # Ensure this matches your org's instance (e.g., 'login' for production, 'test' for sandboxes)
+        domain='login'
     )
 except Exception as e:
     print(f"Salesforce connection failed: {str(e)}")
@@ -68,12 +68,19 @@ def process_image(images, project_name):
         # Process each image
         detected_elements_all = []
         for image in images:
-            img = Image.open(image)
+            img = None
+            try:
+                img = Image.open(image)
+            except Exception as e:
+                if "cannot identify image file" in str(e):
+                    return f"Error: Unsupported image format '{os.path.basename(image)}'. Please use JPG, JPEG, PNG, or convert to a supported format.", "Failure", "", "", 0
+                return f"Error: Failed to open image '{os.path.basename(image)}' - {str(e)}", "Failure", "", "", 0
+
             image_size_mb = os.path.getsize(image) / (1024 * 1024)
             if image_size_mb > 20:
                 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
+            if not str(image).lower().endswith(('.jpg', '.jpeg', '.png', '.avif')):  # Updated to include .avif
+                return "Error: Only JPG, JPEG, PNG, or AVIF images are supported.", "Failure", "", "", 0
 
             # Save image to public folder temporarily before uploading to Salesforce
             upload_dir = "public_uploads"