Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import pandas as pd
 import json
 import tempfile
+import datetime
 
 from constants import *
 from huggingface_hub import Repository
@@ -26,9 +27,14 @@ def add_new_eval(
     if input_file is None:
         return "Error! Empty file!"
 
-    upload_data=json.loads(input_file)
+    # upload_data=json.loads(input_file)
+    upload_content = input_file.read()
     submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
     submission_repo.git_pull()
+    filename = now.strftime("%Y%m%d_%H%M%S")
+    now = datetime.datetime.now()
+    with open(f'{filename}.zip','wb') as f:
+        f.write(input_file)
     shutil.copyfile(CSV_DIR, os.path.join(SUBMISSION_NAME, f"{input_file}"))
 
     csv_data = pd.read_csv(CSV_DIR)
@@ -50,13 +56,24 @@ def add_new_eval(
     else:
         model_name = '[' + model_name + '](' + model_link + ')'
 
+    os.makedirs(filename, exist_ok=True)
+    with zipfile.ZipFile(io.BytesIO(input_file.read()), 'r') as zip_ref:
+        zip_ref.extractall(filename)
+
+    upload_data = {}
+    for file in os.listdir(filename):
+        with open(os.path.join(filename, file)) as ff:
+            cur_json = json.load(ff)
+            if isinstance(cur_json, dict):
+                for key in cur_json:
+                    upload_data[key] = cur_json[key][0]
     # add new data
     new_data = [
         model_name
         ]
     for key in TASK_INFO:
         if key in upload_data:
-            new_data.append(upload_data[key][0])
+            new_data.append(upload_data[key])
         else:
             new_data.append(0)
     csv_data.loc[col] = new_data