created a consistent usage of the image with the tokens and the bearer

Author: Circuit-Overtime

cogs/imagine_cog.py

@@ -115,8 +115,6 @@ async def regenerate(
             return
 
         image_file = discord.File(image, filename="image.png")
-        if dic["nsfw"]:
-            image_file.filename = f"SPOILER_{image_file.filename}"
         time_taken_delta: datetime.timedelta = datetime.datetime.now() - start
 
         embed: SafeEmbed = await generate_pollinate_embed(
@@ -386,7 +384,12 @@ async def imagine_command(
         )
         try:
             dic, image = await generate_image(
-                prompt, width, height, model, safe, cached, nologo, enhance, private
+                prompt=prompt,
+                width=width,
+                height=height,
+                model=model,
+                safe=safe,
+                enhance=enhance,
             )
             time_taken = (datetime.datetime.now() - start).total_seconds()
             discord_logger.log_image_generation(

utils/image_gen_utils.py

@@ -42,10 +42,7 @@ async def generate_image(
     height: int = config.image_generation.defaults.height,
     model: str = None,  # Don't use config.MODELS[0] as default
     safe: bool = config.image_generation.defaults.safe,
-    cached: bool = config.image_generation.defaults.cached,
-    nologo: bool = config.image_generation.defaults.nologo,
     enhance: bool = config.image_generation.defaults.enhance,
-    private: bool = config.image_generation.defaults.private,
     seed: int = random.randint(0, 2**31 - 1),
     **kwargs,
 ):
@@ -56,21 +53,18 @@ async def generate_image(
             model = config.image_generation.fallback_model
 
     validate_model(model)
-    url: str = f"https://gen.pollinations.ai/image/{quote(prompt)}?seed={seed}&width={width}&height={height}&model={model}&safe={str(safe).lower()}&nologo={str(nologo).lower()}&enhance={str(enhance).lower()}&nofeed={str(private).lower()}&referer={quote(config.image_generation.referer)}"
+    url: str = f"https://gen.pollinations.ai/image/{quote(prompt)}?seed={seed}&width={width}&height={height}&model={model}&safe={str(safe).lower()}&enhance={str(enhance).lower()}"
+
     dic = {
         "prompt": prompt,
         "width": width,
         "height": height,
         "model": model,
         "safe": safe,
-        "cached": cached,
-        "nologo": nologo,
-        "enhance": enhance,
         "url": url,
+        "seed": seed,
     }
 
-    dic["seed"] = None if cached else seed
-
     headers = {
         "Authorization": f"Bearer {os.getenv('ENTER_TOKEN')}",
     }
@@ -109,19 +103,6 @@ async def generate_image(
 
                 try:
                     dic["nsfw"] = user_comment["has_nsfw_concept"]
-                    if (
-                        enhance
-                        or len(prompt)
-                        < config.image_generation.validation.max_enhanced_prompt_length
-                    ):
-                        enhance_prompt = user_comment["prompt"]
-                        if enhance_prompt == prompt:
-                            dic["enhanced_prompt"] = None
-                        else:
-                            enhance_prompt = enhance_prompt[
-                                : enhance_prompt.rfind("\n")
-                            ].strip()
-                            dic["enhanced_prompt"] = enhance_prompt
                 except Exception:
                     dic["nsfw"] = False