Update flashcard_util.py

flashcard_util.py

@@ -1,9 +1,11 @@
 import lang
 import requests
 import json
-
+import httpx
+import aiofiles
 from imgsearch import search_img
 from re import match, sub
+from env import HOST_URL
 
 from env import LLM_API_KEY
 
@@ -36,9 +38,29 @@ async def tldr(content, l=lang.VI_VN):
     # print(_summary)
     return _summary['choices'][0]['message']['content'].split('\n',1)[-1].strip()
 
-async def fetch_img_for_words(words: list[str]):
+import io
+import asyncio
+from functools import partial
+def optimize_image(data, _fn):
+    with Image.open(io.BytesIO(data)) as img:
+        img.save(f'{_fn}.webp', optimize=True)
+
+async def download_img(_url, _prefix=HOST_URL):
+    _r = None
+    async with httpx.AsyncClient() as client:
+        _r = await client.get(url=_url, timeout=None)
+
+    _fn = f'img_{str(hash(_url))}'
+    if _r.status_code//100 == 2:
+
+        _loop = asyncio.get_running_loop()
+        _out = await _loop.run_in_executor(None, partial(optimize_image, _fn))
+        
+    return f"{_prefix}/images/{_fn}.webp"
+
+async def fetch_img_for_words(words: list[str], __url_prefix=None):
     print("fetching images...")
-    _img_link = [search_img(r) for r in words]
+    _img_link = [await download_img(search_img(r)) for r in words]
     return [(word,img) for (word, img) in zip(words, _img_link)]
 
 async def get_definitions_from_words(words: list[str], summary: str = "", lang: str = lang.VI_VN):