Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import pandas as pd
 import os
 import shutil
-import tempfile
 
 # Description and Introduction texts
 DESCRIPTION = """
@@ -54,39 +53,24 @@ def add_new_entry(file, password):
     if password != UPLOAD_SECRET:
         return df, "Incorrect password. Upload failed."
     
-    try:
-        # Create a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.csv') as temp_file:
-            temp_file.write(file.read())
-            temp_file_path = temp_file.name
-
-        # Read the CSV file
-        new_df = pd.read_csv(temp_file_path)
-        
-        columns_order = [
-            "Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count", 
-            "input_length", "output_length"
-        ]
-        for col in columns_order:
-            if col not in new_df.columns:
-                new_df[col] = pd.NA
-        new_df = new_df[columns_order]
-        
-        # Append the new data to the existing DataFrame
-        df = pd.concat([df, new_df], ignore_index=True)
-        
-        # Move the temporary file to the CSV folder
-        filename = os.path.basename(file.name)
-        destination = os.path.join(csv_folder_path, filename)
-        shutil.move(temp_file_path, destination)
-        
-        return df, f"File '{filename}' uploaded and data added successfully!"
-    except Exception as e:
-        return df, f"An error occurred: {str(e)}"
-    finally:
-        # Ensure the temporary file is removed if it still exists
-        if 'temp_file_path' in locals() and os.path.exists(temp_file_path):
-            os.unlink(temp_file_path)
+    new_df = pd.read_csv(file.name)
+    
+    columns_order = [
+        "Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count", 
+        "Input_Tokens", "Output_Tokens"
+    ]
+    for col in columns_order:
+        if col not in new_df.columns:
+            new_df[col] = pd.NA
+    new_df = new_df[columns_order]
+    
+    df = pd.concat([df, new_df], ignore_index=True)
+    
+    filename = os.path.basename(file.name)
+    destination = os.path.join(csv_folder_path, filename)
+    shutil.copy(file.name, destination)
+    
+    return df, f"File '{filename}' uploaded and data added successfully!"
 
 def read_and_process_csv_files(folder_path):
     all_data = []