From 0816ebd26b86555c2b1ec7c5ca31d51ea28a2d83 Mon Sep 17 00:00:00 2001
From: Andreas Tsouchlos <an.tsouchlos@gmail.com>
Date: Thu, 4 Jan 2024 03:19:30 +0100
Subject: [PATCH] Add category column if it does not exist

---
 banking_breakdown/ui.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/banking_breakdown/ui.py b/banking_breakdown/ui.py
index 2072793..ba0ca24 100644
--- a/banking_breakdown/ui.py
+++ b/banking_breakdown/ui.py
@@ -121,6 +121,9 @@ class MainWindow(QMainWindow):
         self._warnings.append((layout, warningIcon, label))
 
     def set_statement_data(self, df: pd.DataFrame):
+        if 'category' not in df.columns:
+            df['category'] = [' '] * len(df.index)
+
         model = PandasModel(df)
         proxyModel = QSortFilterProxyModel(self)
         proxyModel.setSourceModel(model)