Update app.py

app.py

@@ -86,6 +86,7 @@ def add_new_eval(
         else:
             LLM_name = LLM_type
         
+        
         if model_link == '':
             model_name = model_name_textbox  # no url
         else:
@@ -112,9 +113,21 @@ def add_new_eval(
             overall_accuracy]
         # pdb.set_trace()
         csv_data = pd.read_csv(CSV_DIR)
-        col = csv_data.shape[0]
-        csv_data.loc[col] = new_data
-        csv_data = csv_data.to_csv(CSV_DIR, index=False)
+
+        # pdb.set_trace()
+        if revision_name_textbox == '':
+            col = csv_data.shape[0]
+            csv_data.loc[col] = new_data
+            csv_data = csv_data.to_csv(CSV_DIR, index=False)
+        else:
+            model_name_list = csv_data['Model']
+            name_list = [name.split(']')[0][1:] for name in model_name_list]
+            if revision_name_textbox not in name_list:
+                col = csv_data.shape[0]
+            else:
+                col = name_list.index(revision_name_textbox)    
+            csv_data.loc[col] = new_data
+            csv_data = csv_data.to_csv(CSV_DIR, index=False)
     return 0
 
 def get_baseline_df():