Refactor function signatures in imagine_cog.py and multi_imagine_cog.py

Zingzy · Mar 18, 2024 · c3776ca · c3776ca
1 parent 605f4d6
commit c3776ca
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 15 deletions.
diff --git a/Pipfile b/Pipfile
diff --git a/cogs/imagine_cog.py b/cogs/imagine_cog.py
@@ -224,7 +224,7 @@ async def model_autocomplete(self,
     @app_commands.guild_only()
     @app_commands.checks.cooldown(1, 15)
     @app_commands.describe(prompt="Imagine a prompt", height="Height of the image", width="Width of the image", negative="The things not to include in the image", cached="Removes the image seed", nologo="Remove the logo", enhance="Disables Prompt enhancing if set to False", private="Only you can see the generated Image if set to True")
-    async def imagine_command(self, interaction, prompt:str, model: str = "Swizz8", width:int = 1000, height:int = 1000, negative:Optional[str] = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False):
+    async def imagine_command(self, interaction, prompt:str, model: str = "Swizz8", width:int = 1000, height:int = 1000, negative:str|None = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False):
         await interaction.response.send_message(embed=discord.Embed(title="Generating Image", description="Please wait while we generate your image", color=discord.Color.blurple()), ephemeral=True)
 
         if len(prompt) > 1500:

diff --git a/cogs/multi_imagine_cog.py b/cogs/multi_imagine_cog.py
@@ -128,7 +128,7 @@ async def delete(self, interaction: discord.Interaction, button: discord.ui.Butt
     @app_commands.checks.cooldown(1, 30)
     @app_commands.guild_only()
     @app_commands.describe(prompt="Imagine a prompt", height="Height of the image", width="Width of the image", negative="The things not to include in the image", cached="Removes the image seed", nologo="Remove the logo", enhance="Disables Prompt enhancing if set to False", private="Only you can see the generated Image if set to True")
-    async def multiimagine_command(self, interaction, prompt:str, width:int = 1000, height:int = 1000, negative:Optional[str] = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False):
+    async def multiimagine_command(self, interaction, prompt:str, width:int = 1000, height:int = 1000, negative:str|None = None, cached:bool = False, nologo:bool = False, enhance:bool = True, private:bool = False):
 
         await interaction.response.send_message(embed=discord.Embed(title="Generating Image", description="Please wait while we generate your image", color=discord.Color.blurple()), ephemeral=True)
 

diff --git a/utils.py b/utils.py
@@ -6,7 +6,6 @@
 from pymongo import MongoClient
 import sys
 import itertools
-from typing import Optional, Any
 
 client = MongoClient(MONGODB_URI)
 
@@ -144,7 +143,7 @@ async def generate_image(
     width: int = 500,
     height: int = 500,
     model: str = "turbo",
-    negative: Optional[str] = None,
+    negative: str|None = None,
     cached: bool = False,
     nologo: bool = False,
     enhance: bool = True,