Update app.py

app.py

@@ -41,6 +41,37 @@ HOW_WE_TESTED = """
 
 csv_folder_path = 'result_csv/'
 
+UPLOAD_SECRET = os.getenv("UPLOAD_SECRET")
+
+# ... [Previous functions remain the same: read_and_process_csv_files, get_leaderboard_df, filter_and_search] ...
+
+def add_new_entry(file, password):
+    global df
+    if file is None:
+        return df, "No file uploaded."
+    
+    if password != UPLOAD_SECRET:
+        return df, "Incorrect password. Upload failed."
+    
+    new_df = pd.read_csv(file.name)
+    
+    columns_order = [
+        "Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count", 
+        "Input_Tokens", "Output_Tokens"
+    ]
+    for col in columns_order:
+        if col not in new_df.columns:
+            new_df[col] = pd.NA
+    new_df = new_df[columns_order]
+    
+    df = pd.concat([df, new_df], ignore_index=True)
+    
+    filename = os.path.basename(file.name)
+    destination = os.path.join(csv_folder_path, filename)
+    shutil.copy(file.name, destination)
+    
+    return df, f"File '{filename}' uploaded and data added successfully!"
+
 def read_and_process_csv_files(folder_path):
     all_data = []
     for filename in os.listdir(folder_path):
@@ -53,7 +84,7 @@ def read_and_process_csv_files(folder_path):
     
     columns_order = [
         "Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count", 
-        "input_length", "output_length", "Input", "Output"
+        "input_length", "output_length"
     ]
     
     for col in columns_order:
@@ -67,30 +98,6 @@ df = read_and_process_csv_files(csv_folder_path)
 def get_leaderboard_df():
     return df
 
-def add_new_entry(file):
-    global df
-    if file is None:
-        return df, "No file uploaded."
-    
-    new_df = pd.read_csv(file.name)
-    
-    columns_order = [
-        "Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count", 
-        "input_length", "output_length", "Input", "Output"
-    ]
-    for col in columns_order:
-        if col not in new_df.columns:
-            new_df[col] = pd.NA
-    new_df = new_df[columns_order]
-    
-    df = pd.concat([df, new_df], ignore_index=True)
-    
-    filename = os.path.basename(file.name)
-    destination = os.path.join(csv_folder_path, filename)
-    shutil.copy(file.name, destination)
-    
-    return df, f"File '{filename}' uploaded and data added successfully!"
-
 def filter_and_search(search_term, library_filter):
     filtered_df = df.copy()