Update app.py

app.py

@@ -1,41 +1,23 @@
 import pandas as pd
 import gradio as gr
 
-# Sample account mapping for the trial balance accounts
-account_mapping = {
-    'Revenue': 'Revenue',
-    'Sales': 'Revenue',
-    'Cost of Sales': 'Cost of Sales',
-    'COGS': 'Cost of Sales',
-    'Operating Expenses': 'Operating Expenses',
-    'Selling': 'Operating Expenses',
-    'Administrative': 'Operating Expenses',
-    'Finance Costs': 'Finance Costs',
-    'Interest Expense': 'Finance Costs',
-    'Tax Expense': 'Tax Expense',
-    'Income Tax': 'Tax Expense',
-    'Assets': 'Assets',
-    'Liability': 'Liabilities',
-    'Equity': 'Equity'
-}
-
-def map_account_category(account_name):
-    for key, value in account_mapping.items():
-        if key.lower() in account_name.lower():
-            return value
-    return 'Other'
+# Load account mapping from a mapping file
+def load_mapping(mapping_file):
+    try:
+        mapping_df = pd.read_excel(mapping_file)
+        return dict(zip(mapping_df['Account Name'].str.lower(), mapping_df['Category']))
+    except Exception as e:
+        raise Exception(f"Error loading mapping file: {e}")
 
-def generate_financial_statements(file):
+# Generate financial statements based on the trial balance and mapping
+def generate_financial_statements(file, mapping):
     try:
         df = pd.read_excel(file, usecols=['Account', 'Balance'])
-        # Ensure that 'Account' column has no NaN and is treated as string
         df['Account'] = df['Account'].astype(str).fillna('')
+        df['Balance'] = df['Balance'] / 1000  # Convert values to thousands (000)
         
-        # Convert values in the 'Balance' column to thousands (000)
-        df['Balance'] = df['Balance'] / 1000
-
         # Apply account mapping to categorize accounts
-        df['Category'] = df['Account'].apply(map_account_category)
+        df['Category'] = df['Account'].str.lower().map(mapping).fillna('Other')
 
         # Income Statement calculation based on categories
         revenue = df[df['Category'] == 'Revenue']['Balance'].sum()
@@ -75,7 +57,7 @@ def generate_financial_statements(file):
         Profit for the Year:             {profit_for_the_year:,.2f}
         """
 
-        # Create Balance Sheet - Filter for Asset, Liability, or Equity Accounts
+        # Create Balance Sheet
         balance_sheet = df[df['Category'].isin(['Assets', 'Liabilities', 'Equity'])].copy()
         balance_sheet_summary = balance_sheet.groupby('Category').agg({'Balance': 'sum'}).reset_index()
 
@@ -87,18 +69,20 @@ def generate_financial_statements(file):
 # Gradio Blocks interface
 with gr.Blocks() as demo:
     gr.Markdown("# Financial Statement Generator")
-    gr.Markdown("Upload a trial balance Excel file to generate an Income Statement and a Balance Sheet.")
+    gr.Markdown("Upload a trial balance Excel file and a mapping file to generate an Income Statement and a Balance Sheet.")
 
     file_input = gr.File(label="Upload Trial Balance Excel File")
+    mapping_input = gr.File(label="Upload Mapping File")
     income_output = gr.Textbox(label="Income Statement", lines=20)
     balance_output = gr.Textbox(label="Balance Sheet", lines=10)
 
-    def process_file(file):
-        income_statement, balance_sheet = generate_financial_statements(file)
+    def process_file(trial_balance_file, mapping_file):
+        mapping = load_mapping(mapping_file)
+        income_statement, balance_sheet = generate_financial_statements(trial_balance_file, mapping)
         return income_statement, balance_sheet
 
-    # Trigger processing when file is uploaded
-    file_input.change(fn=process_file, inputs=file_input, outputs=[income_output, balance_output])
+    # Trigger processing when files are uploaded
+    file_input.change(fn=process_file, inputs=[file_input, mapping_input], outputs=[income_output, balance_output])
 
 # Launch the Gradio app
 if __name__ == "__main__":