add rules to calculate team scores

calculate_personal_scores.py

@@ -1,14 +1,11 @@
 import json
 import os
-import time
 from collections import defaultdict
 from functools import lru_cache
 
 import argilla as rg
-import gradio as gr
 import pandas as pd
 from dotenv import load_dotenv
-from fastapi import FastAPI
 
 load_dotenv()
 
@@ -361,4 +358,4 @@ def calculate_personal_scores():
 
 
 if __name__ == "__main__":
-    calculate_scores()
+    calculate_personal_scores()

calculate_team_scores.py

@@ -1,14 +1,7 @@
-import json
 import os
-import time
-from collections import defaultdict
-from functools import lru_cache
 
-import argilla as rg
-import gradio as gr
 import pandas as pd
 from dotenv import load_dotenv
-from fastapi import FastAPI
 
 from calculate_personal_scores import get_discord_username
 
@@ -31,6 +24,33 @@ TEAM_COLUMNS = {
     "email_5": "Email 5",
 }
 
+# INCLUDE: Cada 100 preguntas validadas extraídas por equipo = 0.5 ptos (máx 1 pto)
+# Estereotipos: Cada 10 estereotipos validados por equipo = 0.1 ptos (máx 1 pto)
+# BLEND: Cada 50 preguntas por equipo = 0.5 ptos (máx 2 ptos)
+# Arena: Cada 50 prompts por equipo = 0.5 ptos (máx 2 ptos)
+
+POINTS = {
+    "INCLUDE": {"points": 0.5, "max_points": 1, "questions": 100},
+    "ESTEREOTIPOS": {"points": 0.1, "max_points": 1, "questions": 10},
+    "BLEND": {"points": 0.5, "max_points": 2, "questions": 50},
+    "ARENA": {"points": 0.5, "max_points": 2, "questions": 50},
+}
+
+
+def calculate_challenge_points(total_score, challenge_name):
+    """Calculate points for a specific challenge based on the POINTS rules."""
+    if challenge_name not in POINTS:
+        return 0
+
+    challenge_rules = POINTS[challenge_name]
+    points_per_unit = challenge_rules["points"]
+    max_points = challenge_rules["max_points"]
+    questions_per_unit = challenge_rules["questions"]
+
+    calculated_points = (total_score / questions_per_unit) * points_per_unit
+    final_points = min(calculated_points, max_points)
+    return round(final_points, 2)
+
 
 def get_team_leaderboard(personal_leaderboard_df):
     """Calculate team leaderboard based on personal scores."""
@@ -82,6 +102,17 @@ def get_team_leaderboard(personal_leaderboard_df):
             while len(discord_usernames) < 5:
                 discord_usernames.append("")
 
+            # Calculate points for each challenge
+            ptos_arena = calculate_challenge_points(team_scores["arena"], "ARENA")
+            ptos_blend_es = calculate_challenge_points(team_scores["blend_es"], "BLEND")
+            ptos_estereotipos = calculate_challenge_points(
+                team_scores["estereotipos"], "ESTEREOTIPOS"
+            )
+            ptos_include = calculate_challenge_points(team_scores["include"], "INCLUDE")
+
+            # Calculate total points
+            ptos_total = ptos_arena + ptos_blend_es + ptos_estereotipos + ptos_include
+
             # Create team row
             team_row_data = {
                 "team_name": team_name,
@@ -91,22 +122,22 @@ def get_team_leaderboard(personal_leaderboard_df):
                 "discord_4": discord_usernames[3],
                 "discord_5": discord_usernames[4],
                 "total_arena": team_scores["arena"],
-                "ptos_arena": 0,  # Set to 0 for now as requested
+                "ptos_arena": ptos_arena,
                 "total_blend_es": team_scores["blend_es"],
-                "ptos_blend_es": 0,  # Set to 0 for now as requested
+                "ptos_blend_es": ptos_blend_es,
                 "total_estereotipos": team_scores["estereotipos"],
-                "ptos_estereotipos": 0,  # Set to 0 for now as requested
+                "ptos_estereotipos": ptos_estereotipos,
                 "total_include": team_scores["include"],
-                "ptos_include": 0,  # Set to 0 for now as requested
-                "ptos_total": 0,  # Set to 0 for now as requested
+                "ptos_include": ptos_include,
+                "ptos_total": ptos_total,
             }
 
             team_leaderboard.append(team_row_data)
 
-        # Create DataFrame and sort by total_arena
+        # Create DataFrame and sort by total points (ptos_total) instead of just arena
         if team_leaderboard:
             team_df = pd.DataFrame(team_leaderboard)
-            team_df.sort_values("total_arena", ascending=False, inplace=True)
+            team_df.sort_values("ptos_total", ascending=False, inplace=True)
             return team_df
         else:
             return pd.DataFrame()
@@ -120,6 +151,8 @@ def calculate_team_scores():
     personal_leaderboard_df = pd.read_csv(LEADERBOARD_PERSONAL_CSV)
     team_leaderboard_df = get_team_leaderboard(personal_leaderboard_df)
     team_leaderboard_df.to_csv(LEADERBOARD_EQUIPOS_CSV, index=False, encoding="utf-8")
+    print("Team scores calculated and saved successfully!")
+    print(f"Generated leaderboard with {len(team_leaderboard_df)} teams")
 
 
 if __name__ == "__main__":

leaderboard_equipos.csv

@@ -1,13 +1,13 @@
 team_name,discord_1,discord_2,discord_3,discord_4,discord_5,total_arena,ptos_arena,total_blend_es,ptos_blend_es,total_estereotipos,ptos_estereotipos,total_include,ptos_include,ptos_total
-Cresia,roverico,cresia.aillm,alberto.medina,,,140,0,0,0,0,0,304,0,0
-TralaleloTralala-MemeAlign,andres_seba,peroloso,,,,120,0,0,0,0,0,440,0,0
-IberoTales,mcdaqc,,,,,118,0,0,0,0,0,449,0,0
-SabiduriaPopular Castellana,alvaro8gb,enrique.solera.navarro,angustias22,comasberto,,105,0,0,0,0,0,2,0,0
-Think Paraguayo,enpaiva93,luceldasilva,edmenciab,,,93,0,0,0,0,0,505,0,0
-HoCV-COL,dreamripper1,,,,,83,0,0,0,0,0,987,0,0
-Refranero Afro-Cubano,rasel3132,bel21093,,,,0,0,0,0,0,0,0,0,0
-Comida Colombia + Ecuador,pablo.ce,magodyboy,nestoralarcon.ed,diegomarquez2008,,0,0,0,0,0,0,2830,0,0
-Equipo LeIA,susanazhou,cosntanzapjeldres,,,,0,0,0,0,0,0,560,0,0
-PeruWhisper,vandread13,,,,,0,0,0,0,0,0,0,0,0
-Falsos_Amigos,yeisonm3011,erick bustamante,dinho9779,,,0,0,0,0,0,0,0,0,0
-MP,rvk6212,,,,,0,0,0,0,0,0,0,0,0
+Cresia,roverico,cresia.aillm,alberto.medina,,,140,1.4,0,0.0,0,0.0,304,1.0,2.4
+TralaleloTralala-MemeAlign,andres_seba,peroloso,,,,120,1.2,0,0.0,0,0.0,440,1.0,2.2
+IberoTales,mcdaqc,,,,,118,1.18,0,0.0,0,0.0,449,1.0,2.1799999999999997
+Think Paraguayo,enpaiva93,luceldasilva,edmenciab,,,93,0.93,0,0.0,0,0.0,505,1.0,1.9300000000000002
+HoCV-COL,dreamripper1,,,,,83,0.83,0,0.0,0,0.0,987,1.0,1.83
+SabiduriaPopular Castellana,alvaro8gb,enrique.solera.navarro,angustias22,comasberto,,105,1.05,0,0.0,0,0.0,2,0.01,1.06
+Comida Colombia + Ecuador,pablo.ce,magodyboy,nestoralarcon.ed,diegomarquez2008,,0,0.0,0,0.0,0,0.0,2830,1.0,1.0
+Equipo LeIA,susanazhou,cosntanzapjeldres,,,,0,0.0,0,0.0,0,0.0,560,1.0,1.0
+Refranero Afro-Cubano,rasel3132,bel21093,,,,0,0.0,0,0.0,0,0.0,0,0.0,0.0
+PeruWhisper,vandread13,,,,,0,0.0,0,0.0,0,0.0,0,0.0,0.0
+Falsos_Amigos,yeisonm3011,erick bustamante,dinho9779,,,0,0.0,0,0.0,0,0.0,0,0.0,0.0
+MP,rvk6212,,,,,0,0.0,0,0.0,0,0.0,0,0.0,0.0