Update app.py

app.py

@@ -6,16 +6,16 @@ from typing import Optional
 
 import discord
 import gradio as gr
+import gradio_client as grc
 from discord import Permissions
 from discord.ext import commands
 from discord.utils import oauth_url
-
-import gradio_client as grc
 from gradio_client.utils import QueueError
 
 event = Event()
 
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
+HF_TOKEN = os.getenv("HF_TOKEN")
 
 
 async def wait(job):
@@ -24,7 +24,7 @@ async def wait(job):
 
 
 def get_client(session: Optional[str] = None) -> grc.Client:
-    client = grc.Client("https://wop-xxx-opengpt.hf.space/", hf_token=os.getenv("HF_TOKEN"))
+    client = grc.Client("https://wop-xxx-opengpt.hf.space/", hf_token=HF_TOKEN)
     if session:
         client.session_hash = session
     return client
@@ -46,80 +46,15 @@ bot = commands.Bot(command_prefix="$", intents=intents)
 @bot.event
 async def on_ready():
     print(f"Logged in as {bot.user} (ID: {bot.user.id})")
-    synced = await bot.tree.sync()
-    print(f"Synced commands: {', '.join([s.name for s in synced])}.")
     event.set()
     print("------")
 
 
-thread_to_client = {}
-thread_to_user = {}
-
-
-@bot.hybrid_command(
-    name="chat",
-    description="Enter some text to chat with the bot! Like this: $chat Hello, how are you?",
-)
-async def chat(ctx, prompt: str):
-    if ctx.author.id == bot.user.id:
-        return
-    try:
-        message = await ctx.send("Creating thread...")
-
-        thread = await message.create_thread(name=prompt)
-        loop = asyncio.get_running_loop()
-        client = await loop.run_in_executor(None, get_client, None)
-        job = client.submit(prompt, 0.9, 2048, 0.9, 1.2, api_name="/chat")
-        await wait(job)
-
-        try:
-            job.result()
-            response = job.outputs()[-1]
-            await thread.send(truncate_response(response))
-            thread_to_client[thread.id] = client
-            thread_to_user[thread.id] = ctx.author.id
-        except QueueError:
-            await thread.send(
-                "The gradio space powering this bot is really busy! Please try again later!"
-            )
-
-    except Exception as e:
-        print(f"{e}")
-
-
-async def continue_chat(message):
-    """Continues a given conversation based on chathistory"""
-    try:
-        client = thread_to_client[message.channel.id]
-        prompt = message.content
-        job = client.submit(prompt, 0.9, 2048, 0.9, 1.2, api_name="/chat")
-        await wait(job)
-        try:
-            job.result()
-            response = job.outputs()[-1]
-            await message.reply(truncate_response(response))
-        except QueueError:
-            await message.reply(
-                "The gradio space powering this bot is really busy! Please try again later!"
-            )
-
-    except Exception as e:
-        print(f"Error: {e}")
-
-
 @bot.event
 async def on_message(message):
-    """Continue the chat"""
-    try:
-        if not message.author.bot:
-            if message.channel.id in thread_to_user:
-                if thread_to_user[message.channel.id] == message.author.id:
-                    await continue_chat(message)
-            else:
-                await bot.process_commands(message)
-
-    except Exception as e:
-        print(f"Error: {e}")
+    if message.channel.id == 1210559547874222083 and not message.author.bot:
+        await message.channel.send(f"Hello {message.author.mention}!")
+    await bot.process_commands(message)
 
 
 # running in thread
@@ -137,13 +72,10 @@ event.wait()
 
 if not DISCORD_TOKEN:
     welcome_message = """
-
     ## You have not specified a DISCORD_TOKEN, which means you have not created a bot account. Please follow these steps:
-
     ### 1. Go to https://discord.com/developers/applications and click 'New Application'
     
     ### 2. Give your bot a name 🤖
-
     ![](https://gradio-builds.s3.amazonaws.com/demo-files/discordbots/BotName.png)
     
     ## 3. In Settings > Bot, click the 'Reset Token' button to get a new token. Write it down and keep it safe 🔐
@@ -155,9 +87,7 @@ if not DISCORD_TOKEN:
     ## 5. Scroll down and enable 'Message Content Intent' under 'Priviledged Gateway Intents'
     
     ![](https://gradio-builds.s3.amazonaws.com/demo-files/discordbots/MessageContentIntent.png)
-
     ## 6. Save your changes!
-
     ## 7. The token from step 3 is the DISCORD_TOKEN. Rerun the deploy_discord command, e.g client.deploy_discord(discord_bot_token=DISCORD_TOKEN, ...), or add the token as a space secret manually.
 """
 else:
@@ -167,15 +97,12 @@ else:
     ## Add this bot to your server by clicking this link: 
     
     {url}
-
     ## How to use it?
-
     The bot can be triggered via `/chat` followed by your text prompt.
     
     This will create a thread with the bot's response to your text prompt.
     You can reply in the thread (without `/chat`) to continue the conversation.
     In the thread, the bot will only reply to the original author of the command.
-
     ⚠️ Note ⚠️: Please make sure this bot's command does have the same name as another command in your server.
     
     ⚠️ Note ⚠️: Bot commands do not work in DMs with the bot as of now.