Update app.py

app.py

@@ -227,19 +227,97 @@ async def on_message(message:discord.Message):
             embed.set_image(url=f"attachment://{imgn}.png")
             await e.edit(embed=embed, attachments=[file])
 
-@bot.slash_command()
-async def setup(ctx:discord.Interaction):
+@bot.command(description="Start a chatbot conversation in the channel.")
+async def setup(ctx:discord.ApplicationContext):
     with open(f"{ctx.guild.name}.guild", "a") as f:
         f.write(f"{ctx.channel.id}\n")
-    embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{ctx.channel.id}>")
+    embed = discord.Embed(title="setup", description=f"You can now chat with the bot in <#{ctx.channel.id}>")
     await ctx.respond(embed=embed)
-@bot.slash_command()
-async def remove(ctx:discord.Interaction):
-    with open(f"{ctx.guild.name}.guild", "a") as f:
-        f.write(f"{ctx.channel.id}\n")
-    embed = discord.Embed(title="Success!", description=f"You can now chat with the bot in <#{ctx.channel.id}>")
+
+@bot.command(name="reset", description="Reset the chat history of a channel.")
+@discord.commands.option(
+    "sys", 
+    description="The new system prompt.",
+    required=False,
+    default=sysrp
+)
+async def reset(ctx:discord.ApplicationContext, sys:str):
+    with open(f"data/{ctx.guild.name}/{ctx.channel.name}", "w") as f:
+        f.write(f"{sot}system: {sys}{eot}")
+    if sys == sysrp:
+        embed = discord.Embed(title="reset", description=f"Reset the channel history with default system prompt.")
+    else:
+        embed = discord.Embed(title="reset", description=f"Reset the channel history with custom system prompt.\n```\n{sys}```")
     await ctx.respond(embed=embed)
 
+@bot.command(name="create", description="Create an image using a specified model. Defaults to sdxl-turbo.")
+@discord.commands.option(
+    "prompt",
+    description="The prompt for the image.",
+    required=True,
+)
+@discord.commands.option(
+    "model",
+    description="The model to use.",
+    required=False,
+    choices=["Stable Diffusion 2-1", "stable-diffusion-xl-base-1.0", "sdxl-turbo", "sd-turbo"],
+    default="sdxl-turbo"
+)
+@discord.commands.option(
+    "negative",
+    description="The negative prompt for the image. Defaults to None.",
+    required=False,
+    default=""
+)
+@discord.commands.option(
+    "steps",
+    description="The amount of steps for image generation. Range: 1 - 70.",
+    required=False,
+    default=35
+)
+async def create(ctx:discord.ApplicationContext, prompt:str, negative:str, steps:int, model:str):
+    refine = True
+    await ctx.defer()
+    st = time.time()
+    imgn = random.choice(['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AB', 'AC', 'AD', 'AE', 'AF', 'AG', 'AH', 'AI', 'AJ', 'AK', 'AL', 'AM', 'AN', 'AO', 'AP', 'AQ', 'AR', 'AS', 'AT', 'AU', 'AV', 'AW', 'AX', 'AY', 'AZ', 'BA', 'BB', 'BC', 'BD', 'BE', 'BF', 'BG', 'BH', 'BI', 'BJ', 'BK', 'BL', 'BM', 'BN', 'BO', 'BP', 'BQ', 'BR', 'BS', 'BT', 'BU', 'BV', 'BW', 'BX', 'BY', 'BZ', 'CA', 'CB', 'CC', 'CD', 'CE', 'CF', 'CG', 'CH', 'CI', 'CJ', 'CK', 'CL', 'CM', 'CN', 'CO', 'CP', 'CQ', 'CR', 'CS', 'CT', 'CU', 'CV', 'CW', 'CX', 'CY', 'CZ', 'DA', 'DB', 'DC', 'DD', 'DE', 'DF', 'DG', 'DH', 'DI', 'DJ', 'DK', 'DL', 'DM', 'DN', 'DO', 'DP', 'DQ', 'DR', 'DS', 'DT', 'DU', 'DV', 'DW', 'DX', 'DY', 'DZ', 'EA', 'EB', 'EC', 'ED', 'EE', 'EF', 'EG', 'EH', 'EI', 'EJ', 'EK', 'EL', 'EM', 'EN', 'EO', 'EP', 'EQ', 'ER', 'ES', 'ET', 'EU', 'EV', 'EW', 'EX', 'EY', 'EZ', 'FA', 'FB', 'FC', 'FD', 'FE', 'FF', 'FG', 'FH', 'FI', 'FJ', 'FK', 'FL', 'FM', 'FN', 'FO', 'FP', 'FQ', 'FR', 'FS', 'FT', 'FU', 'FV', 'FW', 'FX', 'FY', 'FZ', 'GA', 'GB', 'GC', 'GD', 'GE', 'GF', 'GG', 'GH', 'GI', 'GJ', 'GK', 'GL', 'GM', 'GN', 'GO', 'GP', 'GQ', 'GR', 'GS', 'GT', 'GU', 'GV', 'GW', 'GX', 'GY', 'GZ', 'HA', 'HB', 'HC', 'HD', 'HE', 'HF', 'HG', 'HH', 'HI', 'HJ', 'HK', 'HL', 'HM', 'HN', 'HO', 'HP', 'HQ', 'HR', 'HS', 'HT', 'HU', 'HV', 'HW', 'HX', 'HY', 'HZ', 'IA', 'IB', 'IC', 'ID', 'IE', 'IF', 'IG', 'IH', 'II', 'IJ', 'IK', 'IL', 'IM', 'IN', 'IO', 'IP', 'IQ', 'IR', 'IS', 'IT', 'IU', 'IV', 'IW', 'IX', 'IY', 'IZ', 'JA', 'JB', 'JC', 'JD', 'JE', 'JF', 'JG', 'JH', 'JI', 'JJ', 'JK', 'JL', 'JM', 'JN', 'JO', 'JP', 'JQ', 'JR', 'JS', 'JT', 'JU', 'JV', 'JW', 'JX', 'JY', 'JZ', 'KA', 'KB', 'KC', 'KD', 'KE', 'KF', 'KG', 'KH', 'KI', 'KJ', 'KK', 'KL', 'KM', 'KN', 'KO', 'KP', 'KQ', 'KR', 'KS', 'KT', 'KU', 'KV', 'KW', 'KX', 'KY', 'KZ', 'LA', 'LB', 'LC', 'LD', 'LE', 'LF', 'LG', 'LH', 'LI', 'LJ', 'LK', 'LL', 'LM', 'LN', 'LO', 'LP', 'LQ', 'LR', 'LS', 'LT', 'LU', 'LV', 'LW', 'LX', 'LY', 'LZ', 'MA', 'MB', 'MC', 'MD', 'ME', 'MF', 'MG', 'MH', 'MI', 'MJ', 'MK', 'ML', 'MM', 'MN', 'MO', 'MP', 'MQ', 'MR', 'MS', 'MT', 'MU', 'MV', 'MW', 'MX', 'MY', 'MZ', 'NA', 'NB', 'NC', 'ND', 'NE', 'NF', 'NG', 'NH', 'NI', 'NJ', 'NK', 'NL', 'NM', 'NN', 'NO', 'NP', 'NQ', 'NR', 'NS', 'NT', 'NU', 'NV', 'NW', 'NX', 'NY', 'NZ', 'OA', 'OB', 'OC', 'OD', 'OE', 'OF', 'OG', 'OH', 'OI', 'OJ', 'OK', 'OL', 'OM', 'ON', 'OO', 'OP', 'OQ', 'OR', 'OS', 'OT', 'OU', 'OV', 'OW', 'OX', 'OY', 'OZ', 'PA', 'PB', 'PC', 'PD', 'PE', 'PF', 'PG', 'PH', 'PI', 'PJ', 'PK', 'PL', 'PM', 'PN', 'PO', 'PP', 'PQ', 'PR', 'PS', 'PT', 'PU', 'PV', 'PW', 'PX', 'PY', 'PZ', 'QA', 'QB', 'QC', 'QD', 'QE', 'QF', 'QG', 'QH', 'QI', 'QJ', 'QK', 'QL', 'QM', 'QN', 'QO', 'QP', 'QQ', 'QR', 'QS', 'QT', 'QU', 'QV', 'QW', 'QX', 'QY', 'QZ', 'RA', 'RB', 'RC', 'RD', 'RE', 'RF', 'RG', 'RH', 'RI', 'RJ', 'RK', 'RL', 'RM', 'RN', 'RO', 'RP', 'RQ', 'RR', 'RS', 'RT', 'RU', 'RV', 'RW', 'RX', 'RY', 'RZ', 'SA', 'SB', 'SC', 'SD', 'SE', 'SF', 'SG', 'SH', 'SI', 'SJ', 'SK', 'SL', 'SM', 'SN', 'SO', 'SP', 'SQ', 'SR', 'SS', 'ST', 'SU', 'SV', 'SW', 'SX', 'SY', 'SZ', 'TA', 'TB', 'TC', 'TD', 'TE', 'TF', 'TG', 'TH', 'TI', 'TJ', 'TK', 'TL', 'TM', 'TN', 'TO', 'TP', 'TQ', 'TR', 'TS', 'TT', 'TU', 'TV', 'TW', 'TX', 'TY', 'TZ', 'UA', 'UB', 'UC', 'UD', 'UE', 'UF', 'UG', 'UH', 'UI', 'UJ', 'UK', 'UL', 'UM', 'UN', 'UO', 'UP', 'UQ', 'UR', 'US', 'UT', 'UU', 'UV', 'UW', 'UX', 'UY', 'UZ', 'VA', 'VB', 'VC', 'VD', 'VE', 'VF', 'VG', 'VH', 'VI', 'VJ', 'VK', 'VL', 'VM', 'VN', 'VO', 'VP', 'VQ', 'VR', 'VS', 'VT', 'VU', 'VV', 'VW', 'VX', 'VY', 'VZ', 'WA', 'WB', 'WC', 'WD', 'WE', 'WF', 'WG', 'WH', 'WI', 'WJ', 'WK', 'WL', 'WM', 'WN', 'WO', 'WP', 'WQ', 'WR', 'WS', 'WT', 'WU', 'WV', 'WW', 'WX', 'WY', 'WZ', 'XA', 'XB', 'XC', 'XD', 'XE', 'XF', 'XG', 'XH', 'XI', 'XJ', 'XK', 'XL', 'XM', 'XN', 'XO', 'XP', 'XQ', 'XR', 'XS', 'XT', 'XU', 'XV', 'XW', 'XX', 'XY', 'XZ', 'YA', 'YB', 'YC', 'YD', 'YE', 'YF', 'YG', 'YH', 'YI', 'YJ', 'YK', 'YL', 'YM', 'YN', 'YO', 'YP', 'YQ', 'YR', 'YS', 'YT', 'YU', 'YV', 'YW', 'YX', 'YY', 'YZ', 'ZA', 'ZB', 'ZC', 'ZD', 'ZE', 'ZF', 'ZG', 'ZH', 'ZI', 'ZJ', 'ZK', 'ZL', 'ZM', 'ZN', 'ZO', 'ZP', 'ZQ', 'ZR', 'ZS', 'ZT', 'ZU', 'ZV', 'ZW', 'ZX', 'ZY', 'ZZ'])
+    if not 1 <= steps <= 70:
+        embed = discord.Embed(title="Invalid Step Count", description=f"Please enter a number between 1 and 70.")
+        await ctx.respond(embed=embed)
+        return
+    if model == "Stable Diffusion 2-1":
+        image, m = (await SD.text_to_image(prompt=prompt, negative_prompt=negative, num_inference_steps=steps), "Stable Diffusion 2-1")
+    elif model == "stable-diffusion-xl-base-1.0":
+        image, m = (await SDXL.text_to_image(prompt=prompt, negative_prompt=negative, num_inference_steps=steps), "stable-diffusion-xl-base-1.0")
+    elif model == "sdxl-turbo":
+        image, m = (await SDXLT.text_to_image(prompt=prompt, negative_prompt=negative, num_inference_steps=steps, guidance_scale=0.0), "sdxl-turbo")
+    elif model == "sd-turbo":
+        image, m = (await SDT.text_to_image(prompt=prompt, negative_prompt=negative, num_inference_steps=steps, guidance_scale=0.0), "sd-turbo")
+    elif model == "Proteus v0.2":
+        image, m = (await PT.text_to_image(prompt=prompt, negative_prompt=negative, num_inference_steps=steps), "Proteus v0.2")
+    else:
+        raise NotImplementedError(f"Model {imageModel} not found. Report this to @araeyn if this keeps happening.")
+    embed = discord.Embed(title="create", description=f"```\n{prompt}\n```")
+    image.save(f"{imgn}.png")
+    file = discord.File(f"{imgn}.png", filename=f"{imgn}.png", description=prompt)
+    embed.set_image(url=f"attachment://{imgn}.png")
+    gt = time.time()
+    image, r = (await RF.image_to_image(f"{imgn}.png", num_inference_steps=35, prompt=prompt, negative_prompt=negative), "stable-diffusion-xl-refiner-1.0")
+    rt = time.time()
+    if refine:
+        embed.set_footer(
+            text=
+            f"Image generation model is {m}. Refiner model is {r}. Took {round((gt - st) * 100) / 100} seconds to generate. Took {round((rt - gt) * 100) / 100} seconds to refine."
+        )
+    else:
+        embed.set_footer(
+            text=
+            f"Image generation model is {m}. Took {round((gt - st) * 100) / 100} seconds to generate."
+        )
+    image.save(f"{imgn}.png")
+    file = discord.File(f"{imgn}.png", filename=f"{imgn}.png")
+    embed.set_image(url=f"attachment://{imgn}.png")
+    await ctx.respond(embed=embed, file=file)
+
 def greet(name, intensity):
     return "Hello, " + name + "!" * int(intensity)