Update app.py

app.py

@@ -2,6 +2,10 @@ import pandas as pd
 from openpyxl import load_workbook
 import gradio as gr
 import os
+import warnings
+
+# Suppress openpyxl warnings
+warnings.filterwarnings("ignore", category=UserWarning, module="openpyxl")
 
 # Load the constant mapping file (embedded in the app)
 def load_mapping():
@@ -30,7 +34,7 @@ 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] * len(input_workbook.sheet_names)  # Placeholder for rows
+            output_data[output_column] = [fixed_value] * 10  # Placeholder for rows
 
         # Handle column mapping from input workbook
         elif "." in input_rule:
@@ -40,6 +44,11 @@ def transform_data(input_path, mapping_df):
                 if column_name in sheet_data.columns:
                     output_data[output_column] = sheet_data[column_name].tolist()
 
+    # Fill missing columns with empty lists
+    for key in output_data:
+        if not output_data[key]:
+            output_data[key] = ["" for _ in range(10)]
+
     return pd.DataFrame(output_data)
 
 # Main processing function
@@ -53,6 +62,9 @@ def process_files(input_workbook):
 
         # Load the output template (embedded in the app)
         output_template_path = "output_template.xlsx"  # Replace with the actual template file path
+        if not os.path.exists(output_template_path):
+            return "Output template file is missing."
+
         output_workbook = load_workbook(output_template_path)
         output_sheet = output_workbook["363040"]
 
@@ -68,7 +80,7 @@ def process_files(input_workbook):
         return output_file_path
 
     except Exception as e:
-        return f"An error occurred: {e}"
+        return f"An error occurred during file generation: {e}"
 
 # Define the Gradio interface
 def generate_excel(input_workbook):