Update app.py

app.py

@@ -60,12 +60,14 @@ def process_row(row, vendor_data, vendor_scores):
     return combined_row
 
 # Function to process file
-def process_file(vendor_name, mode, file):
+def process_file(vendor_name_input, mode, file):
     global vendor_df
     global vectorizer
 
+    # Sanitize the vendor name input
+    vendor_name = vendor_name_input.strip().lower().replace(" ", "_")
+
     if mode == 'Upload Vendor File':
-        vendor_name = os.path.splitext(os.path.basename(file.name))[0]  
         vendor_df = pd.read_excel(file.name)
         vendor_df.iloc[:, 2] = vendor_df.iloc[:, 2].apply(preprocess_text)
         vendor_df['score_vendor'] = vendor_df.iloc[:, 4].apply(parse_number)
@@ -102,11 +104,10 @@ def process_file(vendor_name, mode, file):
         vendor_scores = {}
         for csv_file, vector_file, score_file in zip(csv_files, vector_files, score_files):
             with open(os.path.join('data', vector_file), 'rb') as f:
-                vendor_name = os.path.splitext(vector_file)[0].replace('_vectors','')
                 vendor_vectors = pickle.load(f)
                 vendor_data[vendor_name] = vendor_vectors
             with open(os.path.join('data', score_file), 'rb') as f:
-                vendor_scores.update(pickle.load(f))
+                vendor_scores[vendor_name] = pickle.load(f)
 
         client_df = pd.read_excel(file.name)
         client_df.iloc[:, 2] = client_df.iloc[:, 2].fillna('3 - Medium')