Duplicate from osanseviero/discord

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,14 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY ./main.py /code/app/main.py
+
+RUN --mount=type=secret,id=DISCORD_PAINTER_TOKEN,mode=0444,required=true \
+   cat /run/secrets/DISCORD_PAINTER_TOKEN > /test
+
+CMD ["uvicorn", "APP.main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -0,0 +1,11 @@
+---
+title: Discord
+emoji: 👁
+colorFrom: purple
+colorTo: gray
+sdk: docker
+pinned: false
+duplicated_from: osanseviero/discord
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

main.py

@@ -0,0 +1,58 @@
+import discord
+import os
+
+import gradio_client as gr 
+
+from fastapi import FastAPI
+
+# Get Gradio client
+jojogan = gr.Client("akhaliq/JoJoGAN")
+
+# Set up discord bot 
+class MyClient(discord.Client):
+    async def on_ready(self):
+        print('Logged on as', self.user)
+
+    async def on_message(self, message):
+        # don't respond to ourselves
+        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("DISCORD_PAINTER_TOKEN", None)
+intents = discord.Intents.default()
+intents.message_content = True
+client = MyClient(intents=intents)
+
+
+app = FastAPI()
+
+async def run_bot():
+  client.run(DISCORD_TOKEN)
+
+# Hacky, keep FastAPI running so Space is running
+@app.get("/")
+async def root():
+    return {"message": "Hello World"}
+
+asyncio.create_task(run_bot)

requirements.txt

@@ -0,0 +1,9 @@
+fastapi==0.74.*
+requests==2.27.*
+sentencepiece==0.1.*
+torch==1.11.*
+transformers==4.*
+uvicorn[standard]==0.17.*
+git+https://github.com/Rapptz/discord.py.git
+Pillow 
+gradio_client