Update app.py

app.py

@@ -26,10 +26,12 @@ def add_new_eval(
     model_name_textbox: str,
     revision_name_textbox: str,
     model_link: str,
+    contact_email: str
 ):
     if input_file is None:
         return "Error! Empty file!"
-
+    if  model_link == '' or model_name == '' or contact_email == '':
+        return gr.update(visible=True), gr.update(visible=False), gr.update(visible=True)
     # upload_data=json.loads(input_file)
     upload_content = input_file
     submission_repo = Repository(local_dir=SUBMISSION_NAME, clone_from=SUBMISSION_URL, use_auth_token=HF_TOKEN, repo_type="dataset")
@@ -53,7 +55,6 @@ def add_new_eval(
             col = csv_data.shape[0]
         else:
             col = name_list.index(revision_name_textbox)    
-    
     if model_link == '':
         model_name = model_name  # no url
     else:
@@ -97,18 +98,19 @@ def add_new_eval(
             new_data.append(upload_data[key])
         else:
             new_data.append(0)
-    new_data.append("User Upload.")
+    new_data.append("User Upload")
+    new_data.append(contact_email)
     csv_data.loc[col] = new_data
     csv_data = csv_data.to_csv(CSV_DIR, index=False)
     submission_repo.push_to_hub()
     print("success update", model_name)
-    return gr.update(visible=False), gr.update(visible=True)
+    return gr.update(visible=False), gr.update(visible=True), gr.update(visible=False)
 
 def get_normalized_df(df):
     # final_score = df.drop('name', axis=1).sum(axis=1)
     # df.insert(1, 'Overall Score', final_score)
     normalize_df = df.copy().fillna(0.0)
-    for column in normalize_df.columns[1:-1]:
+    for column in normalize_df.columns[1:-2]:
         min_val = NORMALIZE_DIC[column]['Min']
         max_val = NORMALIZE_DIC[column]['Max']
         normalize_df[column] = (normalize_df[column] - min_val) / (max_val - min_val)
@@ -505,15 +507,18 @@ with block:
             with gr.Row():
                 with gr.Column():
                     model_name_textbox = gr.Textbox(
-                        label="Model name", placeholder="LaVie"
+                        label="**Model name**", placeholder="Required field"
                         )
                     revision_name_textbox = gr.Textbox(
-                        label="Revision Model Name", placeholder="LaVie"
+                        label="Revision Model Name(Optional)", placeholder="LaVie"
                     )
 
                 with gr.Column():
                     model_link = gr.Textbox(
-                        label="Model Link", placeholder="https://huggingface.co/decapoda-research/llama-7b-hf"
+                        label="**Project Page/Paper Link**", placeholder="Required field"
+                    )
+                    contact_email = gr.Textbox(
+                        label="E-Mail(**Will not be displayed**)", placeholder="Required field"
                     )
 
 
@@ -521,7 +526,8 @@ with block:
 
                 input_file = gr.components.File(label = "Click to Upload a ZIP File", file_count="single", type='binary')
                 submit_button = gr.Button("Submit Eval")
-                submit_succ_button = gr.Button("Submit Success! Please press refresh and return to LeaderBoard!", visible=False, interactive=False)
+                submit_succ_button = gr.Markdown("Submit Success! Please press refresh and return to LeaderBoard!", visible=False)
+                fail_textbox = gr.Markdown('<span style="color:red;">Please ensure that the `Model Name`, `Project Page`, and `Email` are filled in correctly.</span>', elem_classes="markdown-text",visible=False)
                 
     
                 submission_result = gr.Markdown()
@@ -532,8 +538,9 @@ with block:
                         model_name_textbox,
                         revision_name_textbox,
                         model_link,
+                        contact_email
                     ],
-                    outputs=[submit_button, submit_succ_button]
+                    outputs=[submit_button, submit_succ_button, fail_textbox]
                 )