upload

uploads.py

@@ -24,6 +24,21 @@ def model_hyperlink(link, model_name):
     return f'<a target="_blank" href="{link}" style="color: var(--link-text-color); text-decoration: underline;text-decoration-style: dotted;">{model_name}</a>'
 
 
+def input_verification(model, model_family, forget_rate, url, path_to_file, organisation, mail):
+    for input in [model, model_family, forget_rate, url, organisation]:
+        if input is "":
+            return format_warning("Please fill all the fields.")
+    
+    # Very basic email parsing
+    _, parsed_mail = parseaddr(mail)
+    if not "@" in parsed_mail:
+        return format_warning("Please provide a valid email adress.")
+
+    
+    if path_to_file is None:
+        return format_warning("Please attach a file.")
+    
+    return parsed_mail
 
 def add_new_eval(
     model: str,
@@ -34,20 +49,11 @@ def add_new_eval(
     organisation: str,
     mail: str,
 ):
-    # Very basic email parsing
-    _, parsed_mail = parseaddr(mail)
-    if not "@" in parsed_mail:
-        return format_warning("Please provide a valid email adress.")
-
-    return format_log(f"Model {model} submitted by {organisation} successfully. \nPlease refresh the leaderboard, and wait a bit to see the score displayed")
-    file_value = path_to_file.value
-    if file_value is None:
-        return format_warning("Please attach a file.")
     
+    parsed_mail  = input_verification(model, model_family, forget_rate, url, path_to_file, organisation, mail)
     
-
     # load the file 
-    df = pd.read_csv(file_value)
+    df = pd.read_csv(path_to_file)
 
     # modify the df to include metadata
     df["model"] = model
@@ -55,7 +61,7 @@ def add_new_eval(
     df["forget_rate"] = forget_rate
     df["url"] = url
     df["organisation"] = organisation
-    df["mail"] = mail
+    df["mail"] = parsed_mail
     df["timestamp"] = datetime.datetime.now()
 
     #upload to spaces using the hf api at 
@@ -64,14 +70,20 @@ def add_new_eval(
     file_name = f"{model}-{organisation}-{datetime.datetime.now().strftime('%Y-%m-%d')}.csv"
     
     # upload the df to spaces
-    new_file = df.to_csv(index=False)
+    import io
+    
+    buffer = io.BytesIO()
+    df.to_csv(buffer, index=False)  # Write the DataFrame to a buffer in CSV format
+    buffer.seek(0)  # Rewind the buffer to the beginning
 
     api.upload_file(
         repo_id = RESULTS_PATH,
         path_in_repo = f"{path_in_repo}/{file_name}",
-        file_name = file_value,
+        path_or_fileobj = buffer,
         token=TOKEN,
         repo_type="space",
     )
 
+    return format_log(f"Model {model} submitted by {organisation} successfully. \nPlease refresh the leaderboard, and wait a bit to see the score displayed")
+