Update app.py

app.py

@@ -16,7 +16,6 @@ import logging
 import time
 import pandas as pd
 from apscheduler.schedulers.background import BackgroundScheduler
-import sys
 
 import gradio_client
 import gradio as gr
@@ -56,31 +55,32 @@ async def on_ready():
     # 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 = pd.read_csv("https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w/export?format=csv&gid=0")
-    #print(f"csv successfully retrieved: {data}")
-    #data.to_csv(csv_file, index=False)
+    data = pd.read_csv("https://docs.google.com/spreadsheets/d/1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w/export?format=csv&gid=0")
+    print(f"csv successfully retrieved: {data}")
+    data.to_csv(csv_file, index=False)
 
 
-def update_google_sheet():
+async def update_google_sheet():
     """save data from HF Space -> google sheets (makes 1 API call)"""
     print("test")
     data = open(csv_file, 'r').read()
-    print("test")
+        print("test")
     # 1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w = live
     # 1YJEwP7kZChI3Z2GKv4zFEfwwyQrO3zwXclv0QMrkOqA = test
     spreadsheet_id = "1hQSsIg1Y9WdBF_CdNM1L1rUUREoxKqRTe3_ILo-WK8w"
     print("test")
-    gspread_bot.import_csv(spreadsheet_id, data)
+    gspread_bot.import_csv('', data)
     print({f"Google sheet {spreadsheet_id} successfully updated!"})
 
+
 """
 scheduler = BackgroundScheduler()
 scheduler.add_job(update_google_sheet, "interval", seconds=60)
 scheduler.start()
-
 """
 
 
+
 def calculate_level(xp):
     return int(xp ** (1.0 / 3.0))
 
@@ -184,8 +184,6 @@ async def add_exp(member_id):
                 new_xp = old_xp + XP_PER_MESSAGE
                 print(f"new_xp = old_xp + XP_PER_MESSAGE / {new_xp} = {old_xp} + {XP_PER_MESSAGE}")
                 data.loc[index, 'discord_exp'] = new_xp # do not change column name
-                test = data.loc[index, 'discord_exp']
-                print(test)
                 print(f"Record for {member} updated from {old_xp} to {new_xp} (+{XP_PER_MESSAGE}) ")
           
                 # level up
@@ -217,10 +215,11 @@ async def add_exp(member_id):
             if current_level == 1:
                 if lvl1 not in member.roles:
                     await member.add_roles(lvl1)
-                    print(f"Gave {member} {lvl1}")  # can log this better          
+                    print(f"Gave {member} {lvl1}")  # can log this better  
 
         if member_id == 811235357663297546:
-            update_google_sheet()
+            await update_google_sheet()
+
     except Exception as e:
         print(f"add_exp Error: {e}")   
 
@@ -231,11 +230,6 @@ async def on_message(message):
         if message.author.id not in bot_ids: # could change to if author does not have bot role (roleid)
             if "!help_xp" not in message.content:
                 print(f"adding exp from message {message.author}")
-                if message.author.id == 811235357663297546:
-                    if message.content == '!stop':
-                        print('Stopping bot...')
-                        await bot.close()
-                        sys.exit()
                 await asyncio.sleep(1)
                 await add_exp(message.author.id)
                 #await periodic_api_test() # run this ~1 per day otherwise will lead to issues
@@ -336,7 +330,14 @@ def run_bot():
     bot.run(DISCORD_TOKEN)
 threading.Thread(target=run_bot).start()
 
+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
@@ -345,7 +346,18 @@ threading.Thread(target=run_bot).start()
 
 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)
@@ -353,7 +365,19 @@ with demo:
         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):