Untitled

 avatar
unknown
python
a year ago
929 B
10
Indexable
	def getProperty(self, png_file_path, node_id, input_key):
		# Open the PNG file
		with Image.open(png_file_path) as img:
			# Extract metadata
			metadata = img.info

		# Look for the custom JSON key in the metadata
		json_data = metadata.get('prompt')
		if json_data is None:
			raise ValueError("No 'prompt' JSON metadata found in the PNG file.")

		# Parse the JSON metadata
		try:
			data = json.loads(json_data)
		except json.JSONDecodeError as e:
			raise ValueError("Invalid JSON data found in the metadata.") from e

		# Search for the node with the given ID
		node_info = data.get(str(node_id))
		if node_info is None:
			raise ValueError(f"Node with ID {node_id} not found.")

		# Extract the input value
		input_value = node_info.get('inputs', {}).get(input_key)
		if input_value is None:
			raise ValueError(f"Input key '{input_key}' not found in node ID {node_id}.")

		print(input_value)
Leave a Comment