Update app.py

app.py

@@ -214,21 +214,27 @@ def generate_t_account(account_name):
     cursor.execute("SELECT account_id FROM chart_of_accounts WHERE account_name = ?", (account_name,))
     account_id = cursor.fetchone()
     if not account_id:
+        logging.error(f"Account {account_name} not found.")
         return "Account not found."
     
     account_id = account_id[0]
-    cursor.execute("""
-    SELECT amount, description, 'Debit' as type FROM journal_entries WHERE debit_account_id = ?
-    UNION
-    SELECT amount, description, 'Credit' as type FROM journal_entries WHERE credit_account_id = ?
-    ORDER BY date
-    """, (account_id, account_id))
-    entries = cursor.fetchall()
+    try:
+        cursor.execute("""
+        SELECT date, amount, description, 'Debit' as type FROM journal_entries WHERE debit_account_id = ?
+        UNION
+        SELECT date, amount, description, 'Credit' as type FROM journal_entries WHERE credit_account_id = ?
+        ORDER BY date
+        """, (account_id, account_id))
+        entries = cursor.fetchall()
+        logging.info(f"Retrieved {len(entries)} entries for T-account: {account_name}")
+    except sqlite3.Error as e:
+        logging.error(f"SQL error in generate_t_account: {e}")
+        return "Error retrieving T-account data."
     
     t_account = f"T-Account for {account_name}\n{'='*50}\n{'Debit':<20} | {'Credit':<20} | Description\n{'-'*50}\n"
     debit_total = 0
     credit_total = 0
-    for amount, desc, entry_type in entries:
+    for date, amount, desc, entry_type in entries:
         if entry_type == "Debit":
             t_account += f"${amount:<19} | {'':<20} | {desc}\n"
             debit_total += amount