Only generate a figcaption if image has both prompt and code

[zombocom-ai] / generate-image.py

diff --git a/generate-image.py b/generate-image.py
index 408fe21bebfe8086accbf4f373176970fa6dcaaa..f5502c6321d337ccdd8435f84e01737aeb0a3fdf 100755 (executable)
--- a/generate-image.py
+++ b/generate-image.py
@@ -27,10 +27,11 @@ def save_artifact(artifact, prompt):
     counter = 1
     seed = artifact.seed
     while True:
+        prompt_abbrev = prompt[:200]
         if counter > 1:
-            base = "{}_{}_{}.png".format(seed, prompt, counter)
+            base = "{}_{}_{}.png".format(seed, prompt_abbrev, counter)
         else:
-            base = "{}_{}.png".format(seed, prompt)
+            base = "{}_{}.png".format(seed, prompt_abbrev)
         base = normalize_filename(base)
         filename = "{}{}/{}".format(DOCUMENT_ROOT, IMAGES_PATH, base)
         if not os.path.exists(filename):