|
import discord |
|
import gradio_client |
|
import gradio as gr |
|
import os |
|
import threading |
|
|
|
|
|
jojogan = gradio_client.Client("akhaliq/JoJoGAN") |
|
|
|
|
|
class MyClient(discord.Client): |
|
async def on_ready(self): |
|
print('Logged on as', self.user) |
|
|
|
async def on_message(self, message): |
|
|
|
if message.author == self.user: |
|
return |
|
|
|
if message.content.find("!help") != -1: |
|
await message.reply("Use !jojo !disney !spidey or !sketch. Have fun!", mention_author=True) |
|
|
|
style = None |
|
if message.content.startswith('!jojo'): |
|
style = 'JoJo' |
|
if message.content.startswith('!disney'): |
|
style = 'Disney' |
|
if message.content.startswith('!spidey'): |
|
style = 'Spider-Verse' |
|
if message.content.startswith('!sketch'): |
|
style = 'sketch' |
|
|
|
if style: |
|
if message.attachments: |
|
attachment = message.attachments[0] |
|
im = jojogan.predict(attachment.url, style) |
|
await message.reply(f'Here is the {style} version of it', file=discord.File(im)) |
|
else: |
|
await message.channel.send("No attachments to be found...Can't animify dat! Try sending me an image π") |
|
|
|
DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None) |
|
intents = discord.Intents.default() |
|
intents.message_content = True |
|
client = MyClient(intents=intents) |
|
|
|
def run_bot(): |
|
client.run(DISCORD_TOKEN) |
|
|
|
threading.Thread(target=run_bot).start() |
|
|
|
def greet(name): |
|
return "Hello " + name + "!" |
|
|
|
demo = gr.Interface(fn=greet, inputs="text", outputs="text") |
|
demo.launch() |