Update app.py

app.py

@@ -18,16 +18,24 @@ from aiohttp.resolver import AsyncResolver
 import utils
 
 event = Event()
+logging.basicConfig(level=logging.INFO)
+DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
+HF_TOKEN=os.getenv("HF_TOKEN")
+
+def get_client(session: Optional[str] = None) -> grc.Client:
+    # Setup DNS resolver to avoid potential issues
+    resolver = AsyncResolver(nameservers=["1.1.1.1", "8.8.8.8"])
+    connector = aiohttp.TCPConnector(resolver=resolver)
+    client = Client("https://standarovskid-bot.hf.space/", token=HF_TOKEN)
+    if session:
+        client.session_hash = session
+    return client
 
 # Setup Discord bot 
-logging.basicConfig(level=logging.INFO)
 intents = discord.Intents.default()
 intents.message_content = True
 intents.members = True
-
-DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
-HF_TOKEN=os.getenv("HF_TOKEN")
-
+bot = commands.Bot(intents=intents, connector=connector, command_prefix="!")
 
 async def main():
     # Setup DNS resolver to avoid potential issues
@@ -35,55 +43,37 @@ async def main():
     connector = aiohttp.TCPConnector(resolver=resolver)
     client = Client("https://standarovskid-bot.hf.space/", token=HF_TOKEN)
     
-    # Instantiate bot
-    bot = commands.Bot(intents=intents, connector=connector, command_prefix="!")
     
-    @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("------")
-
-    async def response(input_text):
-        return "fine"
     
-    @bot.hybrid_command(
-        name="echo",
-        description="Enter some text to chat with the bot! Like this: /echo Hello, how are you?",
-    )    
-    async def echo(ctx,*,input_text):
-        try:
-            await ctx.send("toto1")
-            await ctx.channel.send("toto2")
-            # 调用 Gradio 应用 API
-            result = client.predict(
-        		message=input_text,
-        		api_name="/slow_echo"
-            )
-            await ctx.channel.send(f"yeah right lmfao 🥀🥀")
-      
-            return result
-        except Exception as e:
-            await ctx.channel.send("issueo")
-            return f"Erreur：{e}"   
+@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("------")    
 
-    #@bot.event
-    #async def on_message(message):
-        # 忽略自己的消息
-        #if message.author == bot.user:
-            #return
+@bot.command()
+async def test(ctx, *, arg):
+    await ctx.send(arg)
     
-        #if message.content.startswith("/echo"):
-            #user_input = message.content[len("/echo "):]
-      #  result = call_huggingface_space(user_input)
-        #await message.channel.send(f"yeah right lmfao 🥀🥀")
-        #await message.channel.send(result)
-     
-    # Start keepalive server and bot
-    utils.keepalive_run()
-    await bot.start(os.getenv('DISCORD_TOKEN'))
+# running in thread
+def run_bot():
+    if not DISCORD_TOKEN:
+        print("DISCORD_TOKEN NOT SET")
+        event.set()
+    else:
+        bot.run(DISCORD_TOKEN)
+
+threading.Thread(target=run_bot).start()
+
+event.wait()
+
+with gr.Blocks() as demo:
+    gr.Markdown(
+        f"""
+    # Discord bot of https://standarovskid-bot.hf.space
+    """
+    )
 
-if __name__ == "__main__":
-    asyncio.run(main())
+demo.launch()