Update app.py

app.py

@@ -3,7 +3,6 @@ import uvloop
 import os
 import logging
 import threading
-from threading import Event
 from typing import Optional
 import discord
 import gradio as gr
@@ -14,6 +13,7 @@ import gradio_client as grc
 from gradio_client.utils import QueueError
 import aiohttp
 from aiohttp.resolver import AsyncResolver
+import utils
 
 # Setup Discord bot 
 logging.basicConfig(level=logging.INFO)
@@ -21,8 +21,6 @@ intents = discord.Intents.default()
 intents.message_content = True
 intents.members = True
 
-event = Event()
-
 DISCORD_TOKEN = os.getenv("DISCORD_TOKEN")
 
 async def wait(job):
@@ -35,13 +33,6 @@ def get_client(session: Optional[str] = None) -> grc.Client:
         client.session_hash = session
     return client
 
-def truncate_response(response: str) -> str:
-    ending = "...\nTruncating response to 2000 characters due to discord api limits."
-    if len(response) > 2000:
-        return response[: 2000 - len(ending)] + ending
-    else:
-        return response
-
 async def main():
     # Setup DNS resolver to avoid potential issues
     resolver = AsyncResolver(nameservers=["1.1.1.1", "8.8.8.8"])
@@ -85,7 +76,7 @@ async def main():
             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:
@@ -105,7 +96,7 @@ async def main():
             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!"
@@ -128,17 +119,9 @@ async def main():
         except Exception as e:
             print(f"Error: {e}")
     
-    # 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()
+    # Start keepalive server and bot
+    utils.keepalive_run()
+    await bot.start(os.getenv('DISCORD_TOKEN'))
 
 if __name__ == "__main__":
     asyncio.run(main())