From 89908603cf146ef0eab6a55d4ba3fdb02ec24abb Mon Sep 17 00:00:00 2001
From: Bryce <github20210803@accounts.brycedrennan.com>
Date: Fri, 16 Sep 2022 22:34:42 -0700
Subject: [PATCH] fix: don't repeatedly download the same url

---
 imaginairy/schema.py | 34 +++++++++++++++++++++-------------
 1 file changed, 21 insertions(+), 13 deletions(-)

diff --git a/imaginairy/schema.py b/imaginairy/schema.py
index 3ba3cfc..7ba29fe 100644
--- a/imaginairy/schema.py
+++ b/imaginairy/schema.py
@@ -4,6 +4,7 @@ import logging
 import os.path
 import random
 from datetime import datetime, timezone
+from functools import lru_cache
 
 import numpy
 import requests
@@ -48,6 +49,8 @@ class LazyLoadingImage:
         if key == "_img":
             #  http://nedbatchelder.com/blog/201010/surprising_getattr_recursion.html
             raise AttributeError()
+        if self._img:
+            return getattr(self._img, key)
 
         if self._lazy_filepath:
             self._img = Image.open(self._lazy_filepath)
@@ -79,19 +82,19 @@ class WeightedPrompt:
 
 class ImaginePrompt:
     def __init__(
-        self,
-        prompt=None,
-        prompt_strength=7.5,
-        init_image=None,  # Pillow Image, LazyLoadingImage, or filepath str
-        init_image_strength=0.3,
-        seed=None,
-        steps=50,
-        height=512,
-        width=512,
-        upscale=False,
-        fix_faces=False,
-        sampler_type="PLMS",
-        conditioning=None,
+            self,
+            prompt=None,
+            prompt_strength=7.5,
+            init_image=None,  # Pillow Image, LazyLoadingImage, or filepath str
+            init_image_strength=0.3,
+            seed=None,
+            steps=50,
+            height=512,
+            width=512,
+            upscale=False,
+            fix_faces=False,
+            sampler_type="PLMS",
+            conditioning=None,
     ):
         prompt = prompt if prompt is not None else "a scenic landscape"
         if isinstance(prompt, str):
@@ -193,3 +196,8 @@ class ImagineResult:
 
     def save_upscaled(self, save_path):
         self.upscaled_img.save(save_path, exif=self._exif())
+
+
+@lru_cache(maxsize=2)
+def _get_briefly_cached_url(url):
+    return requests.get(url, timeout=60)