svg

image/remove_metadata.py

@@ -201,6 +201,11 @@ class RemoveMetadata(ImageBase):
 		elif format_name in ['BMP', 'GIF']:
 			settings = {}  # These formats have limited metadata anyway
 			
+		elif format_name == 'HEIC':
+			settings = {
+				'quality': 100  # Maximum quality for HEIC
+			}
+			
 		return settings
 	
 	def _clean_png_with_transparency(self, image: Image.Image, output_path: str) -> bool:
@@ -425,9 +430,15 @@ class RemoveMetadata(ImageBase):
 				# Choose appropriate cleaning method based on format
 				success = False
 				
+				# PNG with transparency
 				if original_format == 'PNG' and original_mode in ('RGBA', 'LA', 'P'):
 					success = self._clean_png_with_transparency(image, output_path)
-					
+				
+				# WEBP with transparency (handle like PNG)
+				elif original_format == 'WEBP' and original_mode in ('RGBA', 'LA', 'P'):
+					success = self._clean_png_with_transparency(image, output_path)
+				
+				# JPEG
 				elif original_format == 'JPEG' or output_path.lower().endswith(('.jpg', '.jpeg')):
 					success = self._clean_jpeg_image(image, output_path)
 					
@@ -449,6 +460,13 @@ class RemoveMetadata(ImageBase):
 			
 		except Exception as e:
 			logger_config.error(f"Failed to clean image: {str(e)}")
+			# Error recovery: clean up partial output
+			if 'output_path' in locals() and os.path.exists(output_path):
+				try:
+					os.unlink(output_path)
+					logger_config.info(f"Cleaned up partial output: {output_path}")
+				except:
+					pass
 			return None, {}
 
 # Example usage