Update app.py

app.py

@@ -1,18 +1,25 @@
 import pandas as pd
 import gradio as gr
 
-# Function to read trial balance and prepare financial statements
 def generate_financial_statements(file):
+    print("Starting file read...")
     # Read the Excel file
-    df = pd.read_excel(file)
-
-    # Assuming the trial balance has columns: 'Account', 'Debit', 'Credit'
-    if 'Account' not in df.columns or 'Debit' not in df.columns or 'Credit' not in df.columns:
+    try:
+        df = pd.read_excel(file)
+        print("File read successfully.")
+    except Exception as e:
+        return f"Error reading the file: {e}"
+
+    # Check for required columns
+    required_columns = ['Account', 'Debit', 'Credit']
+    if not all(col in df.columns for col in required_columns):
         return "Invalid file format. Please ensure the columns are 'Account', 'Debit', and 'Credit'."
 
+    print("Calculating totals...")
     # Calculate totals
     total_debit = df['Debit'].sum()
     total_credit = df['Credit'].sum()
+    print(f"Total Debit: {total_debit}, Total Credit: {total_credit}")
 
     # Create Profit & Loss Statement
     pnl = df[df['Debit'] > 0].copy()
@@ -20,19 +27,23 @@ def generate_financial_statements(file):
     pnl.loc[pnl['Credit'] > 0, 'Type'] = 'Income'
     
     pnl_summary = pnl.groupby('Type')['Debit', 'Credit'].sum().reset_index()
+    print("Profit & Loss statement created.")
 
     # Create Balance Sheet
     balance_sheet = df.groupby(df['Account'].str.contains('Asset|Liability|Equity')).agg({'Debit': 'sum', 'Credit': 'sum'}).reset_index()
     balance_sheet.columns = ['Type', 'Debit', 'Credit']
+    print("Balance Sheet created.")
 
-    return pnl_summary, balance_sheet
+    return pnl_summary.to_string(), balance_sheet.to_string()
 
 # Gradio interface
 def gradio_interface(file):
-    pnl_summary, balance_sheet = generate_financial_statements(file)
-    return pnl_summary.to_string(), balance_sheet.to_string()
+    print("Processing file...")
+    result = generate_financial_statements(file)
+    print("Processing complete.")
+    return result
 
-# Launch the Gradio app
+# Set up Gradio interface
 iface = gr.Interface(
     fn=gradio_interface,
     inputs=gr.inputs.File(label="Upload Trial Balance Excel File"),
@@ -44,4 +55,6 @@ iface = gr.Interface(
     description="Upload a trial balance Excel file to generate Profit & Loss account and Balance Sheet."
 )
 
-iface.launch()
+# Launch the Gradio app
+if __name__ == "__main__":
+    iface.launch()