import discord import os import threading from discord.ext import commands import json import datetime import requests import os.path import random import gspread import re import gradio_client import gradio as gr from gradio_client import Client DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None) intents = discord.Intents.all() bot = commands.Bot(command_prefix='!', intents=intents) """""" XP_PER_MESSAGE = 10 # 100k messages = 1M exp = lvl 100 """""" service_account = json.loads(os.environ.get('KEY')) file_path = 'service_account.json' with open(file_path, 'w') as json_file: json.dump(service_account, json_file) gspread_bot = gspread.service_account(filename='service_account.json') worksheet = gspread_bot.open("levelbot").sheet1 """""" bot_ids = [1136614989411655780, 1166392942387265536, 1158038249835610123, 1130774761031610388, 1155489509518098565, 1155169841276260546, 1152238037355474964, 1154395078735953930] """""" @bot.event async def on_ready(): print(f'Logged in as {bot.user.name}') print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}") def calculate_level(xp): return int(xp ** (1.0 / 3.0)) def calculate_xp(level): return (int(level ** 3)) async def add_exp(member): try: guild = bot.get_guild(879548962464493619) lvl1 = guild.get_role(1171861537699397733) lvl2 = guild.get_role(1171861595115245699) lvl3 = guild.get_role(1171861626715115591) lvl4 = guild.get_role(1171861657975259206) lvl5 = guild.get_role(1171861686580412497) lvl6 = guild.get_role(1171861900301172736) lvl7 = guild.get_role(1171861936258941018) lvl8 = guild.get_role(1171861968597024868) lvl9 = guild.get_role(1171862009982242836) lvl10 = guild.get_role(1164188093713223721) lvl11 = guild.get_role(1171524944354607104) lvl12 = guild.get_role(1171524990257082458) lvl13 = guild.get_role(1171525021928263791) lvl14 = guild.get_role(1171525062201966724) lvl15 = guild.get_role(1171525098465918996) lvls = { 1: lvl1, 2: lvl2, 3: lvl3, 4: lvl4, 5: lvl5, 6: lvl6, 7: lvl7, 8: lvl8, 9: lvl9, 10: lvl10, 11: lvl11, 12: lvl12, 13: lvl13, 14: lvl14, 15: lvl15, } #if member.id == 811235357663297546: # does a record already exist? cell = worksheet.find(str(member.id)) length = len(worksheet.col_values(1)) if cell is None: print(f"creating new record for {member}") # if not, create new record string_member_id = str(member.id) xp = 10 current_level = calculate_level(xp) member_name = member.name worksheet.update(values=[[string_member_id, member_name, xp, current_level]], range_name=f'A{length+1}:D{length+1}') # initial role assignment if current_level == 1: if lvl1 not in member.roles: await member.add_roles(lvl1) print(f"Gave {member} {lvl1}") else: if cell: # if so, update that row... xp = worksheet.cell(cell.row, cell.col+2).value xp = int(xp) + XP_PER_MESSAGE current_level = calculate_level(xp) print(f"updating record for {member}: {xp} xp") # write with added xp worksheet.update(values=[[xp, current_level]], range_name=f'C{cell.row}:D{cell.row}') # level up if current_level >= 2 and current_level <=15: current_role = lvls[current_level] if current_role not in member.roles: await member.add_roles(current_role) print(f"Gave {member} {current_role}") await member.remove_roles(lvls[current_level-1]) print(f"Removed {lvls[current_level-1]} from {member}") except Exception as e: print(f"add_exp Error: {e}") memberlist = [880517673665069056, 1107354874410193058, 904483056096772157, 256178664545320985, 533431969489485824, 666044046510391296, 664692763660648500, 556359685306056721, 761341255421788160, 702313896270102588, 779316628382285875, 612687217986502662, 886717151958949909, 915862238215168080, 296112700034121729, 702564345837781107, 365891577010192387, 114878879486836742, 859286466420408360, 312281922502131716, 691380597620473917, 624407349867708446, 432172960158056458, 408799651563700225, 440003727726739456, 739856785309171823, 782548161490321430, 1001717789247164417, 771840318999756840, 741829742667497522, 531839001167134730, 271654301942808576, 183685370003587073, 794928855487610900, 923139529685233665, 790322701260292137, 258494749453844481, 138110479380905984, 700741522404016178, 115707698334400515, 1036824500119031838, 705516123877146717, 806931075607035954, 119238648259543044, 836953463941496838, 284316797631725568, 994516380881780817, 375935758755823617, 885622316170612776, 378159748052287488, 790374047962890292, 442025360763387904, 724733717712470038, 864447241624354856, 900320502789324800, 908839508085182474, 965979226161414174, 969180367175155722, 198924160938934273, 652261149068492827, 936866294031073291, 530814506427744266, 486576362417815562, 1123577306519187507, 827581464841289728, 752885249184366663, 248246366529126401, 545939238621806612, 692991480591024170, 820598424814223371, 583190845491249162, 176340260211392512, 466612127910854677, 340469567421808640, 742289903362113647, 764008173970653185, 743806647431856168, 1090923181910532167, 564107873496596502, 713799720333738044, 997168115802722304, 813008300702171136, 927756109182296095, 994721274871959612, 912392990264295484, 178431838962974720, 902960815328600074, 529256775330037770, 741460384682082355, 481724798443978752, 917749478310699038, 242569325087555584, 692880955400257616, 964060740350791691, 1016263191547883530, 707365701010194592, 426188263317962752, 1064747458120401046, 967817056093343834, 960243025131487402, 479736593674928128, 731149313270612030, 812973418068705282, 493065034709205005, 982346160385712169, 741279012055154709, 216279898128384000, 294949693576839168, 499331168928399361, 551805099782045716, 961903089475018822, 853014693115068426, 699051313685725185, 442193894642417675, 833840009097248851, 423164583310786560, 908146032175620096, 590328553481175080, 748444365713571891, 801067739895824405, 594064588320800788, 738323260143435817, 545210284692733969, 531900905407381540, 685367983564128281, 480439347527680000, 977550251307204628, 595852442760118294, 362710695197933572, 213594857074196480, 652692443099299841, 882082042798538822, 860589887966806037, 260483618143469569, 933071618153013269, 460689816791678976, 226107942498205696, 781095904743784478, 907691778000449646, 616912704732528640, 667760218104397854, 866200438680059907, 232671334306545664, 883618102678536223, 364516913428168714, 860460488944254997, 234763281749770243, 702750627276521504, 533427213257670671, 348869248963182592, 1013391359211741336, 351743892309278721, 174580891245477888, 751093888885719132, 326082865035083778, 243246388224196610, 701473194854449212, 1087249987911561326, 643355951600107520, 754669984625459210, 366282094311112714, 336293882964869131, 768865999050637312, 532169631801147393, 183630669220544513, 874053755006513202, 417428981705605121, 375786646664380416, 225221892045275137, 456523620680728596, 199899258172866560, 966874758807441448, 469160618151378944, 285136467112493061, 288704791641718795, 882266585421713408, 391246796245958656, 712891236813373461, 168869159483080705, 370497630188601344, 449004480826966047, 325447771790508033, 869262942632284230, 622759325923803166, 966878995633287208, 247051633148690433, 889959421797601302, 519222880840450059, 768377731930783766, 800315102271242251, 691351867564752898, 632849691712946186, 547714551802494986, 372660448656621569, 502035836418916359, 402221967744303109, 768123325779935254, 682138038868246548, 232145650085658625, 1063495824711098418, 885786246952124457, 811235357663297546] @bot.command(name='update_sheet') async def update_sheet(ctx): if ctx.author.id == 811235357663297546: guild = bot.get_guild(879548962464493619) #hf_discord_verified_users_test = gspread_bot.open("hf_discord_verified_users_test").sheet1 spreadsheet = gspread_bot.open("levelbot") worksheet2 = spreadsheet.get_worksheet(1) verified_role = guild.get_role(900063512829755413) members = verified_role.members random.shuffle(members) for member in members: if member.id not in memberlist: print(f"Processing member: {member}") string_user_id = str(member.id) string_user_id_2 = string_user_id[:-5] pattern = re.compile(re.escape(string_user_id_2)) cell_list = worksheet2.findall(pattern) for cell in cell_list: row_number = cell.row column_number_to_update = 5 worksheet2.update_cell(row_number, column_number_to_update, string_user_id) memberlist.append(member.id) print(memberlist) @bot.event async def on_message(message): try: if message.author.id not in bot_ids: await add_exp(message.author) await bot.process_commands(message) except Exception as e: print(f"on_message Error: {e}") @bot.event async def on_reaction_add(reaction, user): try: if user.id not in bot_ids: await add_exp(user) except Exception as e: print(f"on_reaction_add Error: {e}") """""" DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None) def run_bot(): bot.run(DISCORD_TOKEN) threading.Thread(target=run_bot).start() def greet(name): return "Hello " + name + "!" demo = gr.Interface(fn=greet, inputs="text", outputs="text") demo.launch()