Add a better hint

Signed-off-by: bghira <bghira@users.github.com>

discord_tron_master/cogs/image/img2img.py

@@ -9,6 +9,7 @@
 from PIL import Image
 from discord_tron_master.bot import DiscordBot
 from discord_tron_master.classes.jobs.image_variation_job import ImageVariationJob
+from discord_tron_master.classes.jobs.image_generation_job import ImageGenerationJob
 from discord_tron_master.classes.jobs.image_upscaling_job import ImageUpscalingJob
 from discord_tron_master.bot import clean_traceback
 from discord_tron_master.cogs.image.generate import Generate
@@ -103,6 +104,8 @@ async def _handle_image_attachment(self, message, attachment):
             # Remove "!upscale" from the contents:
             message.content = message.content.replace("!upscale", "")
             job = ImageUpscalingJob((self.bot, self.config, message, message.content, discord_first_message, attachment.url))
+        elif message.content != "":
+            job = ImageGenerationJob((self.bot, self.config, message, message.content, discord_first_message, attachment.url))
         else:
             # Default to image variation job
             job = ImageVariationJob((self.bot, self.config, message, message.content, discord_first_message, attachment.url))

discord_tron_master/websocket_hub.py

@@ -48,7 +48,7 @@ async def handler(self, websocket, path):
                 result = json.dumps(raw_result)
                 # Did result error? If so, close the websocket connection:
                 if "RegistrationError" in raw_result:
-                    await websocket.close(code=4002, reason=raw_result)
+                    await websocket.close(code=4002, reason="RegistrationError:" + raw_result)
                     return
                 if raw_result is None or "error" in raw_result:
                     if raw_result is None: