Update app.py

app.py

@@ -19,6 +19,9 @@ import utils
 
 event = Event()
 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")
 
@@ -28,50 +31,40 @@ def get_client(session: Optional[str] = None) -> Client:
         client.session_hash = session
     return client
 
-# Setup Discord bot 
- # Setup DNS resolver to avoid potential issues
-resolver = AsyncResolver(nameservers=["1.1.1.1", "8.8.8.8"])
-connector = aiohttp.TCPConnector(resolver=resolver)
-    
-intents = discord.Intents.default()
-intents.message_content = True
-intents.members = True
-bot = commands.Bot(intents=intents, connector=connector, command_prefix="!")
+async def main():
+    # Setup DNS resolver to avoid potential issues
+    resolver = AsyncResolver(nameservers=["1.1.1.1", "8.8.8.8"])
+    connector = aiohttp.TCPConnector(resolver=resolver)
+    # Setup Discord 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("------")    
-
-@bot.command()
-async def test(ctx, *, arg):
-    thread = await ctx.create_thread(name=prompt, auto_archive_duration=60)
-    loop = asyncio.get_running_loop()
-    client = await loop.run_in_executor(None, get_client, None)
-    job = client.submit(prompt, api_name="/slow_echo")
-    wait([job])
-    await ctx.send(arg)
-    
-# running in thread
-def run_bot():
-    if not DISCORD_TOKEN:
-        print("DISCORD_TOKEN NOT SET")
+    @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()
-    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
-    """
-    )
+        print("------")    
+    
+    @bot.command()
+    async def test(ctx, *, arg):
+        thread = await ctx.create_thread(name=prompt, auto_archive_duration=60)
+        loop = asyncio.get_running_loop()
+        client = await loop.run_in_executor(None, get_client, None)
+        job = client.submit(prompt, api_name="/slow_echo")
+        wait([job])
+        await ctx.send(arg)
+        
+    # running in thread
+    def run_bot():
+        if not DISCORD_TOKEN:
+            print("DISCORD_TOKEN NOT SET")
+            event.set()
+        else:
+            bot.run(DISCORD_TOKEN)
+    
+    utils.keepalive_run()
+    await bot.start(os.getenv('DISCORD_TOKEN'))
 
-demo.launch()
+if __name__ == "__main__":
+    asyncio.run(main())