Update app.py

app.py

@@ -1,6 +1,30 @@
 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'
+
 def generate_financial_statements(file):
     try:
         print("Starting file read...")
@@ -18,23 +42,26 @@ def generate_financial_statements(file):
     try:
         # Ensure that 'Account' column has no NaN and is treated as string
         df['Account'] = df['Account'].astype(str).fillna('')
-        
+
         # Convert values in the 'Balance' column to thousands (000)
         df['Balance'] = df['Balance'] / 1000
 
-        # Create Income Statement as per IAS
-        revenue = df[df['Account'].str.contains('Revenue|Sales', case=False, na=False)]['Balance'].sum()
-        cost_of_sales = df[df['Account'].str.contains('Cost of Sales|COGS', case=False, na=False)]['Balance'].sum()
+        # Apply account mapping to categorize accounts
+        df['Category'] = df['Account'].apply(map_account_category)
+
+        # Income Statement calculation based on categories
+        revenue = df[df['Category'] == 'Revenue']['Balance'].sum()
+        cost_of_sales = df[df['Category'] == 'Cost of Sales']['Balance'].sum()
         gross_profit = revenue - cost_of_sales
         
-        other_income = df[df['Account'].str.contains('Other Income', case=False, na=False)]['Balance'].sum()
-        operating_expenses = df[df['Account'].str.contains('Operating Expenses|Administrative|Selling', case=False, na=False)]['Balance'].sum()
+        other_income = df[df['Category'] == 'Other']['Balance'].sum()
+        operating_expenses = df[df['Category'] == 'Operating Expenses']['Balance'].sum()
         operating_profit = gross_profit + other_income - operating_expenses
         
-        finance_costs = df[df['Account'].str.contains('Finance Costs|Interest Expense', case=False, na=False)]['Balance'].sum()
+        finance_costs = df[df['Category'] == 'Finance Costs']['Balance'].sum()
         profit_before_tax = operating_profit - finance_costs
         
-        tax_expense = df[df['Account'].str.contains('Tax Expense|Income Tax', case=False, na=False)]['Balance'].sum()
+        tax_expense = df[df['Category'] == 'Tax Expense']['Balance'].sum()
         profit_for_the_year = profit_before_tax - tax_expense
         
         # Generate the income statement with formatting
@@ -63,8 +90,8 @@ def generate_financial_statements(file):
         print("Income Statement created.")
 
         # Create Balance Sheet - Filter for Asset, Liability, or Equity Accounts
-        balance_sheet = df[df['Account'].str.contains('Asset|Liability|Equity', na=False, case=False)].copy()
-        balance_sheet_summary = balance_sheet.groupby('Account').agg({'Balance': 'sum'}).reset_index()
+        balance_sheet = df[df['Category'].isin(['Assets', 'Liabilities', 'Equity'])].copy()
+        balance_sheet_summary = balance_sheet.groupby('Category').agg({'Balance': 'sum'}).reset_index()
         print("Balance Sheet created.")
 
         return income_statement, balance_sheet_summary.to_string(index=False)