Spoiler support for Live preview

Kilvoctu · Mar 1, 2024 · cfae091 · cfae091
1 parent 265709d
commit cfae091
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/stablecog.py b/core/stablecog.py
@@ -38,7 +38,11 @@ async def update_progress(event_loop, status_message_task, s, queue_object, trie
             buffer = stack.enter_context(io.BytesIO())
             image.save(buffer, 'PNG')
             buffer.seek(0)
-            file = discord.File(fp=buffer, filename=f'{queue_object.seed}.png')
+            filename=f'{queue_object.seed}.png'
+            if queue_object.spoiler:
+                filename=f'SPOILER_{queue_object.seed}.png'
+            fp=buffer
+            file = discord.File(fp, filename)
 
         ips = '?'
         if progress_data["eta_relative"] != 0: