Update app.py

app.py

@@ -7,7 +7,7 @@ import os
 def load_mapping():
     mapping_data = {
         "PO Output Column": ["Column1", "Column2", "Column3"],  # Replace with actual columns
-        "UVM  MMB  POLY STICKER Column": ["Fixed-Value1", "Fixed-Value2", "Fixed-Value3"]  # Replace with mapping logic
+        "UVM  MMB  POLY STICKER Column": ["Sheet1.ColumnA", "Sheet1.ColumnB", "Fixed-Value"]  # Replace with mapping logic
     }
     return pd.DataFrame(mapping_data)
 
@@ -15,7 +15,7 @@ def load_mapping():
 def transform_data(input_path, mapping_df):
     # Load the input workbook
     input_workbook = pd.ExcelFile(input_path)
-    
+
     # Initialize a dictionary to store data for output
     output_data = {col: [] for col in mapping_df["PO Output Column"] if not pd.isna(col)}
 
@@ -30,9 +30,15 @@ def transform_data(input_path, mapping_df):
         # Handle fixed values
         if "Fixed" in input_rule:
             fixed_value = input_rule.split("-")[0].strip()
-            output_data[output_column] = [fixed_value] * 1  # Placeholder for rows
-
-        # TODO: Add logic to map specific columns from input workbook
+            output_data[output_column] = [fixed_value] * len(input_workbook.sheet_names)  # Placeholder for rows
+
+        # Handle column mapping from input workbook
+        elif "." in input_rule:
+            sheet_name, column_name = input_rule.split(".")
+            if sheet_name in input_workbook.sheet_names:
+                sheet_data = pd.read_excel(input_path, sheet_name=sheet_name)
+                if column_name in sheet_data.columns:
+                    output_data[output_column] = sheet_data[column_name].tolist()
 
     return pd.DataFrame(output_data)