Update app.py

app.py

@@ -14,6 +14,7 @@ import csv
 from tabulate import tabulate
 import logging
 import time
+import pandas as pd
 
 import gradio_client
 import gradio as gr
@@ -51,7 +52,6 @@ async def on_ready():
     print(f"XP_PER_MESSAGE: {XP_PER_MESSAGE}")
 
 
-    
 def calculate_level(xp):
     return int(xp ** (1.0 / 3.0))
 
@@ -60,7 +60,6 @@ def calculate_xp(level):
     return (int(level ** 3))
 
 
-
 processed_users = set()
 
 async def periodic_api_test():
@@ -98,44 +97,6 @@ async def periodic_api_test():
         print(f"periodic_api_test Error: {e}")   
 
 
-
-
-"""
-async def search_csv(file_path, column_index, search_value):
-    with open(file_path, 'r', newline='') as csvfile:
-        reader = csv.reader(csvfile)
-        first_row = next(reader)
-        print(first_row)        
-        for row in reader:
-            if row and len(row) > column_index and row[column_index] == search_value:
-                print(row)
-
-
-
-def is_csv_empty(file_path):
-    if not os.path.isfile(file_path):
-        print(f"The file '{file_path}' does not exist.")
-        return False
-
-    with open(file_path, 'r', newline='') as csvfile:
-        reader = csv.reader(csvfile)
-        try:
-            first_row = next(reader)
-            print(first_row)
-        except StopIteration:
-            print(f"The file '{file_path}' is empty.")
-            return False
-
-        if not any(first_row):
-            print(f"The file '{file_path}' has no data.")
-            return False
-
-    return True
-
-"""
-
-
-
 def blocking_io(member_id):
     try:
         time.sleep(1)
@@ -274,8 +235,6 @@ async def on_reaction_add(reaction, user):
         print(f"on_reaction_add Error: {e}")
 
 
-
-
 @bot.command(name='update_leaderboard')
 async def update_leaderboard(ctx, num_results: int = 10):
     if ctx.author.id == 811235357663297546:
@@ -306,7 +265,6 @@ async def update_leaderboard(ctx, num_results: int = 10):
         top_data_pairs = sorted_data_pairs[:num_results]    
         """
     
-    
         # remove huggingfolks
         guild = ctx.guild
         role = discord.utils.get(guild.roles, id=897376942817419265)
@@ -333,106 +291,80 @@ async def update_leaderboard(ctx, num_results: int = 10):
         await message.edit(content=f"Updated Leaderboard:\n```\n{new_table}\n```")
 
 
-
-
 @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)
 
 
-
 @bot.command()
 async def write_csv(ctx):
-    worksheet = gspread_bot.open("levelbot").sheet1
-    values = worksheet.get_all_values()
-    target_id = '811235357663297546'
-    variable_xp = 0
-    
-    csv_file_path = 'test.csv'
-    with open(csv_file_path, 'w', newline='') as csv_file:
-        csv_writer = csv.writer(csv_file)
-        csv_writer.writerows(values)
-
-
-    # modifying csv data 
-    with open(csv_file_path, 'r') as csv_file:
-        csv_reader = csv.reader(csv_file)
-        rows = list(csv_reader)
-    
-        for row in rows:
-            print(row)
-            if target_id in row:
-                index_of_target = row.index(target_id)
+    if ctx.author.id == 811235357663297546:
+        worksheet = gspread_bot.open("levelbot").sheet1
+        values = worksheet.get_all_values()
+        target_id = '811235357663297546'
+        variable_xp = 0
+        
+        csv_file_path = 'test.csv'
+        with open(csv_file_path, 'w', newline='') as csv_file:
+            csv_writer = csv.writer(csv_file)
+            csv_writer.writerows(values)
     
-                # 2 to right
-                cell_to_increment_index = index_of_target + 2
+        # modifying csv data 
+        with open(csv_file_path, 'r') as csv_file:
+            csv_reader = csv.reader(csv_file)
+            rows = list(csv_reader)
+        
+            for row in rows:
+                print(row)
+                if target_id in row:
+                    index_of_target = row.index(target_id)
+        
+                    # 2 to right
+                    cell_to_increment_index = index_of_target + 2
+        
+                    # +10
+                    try:
+                        row[cell_to_increment_index] = str(int(row[cell_to_increment_index]) + variable_xp)
+                        print(row[cell_to_increment_index])
+                    except (IndexError, ValueError):
+                        print(f"Error: Unable to increment cell at index {cell_to_increment_index}")
+        
+                    break
     
-                # +10
-                try:
-                    row[cell_to_increment_index] = str(int(row[cell_to_increment_index]) + variable_xp)
-                    print(row[cell_to_increment_index])
-                except (IndexError, ValueError):
-                    print(f"Error: Unable to increment cell at index {cell_to_increment_index}")
+        # write modified data and send in discord
+        with open(csv_file_path, 'w', newline='') as csv_file:
+            csv_writer = csv.writer(csv_file)
+            csv_writer.writerows(rows)
     
-                break
-
-
-    # write modified data and send in discord
-    with open(csv_file_path, 'w', newline='') as csv_file:
-        csv_writer = csv.writer(csv_file)
-        csv_writer.writerows(rows)
-
-    csv_data_str = '\n'.join(','.join(row) for row in rows)
-    csv_file = discord.File(csv_file_path)
-
-    await ctx.channel.send(content="csv", file=csv_file)
+        csv_data_str = '\n'.join(','.join(row) for row in rows)
+        csv_file = discord.File(csv_file_path)
     
-    #target_user = await bot.fetch_user(int(target_id))
-    #await target_user.send(content="csv", file=csv_file)
-
-
-
-                
-
-    # do simple operation on file (change one value)
-
-    # post current version to discord
-
-    # recover csv from discord channel, read into memory
+        await ctx.channel.send(content="csv", file=csv_file)
+        
+        #target_user = await bot.fetch_user(int(target_id))
+        #await target_user.send(content="csv", file=csv_file)
 
 
 @bot.command()
 async def read_csv(ctx):
-    async for message in ctx.channel.history(limit=10):
-        if message.attachments:
-            attachment = message.attachments[0]
-            if attachment.filename.endswith('.csv'):
-                csv_content = await attachment.read()
-                csv_data = []
-                csv_lines = csv_content.decode('utf-8').splitlines()
-                csv_reader = csv.reader(csv_lines)
-                for row in csv_reader:
-                    csv_data.append(row)
-                print(csv_data)
-
-                await ctx.send("discord -> read -> updated HF Hub")
-                # react to the given message, helps with explainability
-                await message.add_reaction('✅')
-                break
-
-
-
-"""
-@bot.command(name='send_message')
-async def send_message(ctx):
-    channel = ctx.channel
-    leaderboard_data = [("User 1", 100), ("User 2", 80), ("User 3", 60)]
-    table = tabulate(leaderboard_data, headers=["User", "Score"], tablefmt="plain")
-    message = await channel.send(f"Leaderboard:\n```\n{table}\n```")
-
-"""
-
+    if ctx.author.id == 811235357663297546:
+        async for message in ctx.channel.history(limit=10):
+            if message.attachments:
+                attachment = message.attachments[0]
+                if attachment.filename.endswith('.csv'):
+                    csv_content = await attachment.read()
+                    csv_data = []
+                    csv_lines = csv_content.decode('utf-8').splitlines()
+                    csv_reader = csv.reader(csv_lines)
+                    for row in csv_reader:
+                        csv_data.append(row)
+                    print(csv_data)
+    
+                    await ctx.send("discord -> read -> updated HF Hub")
+                    # react to the given message, helps with explainability
+                    await message.add_reaction('✅')
+                    break
 
 
 # embeds with user pfps? 
@@ -458,7 +390,57 @@ 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()    
+
+URL = "https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w/edit#gid=0"
+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
+
+# csv
+# read into pandas dataframe1
+# read levels column and create pandas dataframe2 with first column containing levels from 2-max found in dataframe1
+# create second column in dataframe2 for number of each level found in dataframe1 levels column
+
+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 dataframe1.iloc[:, 3]:
+        counts[value] = counts.get(value, 0) + 1
+    dataframe2['Members'] = dataframe2['Levels'].map(counts)
+
+    print("Dataframe 1:")
+    print(dataframe1)
+    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):
+
+demo.queue(default_concurrency_limit=40).launch()