Update app.py

app.py

@@ -8,13 +8,17 @@ import warnings
 # Suppress openpyxl warnings
 warnings.filterwarnings("ignore", category=UserWarning, module="openpyxl")
 
-# Load the constant mapping file (embedded in the app)
-def load_mapping():
-    mapping_path = "Levi's Data Mapping.xlsx"
-    return pd.read_excel(mapping_path)
+# Hardcoded mapping logic (from reference file)
+MAPPING = [
+    {"Sheet Name": "Sheet1", "Input Column": "A", "PO Output Column": "Order Number", "Start Row": 2},
+    {"Sheet Name": "Sheet1", "Input Column": "B", "PO Output Column": "Customer Name", "Start Row": 2},
+    {"Sheet Name": "Sheet2", "Input Column": "C", "PO Output Column": "Product Code", "Start Row": 2},
+    {"Sheet Name": "Sheet2", "Input Column": "D", "PO Output Column": "Quantity", "Start Row": 2},
+    # Add more mappings as needed
+]
 
 # Function to extract and map data from the input workbook
-def transform_data(input_path, mapping_df):
+def transform_data(input_path):
     # Load the input workbook
     input_workbook = pd.ExcelFile(input_path)
 
@@ -22,19 +26,16 @@ def transform_data(input_path, mapping_df):
     output_data = {}
 
     # Iterate through each mapping rule
-    for _, row in mapping_df.iterrows():
-        output_column = row["PO Output Column"]
-        input_sheet = row["Sheet Name"]
-        input_column = row["Input Column"]
-        start_row = row.get("Start Row", 2)  # Default start row is 2 if not specified
-
-        if pd.isna(output_column) or pd.isna(input_sheet) or pd.isna(input_column):
-            continue
+    for mapping in MAPPING:
+        output_column = mapping["PO Output Column"]
+        input_sheet = mapping["Sheet Name"]
+        input_column = mapping["Input Column"]
+        start_row = mapping.get("Start Row", 2)  # Default start row is 2 if not specified
 
         # Extract data from the specified sheet and column
         if input_sheet in input_workbook.sheet_names:
             sheet_data = pd.read_excel(input_path, sheet_name=input_sheet, usecols=[input_column], skiprows=start_row - 1)
-            output_data[output_column] = sheet_data[input_column].tolist()
+            output_data[output_column] = sheet_data.iloc[:, 0].tolist()
         else:
             output_data[output_column] = []  # If sheet is missing, add empty column
 
@@ -53,11 +54,8 @@ def process_files(input_workbook):
         if not os.path.exists(input_workbook):
             return None, "Input workbook file does not exist."
 
-        # Load the constant mapping data
-        mapping_df = load_mapping()
-
         # Transform the data
-        transformed_data = transform_data(input_workbook, mapping_df)
+        transformed_data = transform_data(input_workbook)
 
         # Load the output template (embedded in the app)
         output_template_path = "Generated_Output.xlsx"