Update app.py

app.py

@@ -18,6 +18,7 @@ import PIL.ImageFont
 import gradio as gr
 import cachetools.func
 from huggingface_hub import hf_hub_download
+import concurrent.futures
 
 
 logger = structlog.getLogger()
@@ -126,6 +127,19 @@ class Image:
         if n == 1: return resp["data"][0]
         return resp["data"]
 
+def create_collage(image1, image2, image3, image4):
+    # assuming images are the same size
+    width, height = image1.size
+
+    new_img = PIL.Image.new('RGB', (2 * width, 2 * height))
+
+    # place images in collage image
+    new_img.paste(image1, (0,0))
+    new_img.paste(image2, (width, 0))
+    new_img.paste(image3, (0, height))
+    new_img.paste(image4, (width, height))
+
+    return new_img
 
 def overlay_text_on_image(img, text, position, text_color=(255, 255, 255), box_color=(0, 0, 0, 128), decode=False):
     # Convert the base64 string back to an image
@@ -206,10 +220,15 @@ Do not include specific numbers.'''.replace('\n', ' '))
     def step(self, zip_code='10001', **kwargs):
         forecast = Weather(zip_code).get_info()
         images, texts = [], []
-        for time, data in forecast.items():
-            img, txt = self.step_one_forecast(data, **kwargs)
-            images.append(overlay_text_on_image(img, time, 'top-right', decode=True))
-            texts.append(txt)
+        with concurrent.futures.ThreadPoolExecutor(max_workers=4) as e:
+            runs = {}
+            for time, data in forecast.items():
+                runs[e.submit(self.step_one_forecast, data, **kwargs)] = time, data
+            for r in concurrent.futures.as_completed(runs.keys()):
+                img, txt = r.result()
+                time, data = runs[r]
+                images.append(overlay_text_on_image(img, time, 'top-right', decode=True))
+                texts.append(txt)
         return create_collage(*images), *texts