Add extract image endpoint

app.py

@@ -9,6 +9,7 @@ from PIL import Image
 import requests
 from io import BytesIO
 from bs4 import BeautifulSoup
+from urllib.parse import urljoin
 
 os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
@@ -33,7 +34,7 @@ def extract_images():
 
         img_tags = soup.select('div img')
         for img_tag in img_tags:
-            img_url = img_tag['src']
+            img_url = urljoin(src, img_tag['src'])
             img_urls.append(img_url)
         return json.dumps({"images":img_urls})
     except Exception as e: