Add AI radio assistant UI

Author: cosven

examples/macos_native_titlebar.py

@@ -0,0 +1,99 @@
+"""macOS PyQt6: native traffic-light buttons, hidden titlebar.
+
+Run: uv run --extra qt python examples/macos_native_titlebar.py
+"""
+
+import ctypes
+import sys
+from ctypes import c_bool, c_char_p, c_long, c_ulong, c_void_p
+
+from PyQt6.QtCore import QTimer, Qt
+from PyQt6.QtWidgets import QApplication, QHBoxLayout, QLabel, QPushButton, QVBoxLayout, QWidget
+
+
+objc = ctypes.cdll.LoadLibrary("/usr/lib/libobjc.A.dylib")
+objc.sel_registerName.restype = c_void_p
+objc.sel_registerName.argtypes = [c_char_p]
+
+
+def sel(name):
+    return objc.sel_registerName(name.encode())
+
+
+def msg(receiver, selector, restype=c_void_p, argtypes=(), *args):
+    send = objc.objc_msgSend
+    send.restype = restype
+    send.argtypes = [c_void_p, c_void_p, *argtypes]
+    return send(receiver, sel(selector), *args)
+
+
+def expand_client_area(widget):
+    for name in ("ExpandedClientAreaHint", "NoTitleBarBackgroundHint"):
+        flag = getattr(Qt.WindowType, name, None)
+        if flag is not None:
+            widget.setWindowFlag(flag, True)
+    attr = getattr(Qt.WidgetAttribute, "WA_ContentsMarginsRespectsSafeArea", None)
+    if attr is not None:
+        widget.setAttribute(attr, False)
+
+
+def make_titlebar_transparent(widget):
+    ns_window = msg(c_void_p(int(widget.winId())), "window")
+    style = msg(ns_window, "styleMask", c_ulong)
+    style |= (1 << 0) | (1 << 1) | (1 << 2) | (1 << 3) | (1 << 15)
+    msg(ns_window, "setStyleMask:", None, (c_ulong,), style)
+    msg(ns_window, "setTitleVisibility:", None, (c_long,), 1)
+    msg(ns_window, "setTitlebarAppearsTransparent:", None, (c_bool,), True)
+    msg(ns_window, "setMovableByWindowBackground:", None, (c_bool,), True)
+
+
+class Toolbar(QWidget):
+    def __init__(self):
+        super().__init__()
+        self.setFixedHeight(38)
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(86, 0, 12, 0)
+        layout.setSpacing(8)
+        layout.addWidget(QLabel("Toolbar at y=0"), 0, Qt.AlignmentFlag.AlignTop)
+        layout.addStretch()
+        for text in ("Back", "Forward", "Search"):
+            button = QPushButton(text)
+            button.setFixedHeight(28)
+            layout.addWidget(button, 0, Qt.AlignmentFlag.AlignTop)
+
+    def mousePressEvent(self, event):
+        window = self.window().windowHandle()
+        if event.button() == Qt.MouseButton.LeftButton and window is not None:
+            if window.startSystemMove():
+                event.accept()
+                return
+        super().mousePressEvent(event)
+
+
+def main():
+    if sys.platform != "darwin":
+        print("This demo is macOS-only.")
+        return 1
+
+    app = QApplication(sys.argv)
+    window = QWidget()
+    expand_client_area(window)
+    window.setWindowTitle("Hidden titlebar")
+    window.resize(720, 420)
+
+    layout = QVBoxLayout(window)
+    layout.setContentsMargins(0, 0, 0, 0)
+    layout.setSpacing(0)
+    layout.addWidget(Toolbar())
+
+    body = QLabel("Native buttons are kept; the titlebar is not visible.")
+    body.setAlignment(Qt.AlignmentFlag.AlignCenter)
+    layout.addWidget(body, 1)
+
+    window.show()
+    QTimer.singleShot(0, lambda: make_titlebar_transparent(window))
+    return app.exec()
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

feeluown/ai/__init__.py

@@ -1,4 +1,5 @@
 # flake8: noqa
 
-from .copilot import AISongModel, Copilot, AISongMatcher
+from .copilot import SongSuggestion, Copilot, SongSuggestionMatcher
 from .ai import AI
+from .radio import AIRadioSession

feeluown/ai/ai.py

@@ -1,5 +1,11 @@
+from typing import TYPE_CHECKING, Iterable, Optional
+
 from feeluown.app import App
 from feeluown.ai.copilot import Copilot
+from feeluown.ai.radio import AIRadioSession
+
+if TYPE_CHECKING:
+    from feeluown.library import BriefSongModel
 
 
 # FIXME: Other packages should only import things from feeluown.ai
@@ -10,6 +16,36 @@ class AI:
     def __init__(self, app: App):
         self._app = app
         self._copilot = Copilot(self._app)
+        self.radio = None
 
     def get_copilot(self):
         return self._copilot
+
+    def get_active_radio(self) -> Optional["AIRadioSession"]:
+        if self.radio is not None and self.radio.is_active:
+            return self.radio
+        return None
+
+    def activate_radio(
+        self,
+        initial_songs: Optional[Iterable["BriefSongModel"]] = None,
+        reset=True,
+    ) -> AIRadioSession:
+        """Activate AI Radio and return the active session."""
+        active_radio = self.get_active_radio()
+        if active_radio is not None:
+            return active_radio
+
+        radio = AIRadioSession(self._app, initial_songs=initial_songs)
+        radio.activate(reset=reset)
+        return radio
+
+    def deactivate_radio(self) -> bool:
+        """Deactivate AI Radio. Return True when a session was active."""
+        radio = self.radio
+        if radio is None:
+            return False
+
+        was_active = radio.is_active
+        radio.deactivate()
+        return was_active