Update app.py

app.py

@@ -1,11 +1,10 @@
 import pandas as pd
-from openpyxl import load_workbook
-from openpyxl.utils import get_column_letter
+from openpyxl import Workbook
 import gradio as gr
 import os
 import warnings
 
-# Suppress openpyxl warnings
+# Suppress warnings
 warnings.filterwarnings("ignore", category=UserWarning, module="openpyxl")
 
 # Hardcoded mapping logic (from reference file)
@@ -57,15 +56,17 @@ def process_files(input_workbook):
         # Transform the data
         transformed_data = transform_data(input_workbook)
 
-        # Load the output template (embedded in the app)
-        output_template_path = "Generated_Output.xlsx"
-        if not os.path.exists(output_template_path):
-            return None, "Output template file is missing."
-
-        output_workbook = load_workbook(output_template_path)
+        # Create a new workbook for the output
+        output_workbook = Workbook()
         output_sheet = output_workbook.active
+        output_sheet.title = "Generated Output"
+
+        # Write headers
+        headers = transformed_data.columns.tolist()
+        for col_idx, header in enumerate(headers, start=1):
+            output_sheet.cell(row=1, column=col_idx, value=header)
 
-        # Write transformed data to the output sheet
+        # Write data rows
         for row_idx, row_data in enumerate(transformed_data.itertuples(index=False), start=2):
             for col_idx, value in enumerate(row_data, start=1):
                 output_sheet.cell(row=row_idx, column=col_idx, value=value)