Update app.py

app.py

@@ -49,6 +49,10 @@ def transform_data(input_path, mapping_df):
 # Main processing function
 def process_files(input_workbook):
     try:
+        # Validate input file
+        if not os.path.exists(input_workbook):
+            return None, "Input workbook file does not exist."
+
         # Load the constant mapping data
         mapping_df = load_mapping()
 
@@ -58,7 +62,7 @@ def process_files(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 "Output template file is missing."
+            return None, "Output template file is missing."
 
         output_workbook = load_workbook(output_template_path)
         output_sheet = output_workbook.active
@@ -72,18 +76,17 @@ def process_files(input_workbook):
         output_file_path = "Generated_Output_Final.xlsx"
         output_workbook.save(output_file_path)
 
-        return output_file_path
+        return output_file_path, None
 
     except Exception as e:
-        return f"An error occurred during file generation: {e}"
+        return None, f"An error occurred during file generation: {e}"
 
 # Define the Gradio interface
 def generate_excel(input_workbook):
-    result = process_files(input_workbook.name)
-    if os.path.exists(result):
-        return result
-    else:
-        return "An error occurred during file generation."
+    file_path, error = process_files(input_workbook.name)
+    if error:
+        return None, error
+    return file_path, "File generated successfully."
 
 with gr.Blocks() as app:
     gr.Markdown("# Excel Sheet Generator")
@@ -93,9 +96,11 @@ with gr.Blocks() as app:
         input_workbook = gr.File(label="Input Workbook", file_types=[".xlsx"])
 
     output_file = gr.File(label="Generated Excel File")
+    status_message = gr.Textbox(label="Status", interactive=False)
+
     generate_button = gr.Button("Generate Excel File")
 
-    generate_button.click(generate_excel, inputs=[input_workbook], outputs=[output_file])
+    generate_button.click(generate_excel, inputs=[input_workbook], outputs=[output_file, status_message])
 
 # Launch the app
 app.launch()