Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
small revert
Browse files
app.py
CHANGED
@@ -53,40 +53,37 @@ api = HfApi()
|
|
53 |
""""""
|
54 |
#csv_file = 'data.csv'
|
55 |
global_df = pd.DataFrame()
|
56 |
-
worksheet2_df = pd.DataFrame()
|
57 |
print(type(global_df))
|
58 |
community_global_df = pd.DataFrame()
|
59 |
data = pd.read_csv("https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w/export?format=csv&gid=0")
|
60 |
-
data2 = pd.read_csv("https://docs.google.com/spreadsheets/d/1T_twNONDSZ8zw7aDcYhw92DlawqPktti1xFw9ssJ13Y/edit#gid=0")
|
61 |
-
|
62 |
|
63 |
@bot.event
|
64 |
async def on_ready():
|
65 |
try:
|
66 |
global global_df
|
67 |
-
global worksheet2_df
|
68 |
await asyncio.sleep(1.1)
|
69 |
print(f'Logged in as {bot.user.name}')
|
70 |
print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}")
|
|
|
|
|
|
|
71 |
|
72 |
-
"""import data from google sheets -> HF Space df (doesn't make API call this way, as it's read-only)"""
|
73 |
data['discord_user_id'] = data['discord_user_id'].astype(str)
|
74 |
global_df = data
|
75 |
print(f"csv successfully retrieved: \n {global_df}")
|
76 |
-
|
77 |
|
78 |
-
# updates both leaderboards
|
79 |
remove_huggingfolks.start()
|
80 |
-
await(periodic_api_test)
|
81 |
print(f"------------------------------------------------------------------------")
|
82 |
except Exception as e:
|
83 |
print(f"on_message Error: {e}")
|
84 |
|
85 |
|
86 |
def update_google_sheet():
|
87 |
-
"""save data from HF Space -> google sheets (makes
|
|
|
88 |
try:
|
89 |
-
print("Updating
|
90 |
name = "levelbot"
|
91 |
#worksheet = gspread_bot.open(name).sheet1
|
92 |
global_df['discord_user_id'] = global_df['discord_user_id'].astype(str)
|
@@ -107,8 +104,14 @@ def update_google_sheet():
|
|
107 |
|
108 |
executor = ThreadPoolExecutor(max_workers=1)
|
109 |
scheduler = BackgroundScheduler(executors={'default': executor})
|
110 |
-
scheduler.add_job(update_google_sheet, trigger='interval', minutes=
|
111 |
scheduler.start()
|
|
|
|
|
|
|
|
|
|
|
|
|
112 |
|
113 |
|
114 |
def calculate_level(xp):
|
@@ -119,7 +122,41 @@ def calculate_xp(level):
|
|
119 |
return (int(level ** 3))
|
120 |
|
121 |
|
122 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
123 |
|
124 |
|
125 |
async def add_exp(member_id):
|
@@ -231,7 +268,7 @@ async def add_exp(member_id):
|
|
231 |
|
232 |
#lunar = bot.get_user(811235357663297546)
|
233 |
await member.send(embed=embed)
|
234 |
-
|
235 |
#You can verify your account to earn 100 points! To verify, do A.
|
236 |
|
237 |
|
@@ -274,7 +311,7 @@ async def on_message(message):
|
|
274 |
async def on_reaction_add(reaction, user):
|
275 |
try:
|
276 |
if user.id not in bot_ids:
|
277 |
-
print(f"adding exp from react {user}")
|
278 |
await asyncio.sleep(0.1)
|
279 |
await add_exp(user.id)
|
280 |
await asyncio.sleep(0.1)
|
@@ -298,25 +335,25 @@ async def remove_huggingfolks():
|
|
298 |
for member_id in members_with_role:
|
299 |
community_global_df = community_global_df[community_global_df.iloc[:, 0] != str(member_id)]
|
300 |
|
301 |
-
|
302 |
|
303 |
community_global_df.drop(community_global_df.columns[0], axis=1, inplace=True)
|
304 |
community_global_df['discord_exp'] = pd.to_numeric(community_global_df['discord_exp'])
|
305 |
community_global_df = community_global_df.nlargest(len(community_global_df), 'discord_exp')
|
306 |
|
307 |
top_30_exp = community_global_df.nlargest(30, 'discord_exp')
|
|
|
308 |
|
309 |
top_30_exp['D'] = ['🥇','🥈','🥉','','','','','','','','','','','','','','','','','','','','','','','','','','','']
|
310 |
top_30_rows = top_30_exp.values.tolist()
|
311 |
|
312 |
-
|
313 |
channel = bot.get_channel(1197143964994773023)
|
314 |
message = await channel.fetch_message(1197148293164187678)
|
315 |
|
316 |
# put into message / leaderboard
|
317 |
new_table = tabulate(top_30_rows, headers=["Name", "Experience", "Level", "Rank"], tablefmt="plain")
|
318 |
await message.edit(content=f"Updated Leaderboard:\n```\n{new_table}\n```")
|
319 |
-
print(f"Updated discord leaderboard!")
|
320 |
|
321 |
except Exception as e:
|
322 |
print(f"remove_huggingfolks Error: {e}")
|
@@ -342,13 +379,24 @@ async def count_users_with_role(ctx, role_id):
|
|
342 |
# embeds with user pfps?
|
343 |
# name, pfp, time in server....
|
344 |
|
|
|
|
|
|
|
|
|
345 |
# weekly do different count
|
346 |
# count number of messages per user for every channel (total messages)
|
347 |
# fix sheet if necessary
|
|
|
|
|
348 |
# add emojis for some color
|
349 |
# check if members are still in the server
|
350 |
|
351 |
|
|
|
|
|
|
|
|
|
|
|
352 |
""""""
|
353 |
DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
|
354 |
def run_bot():
|
@@ -365,7 +413,10 @@ def get_data():
|
|
365 |
return community_global_df
|
366 |
except Exception as e:
|
367 |
print(f"on_message Error: {e}")
|
368 |
-
|
|
|
|
|
|
|
369 |
|
370 |
demo = gr.Blocks()
|
371 |
with demo:
|
@@ -407,4 +458,3 @@ with demo:
|
|
407 |
print(f"on_message Error: {e}")
|
408 |
demo.queue().launch()
|
409 |
|
410 |
-
|
|
|
53 |
""""""
|
54 |
#csv_file = 'data.csv'
|
55 |
global_df = pd.DataFrame()
|
|
|
56 |
print(type(global_df))
|
57 |
community_global_df = pd.DataFrame()
|
58 |
data = pd.read_csv("https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w/export?format=csv&gid=0")
|
|
|
|
|
59 |
|
60 |
@bot.event
|
61 |
async def on_ready():
|
62 |
try:
|
63 |
global global_df
|
|
|
64 |
await asyncio.sleep(1.1)
|
65 |
print(f'Logged in as {bot.user.name}')
|
66 |
print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}")
|
67 |
+
# testing sheet -> read -> paste sheet
|
68 |
+
|
69 |
+
"""import data from google sheets -> HF Space .csv (doesn't make API call this way, as it's read-only)"""
|
70 |
|
|
|
71 |
data['discord_user_id'] = data['discord_user_id'].astype(str)
|
72 |
global_df = data
|
73 |
print(f"csv successfully retrieved: \n {global_df}")
|
74 |
+
#data.to_csv(csv_file, index=False)
|
75 |
|
|
|
76 |
remove_huggingfolks.start()
|
|
|
77 |
print(f"------------------------------------------------------------------------")
|
78 |
except Exception as e:
|
79 |
print(f"on_message Error: {e}")
|
80 |
|
81 |
|
82 |
def update_google_sheet():
|
83 |
+
"""save data from HF Space -> google sheets (makes 1 API call)"""
|
84 |
+
# run this in separate thread?
|
85 |
try:
|
86 |
+
print("Updating...")
|
87 |
name = "levelbot"
|
88 |
#worksheet = gspread_bot.open(name).sheet1
|
89 |
global_df['discord_user_id'] = global_df['discord_user_id'].astype(str)
|
|
|
104 |
|
105 |
executor = ThreadPoolExecutor(max_workers=1)
|
106 |
scheduler = BackgroundScheduler(executors={'default': executor})
|
107 |
+
scheduler.add_job(update_google_sheet, trigger='interval', minutes=2, max_instances=1)
|
108 |
scheduler.start()
|
109 |
+
"""
|
110 |
+
scheduler = BackgroundScheduler()
|
111 |
+
scheduler.add_job(update_google_sheet, "interval", seconds=60)
|
112 |
+
scheduler.start()
|
113 |
+
|
114 |
+
"""
|
115 |
|
116 |
|
117 |
def calculate_level(xp):
|
|
|
122 |
return (int(level ** 3))
|
123 |
|
124 |
|
125 |
+
processed_users = set()
|
126 |
+
|
127 |
+
async def periodic_api_test(): # needs rewrite, can do same thing and interact with csv -> push to google sheets
|
128 |
+
try:
|
129 |
+
await asyncio.sleep(0.1)
|
130 |
+
column_values_3 = worksheet2.col_values(3)
|
131 |
+
column_values_8 = worksheet2.col_values(8)
|
132 |
+
|
133 |
+
for i, user in enumerate(column_values_3):
|
134 |
+
if user in processed_users:
|
135 |
+
continue
|
136 |
+
|
137 |
+
url = f"https://huggingface.co/api/users/{user}/overview"
|
138 |
+
response = requests.get(url)
|
139 |
+
|
140 |
+
if response.status_code == 200:
|
141 |
+
data = response.json()
|
142 |
+
|
143 |
+
likes = data["numLikes"]
|
144 |
+
models = data["numModels"]
|
145 |
+
datasets = data["numDatasets"]
|
146 |
+
spaces = data["numSpaces"]
|
147 |
+
discussions = data["numDiscussions"]
|
148 |
+
papers = data["numPapers"]
|
149 |
+
upvotes = data["numUpvotes"]
|
150 |
+
|
151 |
+
worksheet2.update(values=[[likes, models, datasets, spaces, discussions, papers, upvotes]],
|
152 |
+
range_name=f'G{i+1}:M{i+1}')
|
153 |
+
|
154 |
+
processed_users.add(user)
|
155 |
+
else:
|
156 |
+
print(f"Failed to retrieve data for user {user}. Status code: {response.status_code}")
|
157 |
+
|
158 |
+
except Exception as e:
|
159 |
+
print(f"periodic_api_test Error: {e}")
|
160 |
|
161 |
|
162 |
async def add_exp(member_id):
|
|
|
268 |
|
269 |
#lunar = bot.get_user(811235357663297546)
|
270 |
await member.send(embed=embed)
|
271 |
+
|
272 |
#You can verify your account to earn 100 points! To verify, do A.
|
273 |
|
274 |
|
|
|
311 |
async def on_reaction_add(reaction, user):
|
312 |
try:
|
313 |
if user.id not in bot_ids:
|
314 |
+
print(f"adding exp from react {user.id}")
|
315 |
await asyncio.sleep(0.1)
|
316 |
await add_exp(user.id)
|
317 |
await asyncio.sleep(0.1)
|
|
|
335 |
for member_id in members_with_role:
|
336 |
community_global_df = community_global_df[community_global_df.iloc[:, 0] != str(member_id)]
|
337 |
|
338 |
+
print(community_global_df)
|
339 |
|
340 |
community_global_df.drop(community_global_df.columns[0], axis=1, inplace=True)
|
341 |
community_global_df['discord_exp'] = pd.to_numeric(community_global_df['discord_exp'])
|
342 |
community_global_df = community_global_df.nlargest(len(community_global_df), 'discord_exp')
|
343 |
|
344 |
top_30_exp = community_global_df.nlargest(30, 'discord_exp')
|
345 |
+
#community_global_df = top_30_exp.copy() # for gradio
|
346 |
|
347 |
top_30_exp['D'] = ['🥇','🥈','🥉','','','','','','','','','','','','','','','','','','','','','','','','','','','']
|
348 |
top_30_rows = top_30_exp.values.tolist()
|
349 |
|
350 |
+
print(top_30_rows)
|
351 |
channel = bot.get_channel(1197143964994773023)
|
352 |
message = await channel.fetch_message(1197148293164187678)
|
353 |
|
354 |
# put into message / leaderboard
|
355 |
new_table = tabulate(top_30_rows, headers=["Name", "Experience", "Level", "Rank"], tablefmt="plain")
|
356 |
await message.edit(content=f"Updated Leaderboard:\n```\n{new_table}\n```")
|
|
|
357 |
|
358 |
except Exception as e:
|
359 |
print(f"remove_huggingfolks Error: {e}")
|
|
|
379 |
# embeds with user pfps?
|
380 |
# name, pfp, time in server....
|
381 |
|
382 |
+
# discord_level column
|
383 |
+
# pick 10 highest
|
384 |
+
# update
|
385 |
+
|
386 |
# weekly do different count
|
387 |
# count number of messages per user for every channel (total messages)
|
388 |
# fix sheet if necessary
|
389 |
+
|
390 |
+
# might need cell location data to pull both level and username at same time
|
391 |
# add emojis for some color
|
392 |
# check if members are still in the server
|
393 |
|
394 |
|
395 |
+
|
396 |
+
|
397 |
+
|
398 |
+
|
399 |
+
|
400 |
""""""
|
401 |
DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN", None)
|
402 |
def run_bot():
|
|
|
413 |
return community_global_df
|
414 |
except Exception as e:
|
415 |
print(f"on_message Error: {e}")
|
416 |
+
# csv
|
417 |
+
# read into pandas dataframe1
|
418 |
+
# read levels column and create pandas dataframe2 with first column containing levels from 2-max found in dataframe1
|
419 |
+
# create second column in dataframe2 for number of each level found in dataframe1 levels column
|
420 |
|
421 |
demo = gr.Blocks()
|
422 |
with demo:
|
|
|
458 |
print(f"on_message Error: {e}")
|
459 |
demo.queue().launch()
|
460 |
|
|