From 9d8aa88bd02b6770bd52eda0b49abfbd6e92aa87 Mon Sep 17 00:00:00 2001
From: Johannes Rest <johannes@jr-it-services.de>
Date: Tue, 24 Feb 2026 11:17:16 +0100
Subject: [PATCH] Added button for app close.

---
 sql_ai_gui.py | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/sql_ai_gui.py b/sql_ai_gui.py
index 13d9e73..872911c 100644
--- a/sql_ai_gui.py
+++ b/sql_ai_gui.py
@@ -274,6 +274,9 @@ class MainWindow(QMainWindow):
         self.btn_copy_sql = QPushButton("Copy SQL only")
         self.btn_copy_sql.clicked.connect(self.copy_sql_only)
         right_btn_row.addWidget(self.btn_copy_sql)
+        self.btn_close = QPushButton("Close")
+        self.btn_close.clicked.connect(self.on_close)
+        right_btn_row.addWidget(self.btn_close)
 
         self.btn_model_pull = QPushButton("Modell aktualisieren (pull)")
         self.btn_model_pull.clicked.connect(self.on_pull_model)
@@ -293,7 +296,7 @@ class MainWindow(QMainWindow):
 
     # -------------- UI helpers --------------
     def ui_busy(self, busy: bool) -> None:
-        for w in [self.btn_send, self.btn_model_pull, self.btn_refresh_models, self.btn_copy_sql]:
+        for w in [self.btn_send, self.btn_model_pull, self.btn_refresh_models, self.btn_copy_sql, self.btn_close]:
             w.setEnabled(not busy)
         self.prompt.setEnabled(not busy)
         self.base_url.setEnabled(not busy)
@@ -434,6 +437,19 @@ class MainWindow(QMainWindow):
         self.msg_error("Model pull fehlgeschlagen", err)
         self.msg_error("Ollama Runtime Update fehlgeschlagen", err)
 
+    def on_close(self) -> None:
+        reply = QMessageBox.question(
+            self,
+            "Beenden",
+            "Wirklich beenden?",
+            QMessageBox.Yes | QMessageBox.No,
+            QMessageBox.No,
+        )
+        if reply == QMessageBox.Yes:
+            self.close()
+
+
+
 
 def main() -> int:
     # For Wayland/Hyprland you can force: