Update app.py

app.py

@@ -124,9 +124,9 @@ class Image:
         return resp["data"]
 
 
-def overlay_text_on_image(img, text, position, text_color=(255, 255, 255), box_color=(0, 0, 0, 128)):
+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
-    if isinstance(img, str) or isinstance(img, bytes):
+    if decode:
         img_bytes = base64.b64decode(img)
         img = PIL.Image.open(BytesIO(img_bytes))
 
@@ -202,8 +202,8 @@ Do not include specific numbers.'''.replace('\n', ' '))
         forecast = Weather(zip_code).get_info()
         images, texts = [], []
         for time, data in forecast.items():
-            img, txt = overlay_text_on_image(self.step_one_forecast(data, **kwargs), time, 'top-right')
-            images.append(img)
+            img, txt = self.step_one_forecast(data, **kwargs)
+            images.append(overlay_text_on_image(img, time, 'top-right', decode=True)
             texts.append(txt)
         return create_collage(*images), *texts