[decorators] re-adding The Painter

app.py

@@ -24,10 +24,12 @@ import shutil # for doing image movement magic
 
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
+DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
+
 df = Client("huggingface-projects/IF", DFIF_TOKEN)
+jojogan = gradio_client.Client("akhaliq/JoJoGAN")
 
 
-DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
 intents = discord.Intents.default()
 intents.message_content = True
 
@@ -42,6 +44,26 @@ async def on_ready():
     bot.log_channel = bot.get_channel(1100458786826747945) # 1100458786826747945 = bot-test, 1107006391547342910 = lunarbot server
 
 #---------------------------------------------------------------------------------------------------------------------------------------------- 
+# jojo
+@bot.command
+async def jojo(ctx): # needs safety, easier to do safety function since they'll all have similar constraints
+    
+    await ctx.message.add_reaction('👍')
+    thread = await ctx.message.create_thread(name=f'{ctx.author} Jojo Thread ')
+    
+    style = None
+    style = 'JoJo'
+
+    await thread.send(f'{ctx.author.mention}Generating JoJo image in thread, yare yare daze...')
+
+    if style:
+        if ctx.message.attachments:
+            attachment = ctx.message.attachments[0]
+            im = jojogan.predict(attachment.url, style)
+            await ctx.message.reply(f'Here is the {style} version of it', file=discord.File(im))
+        else:
+            await ctx.message.channel.send("No attachments to be found...Can't animify dat! Try sending me an image 😉")
+#----------------------------------------------------------------------------------------------------------------------------------------------     
 # Stage 1
 @bot.command()
 async def deepfloydif(ctx, *, prompt: str):