Update app.py

app.py

@@ -1,17 +1,13 @@
 import gradio as gr
 from evaluate import run_evaluation
 import pandas as pd
-import os
-print(os.getenv('VG_DATA'))
-print(os.getenv('test'))
 
 LEADERBOARD_CSV = "leaderboard.csv"
-import pandas as pd
 import os
 from datetime import datetime, date
 
 SUBMIT_RECORD = "submissions.csv"
-MAX_SUBMIT_PER_DAY = 2
+MAX_SUBMIT_PER_DAY = 5
 
 def check_submission_limit(username):
     if not os.path.exists(SUBMIT_RECORD):
@@ -35,23 +31,52 @@ def record_submission(username):
         df = pd.DataFrame(columns=["username", "timestamp"])
     df.loc[len(df)] = {"username": username, "timestamp": now}
     df.to_csv(SUBMIT_RECORD, index=False)
+import json
+def check_user_valid(username, password):
+    with open("users.json") as f:
+        users = json.load(f)
+    return username in users and users.get(username) == password
 
-def evaluate_and_update(pred_file, username):
-    if not check_submission_limit(username):
-        return "⛔ Submission limit exceeded for today.", pd.read_csv(LEADERBOARD_CSV)
-    score = run_evaluation(pred_file.name)
-    if os.path.exists(LEADERBOARD_CSV):
-        df = pd.read_csv(LEADERBOARD_CSV)
+def update_leaderboard(username, score1, score2, score3):
+    # 追加记录到总提交记录
+    if os.path.exists("leaderboard.csv"):
+        sub_df = pd.read_csv("leaderboard.csv")
+    else:
+        sub_df = pd.DataFrame(columns=["username", "gtop-1@0.25", "AP@0.25", "AR@0.25"])
+    sub_df.loc[len(sub_df)] = [username, score1, score2, score3]
+    sub_df.to_csv("leaderboard.csv", index=False)
+
+    # 从所有提交中挑选每用户score1最高的记录
+    top_df = sub_df.sort_values("gtop-1@0.25", ascending=False).drop_duplicates("username", keep="first")
+    top_df = top_df.sort_values("gtop-1@0.25", ascending=False)
+    top_df.to_csv("leaderboard.csv", index=False)
+    return top_df
+
+def show_lb():
+    if os.path.exists("leaderboard.csv"):
+        sub_df = pd.read_csv("leaderboard.csv")
     else:
-        df = pd.DataFrame(columns=["name", "score"])
-    df.loc[len(df)] = {"name": username, "score": score}
-    df = df.sort_values(by="score", ascending=False).reset_index(drop=True)
-    df.to_csv(LEADERBOARD_CSV, index=False)
-    return f"Your score: {score:.4f}", df
+        sub_df = pd.DataFrame(columns=["username", "gtop-1@0.25", "AP@0.25", "AR@0.25"])
+    return sub_df
+def evaluate_and_update(pred_file, username,password):
+    
+    if not check_submission_limit(username):
+        return "⛔ Submission limit exceeded for today.", pd.DataFrame()
+    if not check_user_valid(username, password):
+        return "❌ Invalid username or password", pd.DataFrame()
+    score1, score2, score3 = run_evaluation(pred_file.name)
+    record_submission(username)
+    leaderboard_df = update_leaderboard(username, score1, score2, score3)
+
+    return f"✅ gtop-1@0.25: {score1:.4f}, AP@0.25: {score2:.4f}, AR@0.25: {score3:.4f}", leaderboard_df
+
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🧊 3D IoU Challenge")
-    name = gr.Textbox(label="Username")
+    with gr.Row():
+        username = gr.Textbox(label="Username")
+        password = gr.Textbox(label="Password", type="password")
     upload = gr.File(label="Upload your prediction (.json)")
     score_text = gr.Textbox(label="Evaluation score")
     leaderboard = gr.Dataframe(headers=["Name", "Score"], interactive=False)
@@ -59,8 +84,10 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Submit")
     submit_btn.click(
         fn=evaluate_and_update,
-        inputs=[upload, name],
+        inputs=[upload, username,password],
         outputs=[score_text, leaderboard]
     )
+    show_btn = gr.Button("Show Leaderboard")
+    show_btn.click(fn=show_lb,inputs=[],outputs=[leaderboard])
 
 demo.launch()