Update app.py

app.py

@@ -1,13 +1,105 @@
 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:
-        # Try reading the Excel file with only relevant columns to reduce overhead
-        df = pd.read_excel(file, usecols=['Account', 'Balance'], nrows=1000)
-        print("File read successfully.")
+        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('')
+        
+        # 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)
+
+        # 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['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['Category'] == 'Finance Costs']['Balance'].sum()
+        profit_before_tax = operating_profit - finance_costs
+        
+        tax_expense = df[df['Category'] == 'Tax Expense']['Balance'].sum()
+        profit_for_the_year = profit_before_tax - tax_expense
+        
+        # Generate the income statement with formatting
+        income_statement = f"""
+        INCOME STATEMENT (in 000s)
+
+        Revenue:                        {revenue:,.2f}
+        Cost of Sales:                 ({cost_of_sales:,.2f})
+        ------------------------------------------------
+        Gross Profit:                   {gross_profit:,.2f}
+
+        Other Income:                    {other_income:,.2f}
+        Operating Expenses:            ({operating_expenses:,.2f})
+        ------------------------------------------------
+        Operating Profit:                {operating_profit:,.2f}
+
+        Finance Costs:                 ({finance_costs:,.2f})
+        ------------------------------------------------
+        Profit Before Tax:               {profit_before_tax:,.2f}
+
+        Tax Expense:                   ({tax_expense:,.2f})
+        ------------------------------------------------
+        Profit for the Year:             {profit_for_the_year:,.2f}
+        """
+
+        # Create Balance Sheet - Filter for Asset, Liability, or Equity Accounts
+        balance_sheet = df[df['Category'].isin(['Assets', 'Liabilities', 'Equity'])].copy()
+        balance_sheet_summary = balance_sheet.groupby('Category').agg({'Balance': 'sum'}).reset_index()
+
+        return income_statement, balance_sheet_summary.to_string(index=False)
+
     except Exception as e:
-        return f"Error reading the file: {e}", None
+        return f"Error processing the file: {e}", None
+
+# 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.")
+
+    file_input = gr.File(label="Upload Trial Balance Excel 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)
+        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])
 
-    # Continue with the previous logic, now optimized
-    # (Rest of the code remains the same, including the mapping logic)
+# Launch the Gradio app
+if __name__ == "__main__":
+    demo.launch(share=True)