reset every 20min

app.py

@@ -9,14 +9,12 @@ import random
 import asyncio
 import discord
 import logging
-import os.path
-import secrets
 import aiohttp
 import gspread
 import datetime
-import schedule
 import requests
 import threading
+import schedule
 import gradio_client
 
 import numpy as np
@@ -32,33 +30,34 @@ from discord.ui import Button, View
 from discord.ext import commands, tasks
 from datetime import datetime, timedelta
 from urllib.parse import urlparse, parse_qs
-from apscheduler.executors.pool import ThreadPoolExecutor
-from apscheduler.executors.asyncio import AsyncIOExecutor
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from gspread_formatting.dataframe import format_with_dataframe
-from apscheduler.schedulers.background import BackgroundScheduler
 from gspread_dataframe import get_as_dataframe, set_with_dataframe
 from huggingface_hub import HfApi, list_liked_repos, list_models
 
-
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
+
 intents = discord.Intents.all() 
 bot = commands.Bot(command_prefix='!', intents=intents)
 
+scheduler = AsyncIOScheduler()
 
-## testing restart
 def restart_bot():
     print("Restarting bot...")
     os.execv(sys.executable, ['python'] + sys.argv)
 
+@scheduler.scheduled_job('interval', minutes=20)
+def periodic_restart():
+    print("Scheduled restart triggered...")
+    restart_bot()
 
 @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)""" 
+    """Import data from Google Sheets -> HF Space df"""
     print(f"We have logged in as {bot.user}")
+    scheduler.start()
     await give_verified_roles()
 
-
 async def give_verified_roles():
     while True:
         try:
@@ -84,65 +83,9 @@ async def give_verified_roles():
                     print(f"Error fetching CSV: {e}")
                     await asyncio.sleep(60)
                     continue
- 
+
             guild = bot.get_guild(879548962464493619)
             role = guild.get_role(900063512829755413)
-    
-            # Define the invite message
-            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()
-    
-            # Iterate over the dataframe rows
-            for index, row in global_df.iterrows():
-                # Extract Hugging Face username
-                hf_user_name = row['hf_user_name']
-                if pd.notna(hf_user_name) and hf_user_name.lower() != 'n/a':
-                    discord_id = row['discord_user_id'].strip('L')
-                    member = guild.get_member(int(discord_id))
-                    if not member:
-                        continue
-                    if role not in member.roles:
-                        await member.add_roles(role)
-                        await asyncio.sleep(5)
-                        print(f"Role added to member: {member}")
-                        #await asyncio.sleep(10)
-                        lunar = bot.get_user(811235357663297546)
-                        if lunar:
-                            await lunar.send(f"Verified role given to {member}!") 
-                        await member.send(
-                            f"Verification successful! [{member} <---> {row['discord_user_name']}] \n🤗 {org_link} {invite_message}"
-                        )
-                        await asyncio.sleep(5)
-    
-        except Exception as e:
-            print(f"Error encountered: {e}")
-        await asyncio.sleep(30) # was 60
-
-        
-# 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("00:00").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()
+            # Define the invite message
+            org_link = "https://huggingface.co/organizations/discord-community/share/wPKRAHYbAlaEaCxUxcq