client.py

import discord
from discord import app_commands
from discord.ext import commands

from PIL import Image
from typing import Literal, Optional
from SECRET import token 
from image_converter import image_to_ascii as img_to_ascii


intents = discord.Intents.default()
intents.message_content = True

bot = commands.Bot(command_prefix='!', intents=intents)

@bot.event
async def on_ready():
    print(f'Bot is up!\nWe have logged in as {bot.user}')

@bot.event
async def on_message(message):
    if message.author == bot.user:
        return

    if message.content.startswith('$ping'):
        await message.channel.send('pong!')

    if "best song" in message.content:
        await message.channel.send('<https://www.youtube.com/watch?v=dQw4w9WgXcQ>')
    
    if message.content == 'pog':
        await message.channel.send('<:pog:1284563646927736894>')

    await bot.process_commands(message)

# Sync function/command from https://about.abstractumbra.dev/discord.py/2023/01/29/sync-command-example.html
@bot.command()
@commands.guild_only()
@commands.is_owner()
async def sync(ctx: commands.Context, guilds: commands.Greedy[discord.Object], spec: Optional[Literal["~", "*", "^"]] = None) -> None:
    if not guilds:
        if spec == "~":
            synced = await ctx.bot.tree.sync(guild=ctx.guild)
        elif spec == "*":
            ctx.bot.tree.copy_global_to(guild=ctx.guild)
            synced = await ctx.bot.tree.sync(guild=ctx.guild)
        elif spec == "^":
            ctx.bot.tree.clear_commands(guild=ctx.guild)
            await ctx.bot.tree.sync(guild=ctx.guild)
            synced = []
        else:
            synced = await ctx.bot.tree.sync()

        await ctx.send(
            f"Synced {len(synced)} commands {'globally' if spec is None else 'to the current guild.'}"
        )
        return

    ret = 0
    for guild in guilds:
        try:
            await ctx.bot.tree.sync(guild=guild)
        except discord.HTTPException:
            pass
        else:
            ret += 1

    await ctx.send(f"Synced the tree to {ret}/{len(guilds)}.")

def is_valid_image(file_name):
    try:
        with Image.open(file_name) as img:
            img.verify()
            return True
    except (IOError, SyntaxError):
        return False

@bot.tree.command()
@app_commands.allowed_installs(guilds=True, users=True) # users only, no guilds for install
@app_commands.allowed_contexts(guilds=True, dms=True, private_channels=True) # all allowed
async def image_to_ascii(interaction: discord.Interaction, format: str, image: discord.Attachment):
    await interaction.response.defer()

    # save the image so we can process it
    await image.save('image.png')

    # check if the file is an image
    if not is_valid_image("image.png"):
        await interaction.followup.send("File needs to be an image.")
        return

    # check if format is 'file'
    if format == "file":
        def check(m):
            try:
                _ = int(m.content)
                return m.author == interaction.user and m.channel == interaction.channel
            except:
                return False

        await interaction.followup.send("What should the width be? (character length, example: 50)")
        resp = await bot.wait_for("message", check=check, timeout=60.0)
        width = int(resp.content)

        ascii = img_to_ascii('image.png', width=width)

        with open("ascii_art.txt", "rb") as file:
            await interaction.followup.send(file=discord.File(file, "ascii_art.txt"))
    elif format == "message":
        ascii = img_to_ascii('image.png')

        # send the message
        await interaction.followup.send(ascii)
    else:
        await interaction.followup.send("Format needs to be 'file' or 'message'. Please try again.")

bot.run(token)