Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
| 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] | |
| """""" | |
| 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] | |
| 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) | |
| 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}") | |
| 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() | |