From 021e9ce72db525da148a7f7c2525968cc831d38c Mon Sep 17 00:00:00 2001
From: Guillermo Rodriguez <guillermo@telos.net>
Date: Sat, 24 Dec 2022 15:38:40 -0300
Subject: [PATCH] Fix big caption bug, and add logging

---
 skynet/dgpu.py              | 5 ++++-
 skynet/frontend/telegram.py | 6 +++++-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/skynet/dgpu.py b/skynet/dgpu.py
index 4dd8e07..a73a5b6 100644
--- a/skynet/dgpu.py
+++ b/skynet/dgpu.py
@@ -127,7 +127,10 @@ async def open_dgpu_node(
                 image = convert_from_cv2_to_image(up_img)
                 logging.info('done')
 
-            return image.tobytes()
+            raw_img = image.tobytes()
+            logging.info(f'final img size {len(raw_img)} bytes.')
+
+            return raw_img
 
         except BaseException as e:
             logging.error(e)
diff --git a/skynet/frontend/telegram.py b/skynet/frontend/telegram.py
index b40996b..47c2b66 100644
--- a/skynet/frontend/telegram.py
+++ b/skynet/frontend/telegram.py
@@ -23,7 +23,11 @@ PREFIX = 'tg'
 
 
 def prepare_metainfo_caption(meta: dict) -> str:
-    meta_str = f'prompt: \"{meta["prompt"]}\"\n'
+    prompt = meta["prompt"]
+    if len(prompt) > 256:
+        prompt = prompt[:256]
+
+    meta_str = f'prompt: \"{prompt}\"\n'
     meta_str += f'seed: {meta["seed"]}\n'
     meta_str += f'step: {meta["step"]}\n'
     meta_str += f'guidance: {meta["guidance"]}\n'