Update app.py

app.py

@@ -183,21 +183,6 @@ def initialize_leaderboard_file():
             "Total Questions", "Timestamp", "Team Name"
         ]).to_csv(LEADERBOARD_FILE, index=False)
 
-def initialize_leaderboard_pro_file():
-    """
-    Ensure the leaderboard file exists and has the correct headers.
-    """
-    if not os.path.exists(LEADERBOARD_FILE_pro):
-        pd.DataFrame(columns=[
-            "Model Name", "Overall Accuracy", "Correct Predictions", 
-            "Total Questions", "Timestamp", "Team Name"
-        ]).to_csv(LEADERBOARD_FILE_pro, index=False)
-    elif os.stat(LEADERBOARD_FILE_pro).st_size == 0:
-        pd.DataFrame(columns=[
-            "Model Name", "Overall Accuracy", "Correct Predictions", 
-            "Total Questions", "Timestamp", "Team Name"
-        ]).to_csv(LEADERBOARD_FILE_pro, index=False)        
-
 def clean_answer(answer):
     if pd.isna(answer):
         return None
@@ -306,22 +291,22 @@ def update_leaderboard_pro(results):
     try:
         # Update the local leaderboard file
         new_entry_df = pd.DataFrame([new_entry])
-        file_exists = os.path.exists(LEADERBOARD_FILE_pro)
+        file_exists = os.path.exists(LEADERBOARD_FILE)
         
         new_entry_df.to_csv(
-            LEADERBOARD_FILE_pro,
+            LEADERBOARD_FILE,
             mode='a',  # Append mode
             index=False,
             header=not file_exists  # Write header only if the file is new
         )
-        print(f"Leaderboard updated successfully at {LEADERBOARD_FILE_pro}")
+        print(f"Leaderboard updated successfully at {LEADERBOARD_FILE}")
 
         # Push the updated file to the Hugging Face repository using HTTP API
         api = HfApi()
         token = HfFolder.get_token()
         
         api.upload_file(
-            path_or_fileobj=LEADERBOARD_FILE_pro,
+            path_or_fileobj=LEADERBOARD_FILE,
             path_in_repo="leaderboardPro.csv",
             repo_id="SondosMB/Mobile-MMLU",  # Your Space repository
             repo_type="space",
@@ -358,19 +343,6 @@ def load_leaderboard():
             
         })
     return pd.read_csv(LEADERBOARD_FILE)
-
-def load_leaderboard_pro():
-    if not os.path.exists(LEADERBOARD_FILE_pro) or os.stat(LEADERBOARD_FILE_pro).st_size == 0:
-        return pd.DataFrame({
-            "Model Name": [],
-            "Overall Accuracy": [],
-            "Correct Predictions": [],
-            "Total Questions": [],
-            "Timestamp": [],
-            "Team Name": [],
-            
-        })
-    return pd.read_csv(LEADERBOARD_FILE_pro)
     
 # def evaluate_predictions(prediction_file, model_name, add_to_leaderboard):
 #     try:
@@ -514,7 +486,7 @@ def evaluate_predictions_pro(prediction_file, model_name,Team_name ,add_to_leade
         missing_columns = [col for col in required_columns if col not in predictions_df.columns]
         if missing_columns:
             return (f"Error: Missing required columns in prediction file: {', '.join(missing_columns)}.",
-                    load_leaderboard_pro())
+                    load_leaderboard())
 
         # Validate 'Answer' column in ground truth file
         if 'Answer' not in ground_truth_df.columns:
@@ -544,7 +516,7 @@ def evaluate_predictions_pro(prediction_file, model_name,Team_name ,add_to_leade
   
     except Exception as e:
         return f"Error during evaluation: {str(e)}", load_leaderboard_pro()
-initialize_leaderboard_pro_file()
+initialize_leaderboard_file()
 
 
 # Function to set default mode
@@ -1084,14 +1056,14 @@ with gr.Blocks(css=css_tech_theme) as demo:
             )
         with gr.TabItem("🏅 Leaderboard-pro"):
             leaderboard_table = gr.Dataframe(
-                value=load_leaderboard_pro(),
+                value=load_leaderboard,
                 label="Leaderboard",
                 interactive=False,
                 wrap=True,
             )
-            refresh_button_pro = gr.Button("Refresh Leaderboard-pro")
-            refresh_button_pro.click(
-                lambda: load_leaderboard_pro(),
+            refresh_button = gr.Button("Refresh Leaderboard")
+            refresh_button.click(
+                lambda: load_leaderboard,
                 inputs=[],
                 outputs=[leaderboard_table],
             )