test auto restart (20:05)

app.py

@@ -2,6 +2,7 @@ import os
 import io
 import re
 import csv
+import sys
 import json
 import time
 import random
@@ -13,6 +14,7 @@ import secrets
 import aiohttp
 import gspread
 import datetime
+import schedule
 import requests
 import threading
 import gradio_client
@@ -44,14 +46,12 @@ intents = discord.Intents.all()
 bot = commands.Bot(command_prefix='!', intents=intents)
 
 
-
-## testing this, restart bot on errors 
+## testing restart
 def restart_bot():
     print("Restarting bot...")
     os.execv(sys.executable, ['python'] + sys.argv)
 
 
-
 @bot.event
 async def on_ready():
     """import data from google sheets -> HF Space df (doesn't make API call this way, as it's read-only)""" 
@@ -92,7 +92,6 @@ async def give_verified_roles():
             org_link = "https://huggingface.co/organizations/discord-community/share/wPKRAHYbAlaEaCxUxcqVyaaaeZcYagDvqc"
             invite_message = "Click to join our community org on the HF Hub!"
 
-    
             # Cache members in the guild
             await guild.chunk()
     
@@ -120,18 +119,31 @@ async def give_verified_roles():
                         await asyncio.sleep(10)
     
         except Exception as e:
-            print(f"Error encountered, restarting bot: {e}")
-            asyncio.sleep(5)
-            restart_bot()
-
+            print(f"Error encountered: {e}")
         await asyncio.sleep(60)
 
         
-# runs discord bot in thread = helps avoid blocking calls
+# Function to run the bot in a thread
 def run_bot():
     bot.run(DISCORD_TOKEN)
+
+
+# Function to schedule the bot restart at midnight
+def schedule_restart():
+    schedule.every().day.at("20:05").do(restart_bot)
+
+    while True:
+        schedule.run_pending()
+        time.sleep(1)
+
+
+# Launch the bot and restart scheduler in separate threads
 threading.Thread(target=run_bot).start()
+threading.Thread(target=schedule_restart).start()
+
+
+# Example Gradio interface
 def greet(name):
     return "Hello " + name + "!"
 demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+demo.launch()