From a9409f08bac76b041fda8675858290d915a183aa Mon Sep 17 00:00:00 2001 From: Anshul Gupta Date: Fri, 15 Nov 2024 14:13:32 -0800 Subject: Add image processing and extra data to openai call --- core/image.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) create mode 100644 core/image.py (limited to 'core/image.py') diff --git a/core/image.py b/core/image.py new file mode 100644 index 0000000..7e3c68d --- /dev/null +++ b/core/image.py @@ -0,0 +1,23 @@ +from bs4 import BeautifulSoup +from typing import Optional + + +def contains_image(html: str) -> Optional[str]: + """ + Extracts the source URL of the first image found in the given HTML content. + + This function parses the provided HTML string to locate the first image + (``) tag. If an image tag is found, it returns the value of the `src` + attribute. If no image is found, it returns None. + + :param html: A string containing HTML content to be parsed. + :type html: str + :return: The `src` attribute value of the first found image tag, or None if + no image tag is found. + :rtype: Optional[str] + """ + soup = BeautifulSoup(html, "html.parser") + image = soup.find("img", recursive=True) + if image: + return image.get("src") + return None -- cgit v1.2.3