From b05afb1cc8259c03a210c775f071d9522484d848 Mon Sep 17 00:00:00 2001
From: ramforth <ramforth@noreply.localhost>
Date: Sat, 10 Jan 2026 16:52:17 +0100
Subject: [PATCH] Fix exit errors and add UI image placeholder to README

---
 README.md | 5 +++++
 ui/app.py | 8 ++++++++
 2 files changed, 13 insertions(+)

diff --git a/README.md b/README.md
index 26477d5..a3edfbd 100644
--- a/README.md
+++ b/README.md
@@ -8,6 +8,11 @@ A simple, motivating training diary focused on Calisthenics and basic health met
 - **Progress Tracking:** Visual graphs for weight and waist trends.
 - **Motivation:** Streak tracking to keep you consistent.
 
+## Interface
+
+![Exercise Diary Interface](assets/screenshot.png)
+*(Note: Replace `assets/screenshot.png` with a real screenshot of the app)*
+
 ## Installation
 
 1.  **Clone the repository:**
diff --git a/ui/app.py b/ui/app.py
index 7649216..addcc9d 100644
--- a/ui/app.py
+++ b/ui/app.py
@@ -42,6 +42,14 @@ class ExerciseDiaryApp(ctk.CTk):
 
         # Default View
         self.show_dashboard()
+        
+        # Handle closing
+        self.protocol("WM_DELETE_WINDOW", self.on_close)
+
+    def on_close(self):
+        self.withdraw()
+        self.quit()
+        self.destroy()
 
     def clear_main_frame(self):
         for widget in self.main_frame.winfo_children():