Update app.py

app.py

@@ -74,12 +74,24 @@ def map_names(odf,fname):
     return mapped_frame,msg
 
 def get_lob(df):
-  column_names = df.columns
-  for pattern in ["lob", "market_segment", "product", "class_of_business",'type']:
-    matching_columns = [col for col in column_names if pattern in col.lower()]
-    if matching_columns:
-      return sorted(matching_columns, key=len)[0]
-  return None
+    global g_mapping
+    if g_mapping is None:
+        g_mapping = download_csv_as_dataframe('https://docs.google.com/spreadsheets/d/1rVoLrrTEDzU79x2H2Z1lJ7-z_jRbt-NMUdTarjLvSGo/edit?usp=drive_link')
+    mapping = g_mapping
+    
+    column_names = set(df.columns)
+    best_match_col = None
+    max_matches = 0
+    for pattern in ["lob", "market_segment", "product", "class_of_business", 'type']:
+        matching_columns = {col for col in column_names if pattern in col.lower()}
+
+        for col in matching_columns:
+            matches = sum(df[col].isin(g_mapping['original']))
+            if matches > max_matches:
+                best_match_col = col
+                max_matches = matches
+        column_names -= matching_columns
+    return best_match_col if max_matches > 0 else None
 
 def get_paid_amount(df):
   for col in df.columns: