yk07577 commited on
Commit
84caa9d
·
verified ·
1 Parent(s): 233eddf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -114,7 +114,13 @@ def upload_csv(file, history):
114
  global global_df, uploaded_filename, last_uploaded_company
115
  try:
116
  uploaded_filename = os.path.splitext(os.path.basename(file.name))[0]
117
- last_uploaded_company = uploaded_filename
 
 
 
 
 
 
118
  df = pd.read_csv(file.name)
119
 
120
  # Validate necessary columns
 
114
  global global_df, uploaded_filename, last_uploaded_company
115
  try:
116
  uploaded_filename = os.path.splitext(os.path.basename(file.name))[0]
117
+ if "apple" in uploaded_filename:
118
+ last_uploaded_company = "apple"
119
+ elif "amazon" in uploaded_filename:
120
+ last_uploaded_company = "amazon"
121
+ else:
122
+ last_uploaded_company = uploaded_filename
123
+
124
  df = pd.read_csv(file.name)
125
 
126
  # Validate necessary columns