setiastrosuitepro 1.6.0__py3-none-any.whl → 1.6.4.post1__py3-none-any.whl

setiastro/saspro/whitebalance.py

@@ -191,10 +191,18 @@ def apply_white_balance_to_doc(doc, preset: Optional[Dict] = None):
 class WhiteBalanceDialog(QDialog):
     def __init__(self, parent, doc, icon: QIcon | None = None):
         super().__init__(parent)
+        self._main = parent
         self.doc = doc
+
+        # Connect to active document change signal
+        if hasattr(self._main, "currentDocumentChanged"):
+            self._main.currentDocumentChanged.connect(self._on_active_doc_changed)
         if icon:
             self.setWindowIcon(icon)
-        self.setWindowTitle("White Balance")
+        self.setWindowTitle(self.tr("White Balance"))
+        self.setWindowFlag(Qt.WindowType.Window, True)
+        self.setWindowModality(Qt.WindowModality.NonModal)
+        self.setModal(False)
         self.resize(900, 600)
 
         self._build_ui()
@@ -212,29 +220,29 @@ class WhiteBalanceDialog(QDialog):
 
         # Type selector
         row = QHBoxLayout()
-        row.addWidget(QLabel("White Balance Type:"))
+        row.addWidget(QLabel(self.tr("White Balance Type:")))
         self.type_combo = QComboBox()
-        self.type_combo.addItems(["Star-Based", "Manual", "Auto"])
+        self.type_combo.addItems([self.tr("Star-Based"), self.tr("Manual"), self.tr("Auto")])
         row.addWidget(self.type_combo); row.addStretch()
         self.main_layout.addLayout(row)
 
         # Manual group
-        self.manual_group = QGroupBox("Manual Gains")
+        self.manual_group = QGroupBox(self.tr("Manual Gains"))
         g = QGridLayout(self.manual_group)
         self.r_spin = QDoubleSpinBox(); self._cfg_gain(self.r_spin, 1.0)
         self.g_spin = QDoubleSpinBox(); self._cfg_gain(self.g_spin, 1.0)
         self.b_spin = QDoubleSpinBox(); self._cfg_gain(self.b_spin, 1.0)
-        g.addWidget(QLabel("Red gain:"),   0, 0); g.addWidget(self.r_spin, 0, 1)
-        g.addWidget(QLabel("Green gain:"), 1, 0); g.addWidget(self.g_spin, 1, 1)
-        g.addWidget(QLabel("Blue gain:"),  2, 0); g.addWidget(self.b_spin, 2, 1)
+        g.addWidget(QLabel(self.tr("Red gain:")),   0, 0); g.addWidget(self.r_spin, 0, 1)
+        g.addWidget(QLabel(self.tr("Green gain:")), 1, 0); g.addWidget(self.g_spin, 1, 1)
+        g.addWidget(QLabel(self.tr("Blue gain:")),  2, 0); g.addWidget(self.b_spin, 2, 1)
         self.main_layout.addWidget(self.manual_group)
 
         # Star-based controls + preview
-        self.star_group = QGroupBox("Star-Based Settings")
+        self.star_group = QGroupBox(self.tr("Star-Based Settings"))
         sg = QVBoxLayout(self.star_group)
         # threshold slider
         thr_row = QHBoxLayout()
-        thr_row.addWidget(QLabel("Detection threshold (σ):"))
+        thr_row.addWidget(QLabel(self.tr("Detection threshold (σ):")))
         self.thr_slider = QSlider(Qt.Orientation.Horizontal)
         self.thr_slider.setMinimum(1); self.thr_slider.setMaximum(100)
         self.thr_slider.setValue(50); self.thr_slider.setTickInterval(10)
@@ -243,14 +251,14 @@ class WhiteBalanceDialog(QDialog):
         thr_row.addWidget(self.thr_slider); thr_row.addWidget(self.thr_label)
         sg.addLayout(thr_row)
 
-        self.chk_reuse = QCheckBox("Reuse cached star detections"); self.chk_reuse.setChecked(True)
+        self.chk_reuse = QCheckBox(self.tr("Reuse cached star detections")); self.chk_reuse.setChecked(True)
         sg.addWidget(self.chk_reuse)
 
-        self.chk_autostretch_overlay = QCheckBox("Autostretch overlay preview"); self.chk_autostretch_overlay.setChecked(True)
+        self.chk_autostretch_overlay = QCheckBox(self.tr("Autostretch overlay preview")); self.chk_autostretch_overlay.setChecked(True)
         sg.addWidget(self.chk_autostretch_overlay)
 
         # star count + image preview
-        self.star_count = QLabel("Detecting stars…")
+        self.star_count = QLabel(self.tr("Detecting stars..."))
         sg.addWidget(self.star_count)
 
         self.preview = QLabel(); self.preview.setMinimumSize(640, 360)
@@ -261,8 +269,8 @@ class WhiteBalanceDialog(QDialog):
         # Buttons
         btn_row = QHBoxLayout()
         btn_row.addStretch()
-        self.btn_apply = QPushButton("Apply")
-        self.btn_cancel = QPushButton("Cancel")
+        self.btn_apply = QPushButton(self.tr("Apply"))
+        self.btn_cancel = QPushButton(self.tr("Cancel"))
         btn_row.addWidget(self.btn_apply)
         btn_row.addWidget(self.btn_cancel)
         self.main_layout.addLayout(btn_row)
@@ -291,6 +299,18 @@ class WhiteBalanceDialog(QDialog):
         self.manual_group.setVisible(t == "Manual")
         self.star_group.setVisible(t == "Star-Based")
 
+    # ---- active document change ------------------------------------
+    def _on_active_doc_changed(self, doc):
+        """Called when user clicks a different image window."""
+        if doc is None or getattr(doc, "image", None) is None:
+            return
+        self.doc = doc
+        # Force fresh star detection by temporarily disabling cache
+        old_reuse = self.chk_reuse.isChecked()
+        self.chk_reuse.setChecked(False)
+        self._update_star_preview()
+        self.chk_reuse.setChecked(old_reuse)
+
     # ---- preview --------------------------------------------------------
     def _update_star_preview(self):
         if self.type_combo.currentText() != "Star-Based":
@@ -305,7 +325,7 @@ class WhiteBalanceDialog(QDialog):
                 img, threshold=thr, autostretch=auto,
                 reuse_cached_sources=reuse, return_star_colors=False
             )
-            self.star_count.setText(f"Detected {count} stars.")
+            self.star_count.setText(self.tr("Detected {0} stars.").format(count))
             # to pixmap
             h, w, _ = overlay.shape
             qimg = QImage((overlay * 255).astype(np.uint8).data, w, h, 3 * w, QImage.Format.Format_RGB888)
@@ -314,7 +334,7 @@ class WhiteBalanceDialog(QDialog):
                                                 Qt.TransformationMode.SmoothTransformation)
             self.preview.setPixmap(pm)
         except Exception as e:
-            self.star_count.setText("Detection failed.")
+            self.star_count.setText(self.tr("Detection failed."))
             self.preview.clear()
 
     # ---- apply ----------------------------------------------------------
@@ -373,7 +393,8 @@ class WhiteBalanceDialog(QDialog):
 
                 # Use the headless helper so doc metadata is consistent
                 apply_white_balance_to_doc(self.doc, preset)
-                self.accept()
+                # Dialog stays open - refresh document for next operation
+                self._refresh_document_from_active()
 
             elif mode == "Auto":
                 preset = {"mode": "auto"}
@@ -382,7 +403,8 @@ class WhiteBalanceDialog(QDialog):
                 _record_preset_for_replay(preset)
 
                 apply_white_balance_to_doc(self.doc, preset)
-                self.accept()
+                # Dialog stays open - refresh document for next operation
+                self._refresh_document_from_active()
 
             else:  # --- Star-Based: compute here so we can plot like SASv2 ---
                 thr   = float(self.thr_slider.value())
@@ -446,11 +468,25 @@ class WhiteBalanceDialog(QDialog):
 
                 QMessageBox.information(
                     self,
-                    "White Balance",
-                    f"Star-Based WB applied.\nDetected {int(star_count)} stars.",
+                    self.tr("White Balance"),
+                    self.tr("Star-Based WB applied.\nDetected {0} stars.").format(int(star_count)),
                 )
-                self.accept()
+                # Dialog stays open - refresh document for next operation
+                self._refresh_document_from_active()
 
         except Exception as e:
-            QMessageBox.critical(self, "White Balance", f"Failed to apply White Balance:\n{e}")
+            QMessageBox.critical(self, self.tr("White Balance"), self.tr("Failed to apply White Balance:\n{0}").format(e))
 
+    def _refresh_document_from_active(self):
+        """
+        Refresh the dialog's document reference to the currently active document.
+        This allows reusing the same dialog on different images.
+        """
+        try:
+            main = self.parent()
+            if main and hasattr(main, "_active_doc"):
+                new_doc = main._active_doc()
+                if new_doc is not None and new_doc is not self.doc:
+                    self.doc = new_doc
+        except Exception:
+            pass

setiastro/saspro/widgets/common_utilities.py

@@ -21,7 +21,7 @@ import logging
 import atexit
 from typing import TYPE_CHECKING, Optional, List, Any
 
-from PyQt6.QtCore import QThread, pyqtSignal, QTimer, Qt
+from PyQt6.QtCore import QThread, pyqtSignal, QTimer, Qt, QCoreApplication
 from PyQt6.QtWidgets import QDialog, QVBoxLayout, QLabel, QMessageBox
 
 if TYPE_CHECKING:
@@ -65,7 +65,7 @@ class AboutDialog(QDialog):
 
     def __init__(self, parent: Optional[Any] = None, version: str = "", build_timestamp: str = ""):
         super().__init__(parent)
-        self.setWindowTitle("About Seti Astro Suite")
+        self.setWindowTitle(self.tr("About Seti Astro Suite"))
 
         # Get version info if not provided
         if not version:
@@ -84,17 +84,17 @@ class AboutDialog(QDialog):
         # Build about text with optional build timestamp
         about_lines = [
             f"<h2>Seti Astro's Suite Pro {version}</h2>",
-            "<p>Written by Franklin Marek</p>",
-            "<p>Collaborators: Fabio Tempera</p>",
-            "<p>Copyright © 2025 Seti Astro</p>",
+            f"<p>{self.tr('Written by Franklin Marek')}</p>",
+            f"<p>{self.tr('Collaborators: Fabio Tempera')}</p>",
+            f"<p>{self.tr('Copyright © 2025 Seti Astro')}</p>",
         ]
 
         if build_timestamp and build_timestamp != "Unknown":
-            about_lines.append(f"<p><b>Build:</b> {build_timestamp}</p>")
+            about_lines.append(f"<p><b>{self.tr('Build:')}</b> {build_timestamp}</p>")
 
         about_lines.extend([
-            "<p>Website: <a href='http://www.setiastro.com'>www.setiastro.com</a></p>",
-            "<p>Donations: <a href='https://www.setiastro.com/checkout/donate?donatePageId=65ae7e7bac20370d8c04c1ab'>Click here to donate</a></p>",
+            f"<p>{self.tr('Website:')} <a href='http://www.setiastro.com'>www.setiastro.com</a></p>",
+            f"<p>{self.tr('Donations:')} <a href='https://www.setiastro.com/checkout/donate?donatePageId=65ae7e7bac20370d8c04c1ab'>{self.tr('Click here to donate')}</a></p>",
         ])
 
         about_text = "".join(about_lines)
@@ -254,7 +254,8 @@ def install_crash_handlers(app: 'QApplication') -> None:
             m.setIcon(QMessageBox.Icon.Critical)
             m.setWindowTitle(title)
             m.setText(head)
-            m.setInformativeText("Details are available below and in saspro.log.")
+            from PyQt6.QtCore import QCoreApplication
+            m.setInformativeText(QCoreApplication.translate("CrashHandler", "Details are available below and in saspro.log."))
             if details:
                 m.setDetailedText(details)
             m.setStandardButtons(QMessageBox.StandardButton.Ok)
@@ -266,7 +267,7 @@ def install_crash_handlers(app: 'QApplication') -> None:
         tb = "".join(traceback.format_exception(exc_type, exc_value, exc_tb))
         logging.error("Uncaught exception:\n%s", tb)
         _show_dialog(
-            "Unhandled Exception",
+            QCoreApplication.translate("CrashHandler", "Unhandled Exception"),
             f"{exc_type.__name__}: {exc_value}",
             tb
         )
@@ -278,7 +279,7 @@ def install_crash_handlers(app: 'QApplication') -> None:
         tb = "".join(traceback.format_exception(args.exc_type, args.exc_value, args.exc_traceback))
         logging.error("Uncaught thread exception (%s):\n%s", args.thread.name, tb)
         _show_dialog(
-            "Unhandled Thread Exception",
+            QCoreApplication.translate("CrashHandler", "Unhandled Thread Exception"),
             f"{args.exc_type.__name__}: {args.exc_value}",
             tb
         )