Update app.py

app.py

@@ -54,7 +54,7 @@ def generate_table(data):
 protected_methods = ["GPT-4-Turbo-Preview (DFS)", "GPT-3.5-Turbo-1106 (DFS)", "GPT-4-0613 (DFS)", "GPT-3.5-Turbo-0613 (DFS)", "GPT-4-Turbo-Preview (CoT)", "ToolLLaMA v2 (DFS)", "GPT-4-0613 (CoT)", "GPT-3.5-Turbo-1106 (CoT)", "GPT-3.5-Turbo-0613 (CoT)", "ToolLLaMA v2 (CoT)"]
 
 # 合并上传的数据到已有的数据中
-def merge_data(uploaded_data_json, existing_data):
+def merge_data(uploaded_data_json):
     # No need to call json.loads here because uploaded_data is already a Python dict
     new_data = uploaded_data_json
     with scheduler.lock:  # 确保文件操作的线程安全
@@ -94,17 +94,18 @@ def merge_data(uploaded_data_json, existing_data):
     # No need to sort here since we're doing it in the generate_table function
     return existing_data
 
-def process_file(file_info, existing_data):
+def process_file(file_info):
     if file_info is not None:
         # 如果 file_info 是文件的路径字符串，需要使用 'open' 来读取文件
         with open(file_info, "r") as uploaded_file:
             data_content = uploaded_file.read()
             uploaded_data_json = json.loads(data_content)
         # Merge the uploaded data
-        existing_data = merge_data(uploaded_data_json)
+        merge_data(uploaded_data_json)
         scheduler.commit()  # Trigger manual commit after processing
     # If we don't return anything here, gradio will not update the table
     # We need to return the new data for the table if it's interactive
+    existing_data = load_data()
     pass_rate_table = generate_table(existing_data["SolvablePassRateScores"])[1]
     win_rate_table = generate_table(existing_data["SolvableWinRateScores"])[1]
     return pass_rate_table, win_rate_table
@@ -228,7 +229,7 @@ with gr.Blocks() as app:
 
     submit_button.click(
         fn=process_file,
-        inputs= [upload_component,existing_data],
+        inputs= upload_component,
         outputs=[table1, table2]
     )