File size: 821 Bytes
d08fbc6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
import re
from typing import Any, Dict
from .dict_utils import dict_get
from .operators import InstanceFieldOperator
def extract_images(text, instance):
regex = r'<img\s+src=["\'](.*?)["\']'
image_sources = re.findall(regex, text)
images = []
for image_source in image_sources:
image = dict_get(instance, image_source)
images.append(image)
return images
class ImageToText(InstanceFieldOperator):
def process_instance_value(self, value: Any, instance: Dict[str, Any]):
if "media" not in instance:
instance["media"] = {}
if "images" not in instance["media"]:
instance["media"]["images"] = []
idx = len(instance["media"]["images"])
instance["media"]["images"].append(value)
return f'<img src="media/images/{idx}">'
|