Make input prompt drag resizable.

Author: FeepingCreature

ai_diffusion/ui/region.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 from enum import Enum
 from PyQt5.QtWidgets import QWidget, QLabel, QToolButton, QHBoxLayout, QVBoxLayout, QFrame, QMenu
-from PyQt5.QtGui import QGuiApplication, QMouseEvent, QResizeEvent, QPixmap, QImage, QPainter, QIcon
+from PyQt5.QtGui import QGuiApplication, QMouseEvent, QResizeEvent, QPixmap, QImage, QPainter, QIcon, QFontMetrics
 from PyQt5.QtCore import QObject, QEvent, Qt, QMetaObject, QSize, pyqtSignal
 
 from ..root import root
@@ -123,12 +123,13 @@ def __init__(self, root: RootRegion, parent: QWidget, header=PromptHeader.full):
 
         self._header = QWidget(self)
         self._header.setLayout(header_layout)
-
+        
         self.positive = TextPromptWidget(parent=self)
         self.positive.line_count = min(settings.prompt_line_count, self._max_lines)
         self.positive.install_event_filter(self)
 
-        self.negative = TextPromptWidget(line_count=1, is_negative=True, parent=self)
+        self.negative = TextPromptWidget(line_count=1, is_negative=True, parent=self, resize_handle=True)
+        self.negative.handle_dragged.connect(self._handle_dragging)
         self.negative.install_event_filter(self)
 
         self._no_region = QWidget(self)
@@ -398,6 +399,14 @@ def _layout_language_button(self):
             s = QSize(self.fontMetrics().width("EN"), self.fontMetrics().height())
             self._language_button.move(pos.x() - s.width() - 2, pos.y() - s.height() - 2)
             self._language_button.resize(s)
+        
+    def _handle_dragging(self, y_pos: int):
+        new_height = y_pos - self.negative.contentsRect().top() + self.positive.contentsRect().height()
+        fm = QFontMetrics(ensure(self.positive._multi.document()).defaultFont())
+        new_line_count = round((new_height - 3) / fm.lineSpacing()) - 1
+        if 1 <= new_line_count <= 10:
+            settings.prompt_line_count = new_line_count
+            self.positive.line_count = new_line_count
 
     def resizeEvent(self, a0):
         super().resizeEvent(a0)

ai_diffusion/ui/widget.py

@@ -440,18 +440,56 @@ def keyPressEvent(self, a0: QKeyEvent | None):
         super().keyPressEvent(a0)
 
 
+class ResizeHandle(QWidget):
+    """A small resize handle that appears at the bottom of the prompt widget."""
+
+    handle_dragged = pyqtSignal(int)
+
+    def __init__(self, parent: QWidget):
+        super().__init__(parent)
+        self.setCursor(Qt.CursorShape.SizeVerCursor)
+        self.setFixedSize(20, 8)
+        self._dragging = False
+        self.setVisible(True)
+        
+    def mousePressEvent(self, event: QMouseEvent | None) -> None:
+        if ensure(event).button() == Qt.MouseButton.LeftButton:
+            self._dragging = True
+
+    def mouseReleaseEvent(self, event: QMouseEvent | None) -> None:
+        self._dragging = False
+
+    def mouseMoveEvent(self, event: QMouseEvent | None) -> None:
+        if not self._dragging:
+            return
+        y_pos = self.mapToParent(ensure(event).pos()).y()
+        self.handle_dragged.emit(y_pos)
+            
+    def paintEvent(self, event: QPaintEvent | None) -> None:
+        if self.isVisible():
+            painter = QPainter(self)
+            painter.setPen(self.palette().color(QPalette.ColorRole.Mid))
+            painter.setBrush(painter.pen().color())
+            w, h = self.width(), self.height()
+            for i, x in enumerate(range(1, w, 3)):
+                y = 2*h//3 if i % 2 == 0 else h//3
+                painter.drawEllipse(x-1, y-1, 2, 2)
+
+
 class TextPromptWidget(QFrame):
     """Wraps a single or multi-line text widget, with ability to switch between them.
     Using QPlainTextEdit set to a single line doesn't work properly because it still
     scrolls to the next line when eg. selecting and then looks like it's empty."""
 
     activated = pyqtSignal()
     text_changed = pyqtSignal(str)
+    handle_dragged = pyqtSignal(int)
 
     _line_count = 2
     _is_negative = False
+    _resize_handle: ResizeHandle = None
 
-    def __init__(self, line_count=2, is_negative=False, parent=None):
+    def __init__(self, line_count=2, is_negative=False, parent=None, resize_handle=False):
         super().__init__(parent)
         self._line_count = line_count
         self._is_negative = is_negative
@@ -474,6 +512,10 @@ def __init__(self, line_count=2, is_negative=False, parent=None):
         self._layout.addWidget(self._multi)
         self._layout.addWidget(self._single)
 
+        if resize_handle:
+            self._resize_handle = ResizeHandle(self)
+            self._resize_handle.handle_dragged.connect(self.handle_dragged)
+
         palette: QPalette = self._multi.palette()
         self._base_color = palette.color(QPalette.ColorRole.Base)
         self.is_negative = self._is_negative
@@ -510,6 +552,15 @@ def line_count(self, value: int):
         if self._line_count > 1:
             self._multi.line_count = self._line_count
 
+    def resizeEvent(self, event):
+        super().resizeEvent(event)
+        if self._resize_handle:
+            rect = self.geometry()
+            self._resize_handle.move(
+                (rect.width() - self._resize_handle.width()) // 2,
+                rect.height() - self._resize_handle.height()
+            )
+
     @property
     def is_negative(self):
         return self._is_negative