Update app.py

app.py

@@ -19,7 +19,7 @@ event = Event()
 
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
 HF_TOKEN = os.getenv("HF_TOKEN")
-API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
+API_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
 headers = {"Authorization": "Bearer " + HF_TOKEN}
 
 def ask(input_text):
@@ -59,7 +59,7 @@ async def uptime(ctx):
     embed.add_field(name="Uptime", value=f"{days} days, {hours} hours, {minutes} minutes, {seconds} seconds", inline=False)
     embed.set_footer(text="Created by Cosmos")
 
-    await ctx.send(embed=embed)
+    await ctx.reply(embed=embed)
     
 @bot.command()
 async def ai(ctx, *, input_text: str):
@@ -68,24 +68,22 @@ async def ai(ctx, *, input_text: str):
       result = ask(input_text)
         
     # Reply with the embed
-    await ctx.send(result)
+    await ctx.reply(result)
 
-def query(prompt):
-    payload = prompt
-    response = requests.post(API_URL, headers=headers, json=payload)
+def query(payload):
+    response = requests.post(API_URL_imgen, headers=headers_imgen, json=payload)
     return response.content
 
 @bot.command()
 async def img(ctx, *, prompt: str):
     # Query the API to generate the image
     async with ctx.typing():
-      image_data = query(prompt)
-
-      # Encode image data as base64
-      image_b64 = base64.b64encode(image_data).decode('utf-8')
-
+      image_bytes = query({
+                "inputs": prompt,
+            })
+      image_file = discord.File(io.BytesIO(image_bytes), filename='generated_image.png')
     # Send the image as a file
-    await ctx.send(file=discord.File(base64.b64decode(image_b64), filename="generated_image.png"))
+    await ctx.reply(file=image_file)
 
 
 @bot.command()
@@ -103,7 +101,7 @@ async def verse(ctx):
     # Create an embed
     embed = discord.Embed(title="Random Bible Verse", description=passage, color=discord.Color.blue())
     embed.set_footer(text="Created by Cosmos")
-    await ctx.send(embed=embed)
+    await ctx.reply(embed=embed)
 
 @bot.command()
 @commands.has_permissions(kick_members=True)
@@ -111,11 +109,11 @@ async def kick(ctx, member: discord.Member, *, reason: Optional[str] = "No reaso
     """Kicks a member."""
     try:
         await member.kick(reason=reason)
-        await ctx.send(f"{member.mention} has been kicked. Reason: {reason}")
+        await ctx.reply(f"{member.mention} has been kicked. Reason: {reason}")
     except discord.Forbidden:
-        await ctx.send("I don't have permission to kick members.")
+        await ctx.reply("I don't have permission to kick members.")
     except discord.HTTPException:
-        await ctx.send("An error occurred while trying to kick the member. Please try again later.")
+        await ctx.reply("An error occurred while trying to kick the member. Please try again later.")
 
 @bot.command()
 @commands.has_permissions(ban_members=True)
@@ -123,11 +121,11 @@ async def ban(ctx, member: discord.Member, *, reason: Optional[str] = "No reason
     """Bans a member."""
     try:
         await member.ban(reason=reason)
-        await ctx.send(f"{member.mention} has been banned. Reason: {reason}")
+        await ctx.reply(f"{member.mention} has been banned. Reason: {reason}")
     except discord.Forbidden:
-        await ctx.send("I don't have permission to ban members.")
+        await ctx.reply("I don't have permission to ban members.")
     except discord.HTTPException:
-        await ctx.send("An error occurred while trying to ban the member. Please try again later.")
+        await ctx.reply("An error occurred while trying to ban the member. Please try again later.")
 
 
 @bot.command()
@@ -145,7 +143,7 @@ async def cmds(ctx):
     embed.add_field(name="Bot Information", value=bot_info, inline=False)
     embed.set_footer(text="Created by Cosmos")
     
-    await ctx.send(embed=embed)
+    await ctx.reply(embed=embed)
 
 async def update_status():
     await bot.wait_until_ready()  # Wait until the bot is fully ready
@@ -231,7 +229,7 @@ async def search(ctx, *, query: str):
         embed = discord.Embed(title="Search Results", description="Unable to fetch search results.", color=discord.Color.red())
 
     embed.set_footer(text="Created by Cosmos")
-    await ctx.send(embed=embed)
+    await ctx.reply(embed=embed)
 
 @bot.command()
 @commands.has_permissions(manage_messages=True)
@@ -239,21 +237,21 @@ async def purge(ctx, limit: int):
     """Removes N amount of messages."""
     try:
         await ctx.channel.purge(limit=limit + 1)  # Add 1 to include the command message itself
-        await ctx.send(f"{limit} messages have been purged.")
+        await ctx.reply(f"{limit} messages have been purged.")
     except discord.Forbidden:
-        await ctx.send("I don't have permission to manage messages.")
+        await ctx.reply("I don't have permission to manage messages.")
     except discord.HTTPException:
-        await ctx.send("An error occurred while trying to purge messages. Please try again later.")
+        await ctx.reply("An error occurred while trying to purge messages. Please try again later.")
 
 
 @bot.event
 async def on_command_error(ctx, error):
     if isinstance(error, commands.CommandNotFound):
-        await ctx.send("Sorry, I couldn't find that command. Use `$cmds` to see the list of available commands.")
+        await ctx.reply("Sorry, I couldn't find that command. Use `$cmds` to see the list of available commands.")
     elif isinstance(error, commands.MissingRequiredArgument):
-        await ctx.send("Oops! Looks like you're missing some required arguments.")
+        await ctx.reply("Oops! Looks like you're missing some required arguments.")
     elif isinstance(error, commands.CheckFailure):
-        await ctx.send("You do not have the permissions to execute this command.")
+        await ctx.reply("You do not have the permissions to execute this command.")
     else:
         # Log the error to console or your logging system
         print(f"An error occurred: {error}")