Multiple AI image generation models

[surelygabriel]

hey i found your code really interesting so I looked through it, thus stumbling upon the image generation at line 113, and I decided to look at it. I found out that the website you're using has several models, and you could possibly make it so you can select one of your choice. I made a little code that should pretty much show you what I meant:

model_choices = {
    "anythingv3_0-pruned.ckpt [2700c435]": "AnimeM1",
    "meinamix_meinaV9.safetensors [2ec66ab0]": "AnimeM2",
    "meinamix_meinaV11.safetensors [b56ce717]": "AnimeM3",
    "Realistic_Vision_V4.0.safetensors [29a7afaa]": "RealisticM1",
    "anythingV5_PrtRE.safetensors [893e49b9]": "AnimeM4",
}
current_model = None

(...)



try:
    with open("currentmodel.txt", "r") as file:
        current_model = file.read().strip()
except FileNotFoundError:
    pass


(...)


(line 121, where your current model is realistic vision) "model": current_model,

(...)



@bot.command()
async def changemodel(ctx, model_name=None):
    if model_name is None:
        model_list = "\n".join(model_choices.values())
        await ctx.send(f"{ctx.author.mention}, please write ~changemodel and then put one of our available models:\n{model_list}")
    else:
        selected_model = get_model_name(model_name)

        if selected_model:
            # Update the model in generate_job
            global current_model
            current_model = selected_model

            # Save the updated model in the file
            with open("currentmodel.txt", "w") as file:
                file.write(current_model)

            await ctx.send(f"{ctx.author.mention}, the model has been updated to {model_name} ({selected_model}).")
        else:
            await ctx.send("Invalid model choice. Please choose a valid model.")