Update app.py
Browse files
app.py
CHANGED
@@ -54,8 +54,13 @@ def add_new_entry(file, password):
|
|
54 |
return df, "Incorrect password. Upload failed."
|
55 |
|
56 |
try:
|
57 |
-
#
|
58 |
-
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
columns_order = [
|
61 |
"Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count",
|
@@ -69,14 +74,18 @@ def add_new_entry(file, password):
|
|
69 |
# Append the new data to the existing DataFrame
|
70 |
df = pd.concat([df, new_df], ignore_index=True)
|
71 |
|
72 |
-
#
|
73 |
filename = os.path.basename(file.name)
|
74 |
destination = os.path.join(csv_folder_path, filename)
|
75 |
-
|
76 |
|
77 |
return df, f"File '{filename}' uploaded and data added successfully!"
|
78 |
except Exception as e:
|
79 |
return df, f"An error occurred: {str(e)}"
|
|
|
|
|
|
|
|
|
80 |
|
81 |
def read_and_process_csv_files(folder_path):
|
82 |
all_data = []
|
|
|
54 |
return df, "Incorrect password. Upload failed."
|
55 |
|
56 |
try:
|
57 |
+
# Create a temporary file
|
58 |
+
with tempfile.NamedTemporaryFile(delete=False, suffix='.csv') as temp_file:
|
59 |
+
temp_file.write(file.read())
|
60 |
+
temp_file_path = temp_file.name
|
61 |
+
|
62 |
+
# Read the CSV file
|
63 |
+
new_df = pd.read_csv(temp_file_path)
|
64 |
|
65 |
columns_order = [
|
66 |
"Model_Name", "Library", "TTFT", "Tokens-per-Second", "Token_Count",
|
|
|
74 |
# Append the new data to the existing DataFrame
|
75 |
df = pd.concat([df, new_df], ignore_index=True)
|
76 |
|
77 |
+
# Move the temporary file to the CSV folder
|
78 |
filename = os.path.basename(file.name)
|
79 |
destination = os.path.join(csv_folder_path, filename)
|
80 |
+
shutil.move(temp_file_path, destination)
|
81 |
|
82 |
return df, f"File '{filename}' uploaded and data added successfully!"
|
83 |
except Exception as e:
|
84 |
return df, f"An error occurred: {str(e)}"
|
85 |
+
finally:
|
86 |
+
# Ensure the temporary file is removed if it still exists
|
87 |
+
if 'temp_file_path' in locals() and os.path.exists(temp_file_path):
|
88 |
+
os.unlink(temp_file_path)
|
89 |
|
90 |
def read_and_process_csv_files(folder_path):
|
91 |
all_data = []
|