async, try/excepts added

app.py

@@ -55,32 +55,37 @@ data = pd.read_csv("https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L
 
 @bot.event
 async def on_ready():
-    global global_df
-    print(f'Logged in as {bot.user.name}')
-    print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}")
-    # testing sheet -> read -> paste sheet
-
-    """import data from google sheets -> HF Space .csv (doesn't make API call this way, as it's read-only)""" 
-
-    data['discord_user_id'] = data['discord_user_id'].astype(str)
-    global_df = data
-    print(f"csv successfully retrieved: \n {global_df}")
-    #data.to_csv(csv_file, index=False)
-
+    try:
+        global global_df
+        print(f'Logged in as {bot.user.name}')
+        print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}")
+        # testing sheet -> read -> paste sheet
+    
+        """import data from google sheets -> HF Space .csv (doesn't make API call this way, as it's read-only)""" 
+    
+        data['discord_user_id'] = data['discord_user_id'].astype(str)
+        global_df = data
+        print(f"csv successfully retrieved: \n {global_df}")
+        #data.to_csv(csv_file, index=False)
+    except Exception as e:
+        print(f"on_message Error: {e}")
 
-def update_google_sheet():
+async def update_google_sheet():
     """save data from HF Space -> google sheets (makes 1 API call)"""
-    print("test")
-    name = "levelbot"
-    worksheet = gspread_bot.open(name).sheet1
-    global_df['discord_user_id'] = global_df['discord_user_id'].astype(str)
-    set_with_dataframe(worksheet, global_df)
-    worksheet.format(f"A1:A{len(global_df.index)}", {
-    	"numberFormat": {
-    		"type": 'TEXT',
-    	},
-    })
-    print({f"Google sheet {name} successfully updated! \n {global_df}"})
+    try:
+        print("test")
+        name = "levelbot"
+        worksheet = gspread_bot.open(name).sheet1
+        global_df['discord_user_id'] = global_df['discord_user_id'].astype(str)
+        set_with_dataframe(worksheet, global_df)
+        worksheet.format(f"A1:A{len(global_df.index)}", {
+        	"numberFormat": {
+        		"type": 'TEXT',
+        	},
+        })
+        print({f"Google sheet {name} successfully updated! \n {global_df}"})
+    except Exception as e:
+        print(f"on_message Error: {e}")
 
 """
 scheduler = BackgroundScheduler()
@@ -90,11 +95,11 @@ scheduler.start()
 """
 
 
-def calculate_level(xp):
+async def calculate_level(xp):
     return int(xp ** (1.0 / 3.0))
 
 
-def calculate_xp(level):
+async def calculate_xp(level):
     return (int(level ** 3))
 
 
@@ -198,7 +203,7 @@ async def add_exp(member_id):
 
           
                 # level up
-                current_level = calculate_level(new_xp)
+                current_level = await calculate_level(new_xp)
                 print(f"Current_level for {member}: {current_level}")
                 if current_level >= 2 and current_level <=30:
                     current_role = lvls[current_level]
@@ -219,7 +224,7 @@ async def add_exp(member_id):
             xp = 10 # define somewhere else?
             member_id = str(member_id)
             member_name = str(member.name)
-            current_level = calculate_level(xp)
+            current_level = await calculate_level(xp)
             row_data = [member_id, member_name, xp, current_level]
             global_df.loc[len(global_df.index)] = row_data 
             
@@ -235,7 +240,7 @@ async def add_exp(member_id):
             print(f"----------------------------------------------------")
 
         if member_id == 811235357663297546:
-            update_google_sheet()
+            await update_google_sheet()
     except Exception as e:
         print(f"add_exp Error: {e}")   
 
@@ -266,64 +271,68 @@ async def on_reaction_add(reaction, user):
 
 @bot.command(name='update_leaderboard')
 async def update_leaderboard(ctx, num_results: int = 10): # needs rewrite
-    if ctx.author.id == 811235357663297546:
-        await asyncio.sleep(1)
-        worksheet = gspread_bot.open("levelbot").sheet1
-    
-        names_list = worksheet.col_values(2)[1:]
-        levels_list = worksheet.col_values(4)[1:]
-        exp_list = worksheet.col_values(3)[1:]
-        channel = bot.get_channel(1197143964994773023)
-        message = await channel.fetch_message(1197148293164187678)
-    
-        # for 3 lists
-        combined_list = [list(sublist) for sublist in zip(names_list, levels_list, exp_list)]
-        combined_list = [[name, int(level), int(exp)] for name, level, exp in combined_list]
-        combined_list = sorted(combined_list, key=lambda x: x[1], reverse=True)
-    
-        print(combined_list)
-        top_results = combined_list[:num_results]
-        #print(top_results)
-    
-        """
-        # get position, then find that value in updated_names_list
-        levels_list = list(map(int, levels_list))
-        data_pairs = list(zip(names_list, levels_list))
-        sorted_data_pairs = sorted(data_pairs, key=lambda x: x[1], reverse=True)
-        top_data_pairs = sorted_data_pairs[:num_results]    
-        """
-    
-        # remove huggingfolks
-        guild = ctx.guild
-        role = discord.utils.get(guild.roles, id=897376942817419265)
-        if role is None:
-            await ctx.send("Role not found.")
-            return
-        members_with_role = [member.name for member in guild.members if role in member.roles]
-        top_results = [r for r in top_results if r[0] not in members_with_role]
+    try:
+        if ctx.author.id == 811235357663297546:
+            await asyncio.sleep(1)
+            worksheet = gspread_bot.open("levelbot").sheet1
         
-        for name, level, xp in top_results:
-            print(f"Name: {name}, Level: {level}, Exp: {xp}")    
-    
-    
-        def xp_required_to_next_level(current_level, current_xp):
-            level_floor_xp = calculate_xp(current_level)
-            level_ceiling_xp = calculate_xp(current_level+1)
-            xp_to_level_up = level_ceiling_xp - current_xp    
-            return (xp_to_level_up)
-    
-    
-        # put into message / leaderboard
-        new_leaderboard_data = [(name, level, str(xp_required_to_next_level(int(level), int(xp)))) for name, level, xp in top_results]  
-        new_table = tabulate(new_leaderboard_data, headers=["Name", "Level", "XP to level up"], tablefmt="plain")
-        await message.edit(content=f"Updated Leaderboard:\n```\n{new_table}\n```")
-
+            names_list = worksheet.col_values(2)[1:]
+            levels_list = worksheet.col_values(4)[1:]
+            exp_list = worksheet.col_values(3)[1:]
+            channel = bot.get_channel(1197143964994773023)
+            message = await channel.fetch_message(1197148293164187678)
+        
+            # for 3 lists
+            combined_list = [list(sublist) for sublist in zip(names_list, levels_list, exp_list)]
+            combined_list = [[name, int(level), int(exp)] for name, level, exp in combined_list]
+            combined_list = sorted(combined_list, key=lambda x: x[1], reverse=True)
+        
+            print(combined_list)
+            top_results = combined_list[:num_results]
+            #print(top_results)
+        
+            """
+            # get position, then find that value in updated_names_list
+            levels_list = list(map(int, levels_list))
+            data_pairs = list(zip(names_list, levels_list))
+            sorted_data_pairs = sorted(data_pairs, key=lambda x: x[1], reverse=True)
+            top_data_pairs = sorted_data_pairs[:num_results]    
+            """
+        
+            # remove huggingfolks
+            guild = ctx.guild
+            role = discord.utils.get(guild.roles, id=897376942817419265)
+            if role is None:
+                await ctx.send("Role not found.")
+                return
+            members_with_role = [member.name for member in guild.members if role in member.roles]
+            top_results = [r for r in top_results if r[0] not in members_with_role]
+            
+            for name, level, xp in top_results:
+                print(f"Name: {name}, Level: {level}, Exp: {xp}")    
+        
+        
+            def xp_required_to_next_level(current_level, current_xp):
+                level_floor_xp = await calculate_xp(current_level)
+                level_ceiling_xp = await calculate_xp(current_level+1)
+                xp_to_level_up = level_ceiling_xp - current_xp    
+                return (xp_to_level_up)
+        
+        
+            # put into message / leaderboard
+            new_leaderboard_data = [(name, level, str(xp_required_to_next_level(int(level), int(xp)))) for name, level, xp in top_results]  
+            new_table = tabulate(new_leaderboard_data, headers=["Name", "Level", "XP to level up"], tablefmt="plain")
+            await message.edit(content=f"Updated Leaderboard:\n```\n{new_table}\n```")
+    except Exception as e:
+        print(f"on_message Error: {e}")
 
 @bot.command(name='xp_help')
 async def xp_help(ctx):
-    help_message = "How to earn Discord / Hub exp: Post messages, react, Like, discuss, create repos and papers"
-    await ctx.author.send(help_message)
-
+    try:
+        help_message = "How to earn Discord / Hub exp: Post messages, react, Like, discuss, create repos and papers"
+        await ctx.author.send(help_message)
+    except Exception as e:
+        print(f"on_message Error: {e}")
 
 
 
@@ -360,10 +369,13 @@ csv_url = URL.replace('/edit#gid=', '/export?format=csv&gid=')
 
 
 def get_data():
-    data = pd.read_csv(csv_url)
-    first_3_columns = data.iloc[:, 1:4]
-    first_3_columns.to_csv('first_3_columns.csv', index=False)
-    return first_3_columns    
+    try:
+        data = pd.read_csv(csv_url)
+        first_3_columns = data.iloc[:, 1:4]
+        first_3_columns.to_csv('first_3_columns.csv', index=False)
+        return first_3_columns    
+    except Exception as e:
+        print(f"on_message Error: {e}")
 # csv
 # read into pandas dataframe1
 # read levels column and create pandas dataframe2 with first column containing levels from 2-max found in dataframe1
@@ -371,40 +383,42 @@ def get_data():
 
 demo = gr.Blocks()
 with demo:
-    dataframe1 = pd.read_csv(csv_url)
-    column_values_unique = sorted(dataframe1.iloc[:, 3].unique())
-    dataframe2 = pd.DataFrame({'Levels': column_values_unique})
-    counts = {}
-    for value in data.iloc[:, 3]:
-        counts[value] = counts.get(value, 0) + 1
-    dataframe2['Members'] = dataframe2['Levels'].map(counts)
-
-    print("\nDataframe 2:")
-    print(dataframe2)
-
-    TITLE = """<h1 align="center" id="space-title">🤗 Hugging Face Level Leaderboard</h1>"""
-    gr.HTML(TITLE)
-    with gr.Tabs(elem_classes="tab-buttons") as tabs:
-        with gr.TabItem("🏅 Level leaderboard", elem_id="level-table", id=0):
-            #gr.Markdown("# 📈 Experience Leaderboard")
-            with gr.Row():
-                with gr.Column():
-                    gr.DataFrame(get_data, every=5, height=500, interactive=False, col_count=(3, "fixed"), column_widths=["100px","100px","100px"])
-
-                with gr.Column():
-                    gr.BarPlot(
-                        value=dataframe2,
-                        x="Levels",
-                        y="Members",
-                        title="Level Distribution",
-                        height=450,
-                        width=450,
-                        interactive=False
-                    )                   
-        #with gr.TabItem("📈 Members of the Week", elem_id="week-table", id=1):
-     
-        #with gr.TabItem("📈 Hub-only leaderboard", elem_id="hub-table", id=2):
-
+    try:
+        dataframe1 = pd.read_csv(csv_url)
+        column_values_unique = sorted(dataframe1.iloc[:, 3].unique())
+        dataframe2 = pd.DataFrame({'Levels': column_values_unique})
+        counts = {}
+        for value in data.iloc[:, 3]:
+            counts[value] = counts.get(value, 0) + 1
+        dataframe2['Members'] = dataframe2['Levels'].map(counts)
+    
+        print("\nDataframe 2:")
+        print(dataframe2)
+    
+        TITLE = """<h1 align="center" id="space-title">🤗 Hugging Face Level Leaderboard</h1>"""
+        gr.HTML(TITLE)
+        with gr.Tabs(elem_classes="tab-buttons") as tabs:
+            with gr.TabItem("🏅 Level leaderboard", elem_id="level-table", id=0):
+                #gr.Markdown("# 📈 Experience Leaderboard")
+                with gr.Row():
+                    with gr.Column():
+                        gr.DataFrame(get_data, every=5, height=500, interactive=False, col_count=(3, "fixed"), column_widths=["100px","100px","100px"])
+    
+                    with gr.Column():
+                        gr.BarPlot(
+                            value=dataframe2,
+                            x="Levels",
+                            y="Members",
+                            title="Level Distribution",
+                            height=450,
+                            width=450,
+                            interactive=False
+                        )                   
+            #with gr.TabItem("📈 Members of the Week", elem_id="week-table", id=1):
+         
+            #with gr.TabItem("📈 Hub-only leaderboard", elem_id="hub-table", id=2):
+    except Exception as e:
+        print(f"on_message Error: {e}")
 demo.queue().launch()