Update app.py

app.py

@@ -37,7 +37,7 @@ def transform_data(input_path, mapping_df):
     return pd.DataFrame(output_data)
 
 # Main processing function
-def process_files(input_workbook, output_template):
+def process_files(input_workbook):
     try:
         # Load the constant mapping data
         mapping_df = load_mapping()
@@ -45,8 +45,9 @@ def process_files(input_workbook, output_template):
         # Transform the data
         transformed_data = transform_data(input_workbook, mapping_df)
 
-        # Load the output template
-        output_workbook = load_workbook(output_template)
+        # Load the output template (embedded in the app)
+        output_template_path = "output_template.xlsx"  # Replace with the actual template file path
+        output_workbook = load_workbook(output_template_path)
         output_sheet = output_workbook["363040"]
 
         # Write transformed data to the output sheet
@@ -64,8 +65,8 @@ def process_files(input_workbook, output_template):
         return f"An error occurred: {e}"
 
 # Define the Gradio interface
-def generate_excel(input_workbook, output_template):
-    result = process_files(input_workbook.name, output_template.name)
+def generate_excel(input_workbook):
+    result = process_files(input_workbook.name)
     if os.path.exists(result):
         return result
     else:
@@ -73,16 +74,15 @@ def generate_excel(input_workbook, output_template):
 
 with gr.Blocks() as app:
     gr.Markdown("# Excel Sheet Generator")
-    gr.Markdown("Upload the input workbook and output template to generate the final Excel file.")
+    gr.Markdown("Upload the input workbook to generate the final Excel file.")
 
     with gr.Row():
         input_workbook = gr.File(label="Input Workbook", 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, output_template], outputs=[output_file])
+    generate_button.click(generate_excel, inputs=[input_workbook], outputs=[output_file])
 
 # Launch the app
 app.launch()