Update app.py

app.py

@@ -3,6 +3,14 @@ from openpyxl import load_workbook
 import gradio as gr
 import os
 
+# Load the constant mapping file (embedded in the app)
+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
+    }
+    return pd.DataFrame(mapping_data)
+
 # Function to extract and map data from the input workbook
 def transform_data(input_path, mapping_df):
     # Load the input workbook
@@ -29,10 +37,10 @@ def transform_data(input_path, mapping_df):
     return pd.DataFrame(output_data)
 
 # Main processing function
-def process_files(input_workbook, mapping_file, output_template):
+def process_files(input_workbook, output_template):
     try:
-        # Load the data mapping file
-        mapping_df = pd.read_excel(mapping_file, sheet_name="Output")
+        # Load the constant mapping data
+        mapping_df = load_mapping()
 
         # Transform the data
         transformed_data = transform_data(input_workbook, mapping_df)
@@ -56,8 +64,8 @@ def process_files(input_workbook, mapping_file, output_template):
         return f"An error occurred: {e}"
 
 # Define the Gradio interface
-def generate_excel(input_workbook, mapping_file, output_template):
-    result = process_files(input_workbook.name, mapping_file.name, output_template.name)
+def generate_excel(input_workbook, output_template):
+    result = process_files(input_workbook.name, output_template.name)
     if os.path.exists(result):
         return result
     else:
@@ -65,17 +73,16 @@ def generate_excel(input_workbook, mapping_file, output_template):
 
 with gr.Blocks() as app:
     gr.Markdown("# Excel Sheet Generator")
-    gr.Markdown("Upload the input workbook, mapping file, and output template to generate the final Excel file.")
+    gr.Markdown("Upload the input workbook and output template to generate the final Excel file.")
 
     with gr.Row():
         input_workbook = gr.File(label="Input Workbook", file_types=[".xlsx"])
-        mapping_file = gr.File(label="Mapping File", file_types=[".xlsx"])
         output_template = gr.File(label="Output Template", file_types=[".xlsx"])
 
     output_file = gr.File(label="Generated Excel File")
     generate_button = gr.Button("Generate Excel File")
 
-    generate_button.click(generate_excel, inputs=[input_workbook, mapping_file, output_template], outputs=[output_file])
+    generate_button.click(generate_excel, inputs=[input_workbook, output_template], outputs=[output_file])
 
 # Launch the app
 app.launch()