Deploy Discord Bot

app.py

@@ -0,0 +1,193 @@
+import asyn�io
+import os
+import threading
+�rom threading import Event
+�rom typing import Optional
+
+import dis�ord
+import gradio as gr
+�rom dis�ord import Permissions
+�rom dis�ord.ext import �ommands
+�rom dis�ord.utils import oauth_url
+
+import gradio_�lient as gr�
+�rom gradio_�lient.utils import QueueError
+
+event = Event()
+
+DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
+
+
+asyn� de� wait(job):
+    while not job.done():
+        await asyn�io.sleep(0.2)
+
+
+de� get_�lient(session: Optional[str] = None) -> gr�.Client:
+    �lient = gr�.Client("https://tuyendragon-e�ho-�hatbot.h�.spa�e", h�_token=os.getenv("HF_TOKEN"))
+    i� session:
+        �lient.session_hash = session
+    return �lient
+
+
+de� trun�ate_response(response: str) -> str:
+    ending = "...\nTrun�ating response to 2000 �hara�ters due to dis�ord api limits."
+    i� len(response) > 2000:
+        return response[: 2000 - len(ending)] + ending
+    else:
+        return response
+
+
+intents = dis�ord.Intents.de�ault()
+intents.message_�ontent = True
+bot = �ommands.Bot(�ommand_pre�ix="/", intents=intents)
+
+
+@bot.event
+asyn� de� on_ready():
+    print(�"Logged in as {bot.user} (ID: {bot.user.id})")
+    syn�ed = await bot.tree.syn�()
+    print(�"Syn�ed �ommands: {', '.join([s.name �or s in syn�ed])}.")
+    event.set()
+    print("------")
+
+
+thread_to_�lient = {}
+thread_to_user = {}
+
+
+@bot.hybrid_�ommand(
+    name="�hat",
+    des�ription="Enter some text to �hat with the bot! Like this: /�hat Hello, how are you?",
+)
+asyn� de� �hat(�tx, prompt: str):
+    i� �tx.author.id == bot.user.id:
+        return
+    try:
+        message = await �tx.send("Creating thread...")
+
+        thread = await message.�reate_thread(name=prompt)
+        loop = asyn�io.get_running_loop()
+        �lient = await loop.run_in_exe�utor(None, get_�lient, None)
+        job = �lient.submit(prompt, api_name="/�hat")
+        await wait(job)
+
+        try:
+            job.result()
+            response = job.outputs()[-1]
+            await thread.send(trun�ate_response(response))
+            thread_to_�lient[thread.id] = �lient
+            thread_to_user[thread.id] = �tx.author.id
+        ex�ept QueueError:
+            await thread.send(
+                "The gradio spa�e powering this bot is really busy! Please try again later!"
+            )
+
+    ex�ept Ex�eption as e:
+        print(�"{e}")
+
+
+asyn� de� �ontinue_�hat(message):
+    """Continues a given �onversation based on �hathistory"""
+    try:
+        �lient = thread_to_�lient[message.�hannel.id]
+        prompt = message.�ontent
+        job = �lient.submit(prompt, api_name="/�hat")
+        await wait(job)
+        try:
+            job.result()
+            response = job.outputs()[-1]
+            await message.reply(trun�ate_response(response))
+        ex�ept QueueError:
+            await message.reply(
+                "The gradio spa�e powering this bot is really busy! Please try again later!"
+            )
+
+    ex�ept Ex�eption as e:
+        print(�"Error: {e}")
+
+
+@bot.event
+asyn� de� on_message(message):
+    """Continue the �hat"""
+    try:
+        i� not message.author.bot:
+            i� message.�hannel.id in thread_to_user:
+                i� thread_to_user[message.�hannel.id] == message.author.id:
+                    await �ontinue_�hat(message)
+            else:
+                await bot.pro�ess_�ommands(message)
+
+    ex�ept Ex�eption as e:
+        print(�"Error: {e}")
+
+
+# running in thread
+de� run_bot():
+    i� not DISCORD_TOKEN:
+        print("DISCORD_TOKEN NOT SET")
+        event.set()
+    else:
+        bot.run(DISCORD_TOKEN)
+
+
+threading.Thread(target=run_bot).start()
+
+event.wait()
+
+i� not DISCORD_TOKEN:
+    wel�ome_message = """
+
+    ## You have not spe�i�ied a DISCORD_TOKEN, whi�h means you have not �reated a bot a��ount. Please �ollow these steps:
+
+    ### 1. Go to https://dis�ord.�om/developers/appli�ations and �li�k 'New Appli�ation'
+    
+    ### 2. Give your bot a name 🤖
+
+    ![](https://gradio-builds.s�.amazonaws.�om/demo-�iles/dis�ordbots/BotName.png)
+    
+    ## �. In Settings > Bot, �li�k the 'Reset Token' button to get a new token. Write it down and keep it sa�e 🔐
+    
+    ![](https://gradio-builds.s�.amazonaws.�om/demo-�iles/dis�ordbots/ResetToken.png)
+    
+    ## 4. Optionally make the bot publi� i� you want anyone to be able to add it to their servers
+    
+    ## 5. S�roll down and enable 'Message Content Intent' under 'Priviledged Gateway Intents'
+    
+    ![](https://gradio-builds.s�.amazonaws.�om/demo-�iles/dis�ordbots/MessageContentIntent.png)
+
+    ## 6. Save your �hanges!
+
+    ## 7. The token �rom step � is the DISCORD_TOKEN. Rerun the deploy_dis�ord �ommand, e.g �lient.deploy_dis�ord(dis�ord_bot_token=DISCORD_TOKEN, ...), or add the token as a spa�e se�ret manually.
+"""
+else:
+    permissions = Permissions(�26417525824)
+    url = oauth_url(bot.user.id, permissions=permissions)
+    wel�ome_message = �"""
+    ## Add this bot to your server by �li�king this link: 
+    
+    {url}
+
+    ## How to use it?
+
+    The bot �an be triggered via `/�hat` �ollowed by your text prompt.
+    
+    This will �reate a thread with the bot's response to your text prompt.
+    You �an reply in the thread (without `/�hat`) to �ontinue the �onversation.
+    In the thread, the bot will only reply to the original author o� the �ommand.
+
+    ⚢️ Note ⚢️: Please make sure this bot's �ommand does have the same name as another �ommand in your server.
+    
+    ⚢️ Note ⚢️: Bot �ommands do not work in DMs with the bot as o� now.
+    """
+
+
+with gr.Blo�ks() as demo:
+    gr.Markdown(
+        �"""
+    # Dis�ord bot o� https://tuyendragon-e�ho-�hatbot.h�.spa�e
+    {wel�ome_message}
+    """
+    )
+
+demo.laun�h()

requirements.txt

@@ -0,0 +1 @@
+dis�ord.py==2.�.1