Merge pull request #3 from pcphil/feat/widgets

Feat/widgets

Author: pcphil

.claude/settings.local.json

@@ -18,7 +18,8 @@
       "Bash(cd /c/Users/chung/projects/bite-size-notes && .venv/Scripts/python.exe -m pytest 2>&1)",
       "Bash(cd /c/Users/chung/projects/bite-size-notes && .venv/Scripts/ruff.exe check src/ 2>&1)",
       "Bash(uv run:*)",
-      "Bash(uv:*)"
+      "Bash(uv:*)",
+      "Bash(rm:*)"
     ]
   }
 }

pyproject.toml

@@ -12,8 +12,7 @@ dependencies = [
     "PySide6>=6.6,<7",
     "sounddevice>=0.4.6",
     "numpy>=1.24",
-    "soundfile>=0.12",
-    "faster-whisper>=1.0.0",
+"faster-whisper>=1.0.0",
     "pyaudiowpatch>=0.2.12; sys_platform == 'win32'",
     "llama-cpp-python>=0.3",
     "huggingface-hub>=0.20",

src/bite_size_notes/app.py

@@ -14,7 +14,7 @@
 def main():
     app = QApplication(sys.argv)
     app.setApplicationName("Bite-Size Notes")
-    app.setOrganizationName("BiteSize")
+    app.setOrganizationName("pcphil")
 
     # Apply theme
     config = AppConfig()

src/bite_size_notes/gui/chat_bubble.py

@@ -1,6 +1,6 @@
 """Transcript line widget for displaying a single transcript segment."""
 
-from PySide6.QtCore import Qt, Signal
+from PySide6.QtCore import Qt, QTimer, Signal
 from PySide6.QtGui import QFont, QTextOption
 from PySide6.QtWidgets import (
     QFrame,
@@ -30,16 +30,18 @@ def __init__(self, parent=None):
         self.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
         self.setSizePolicy(QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Fixed)
         self.setWordWrapMode(QTextOption.WrapMode.WordWrap)
+        self.setMinimumWidth(0)
 
     def _adjust_height(self):
         doc_height = int(self.document().size().height())
         margins = self.contentsMargins()
         height = doc_height + margins.top() + margins.bottom() + 4
         self.setFixedHeight(max(height, 28))
+        self.updateGeometry()
 
     def resizeEvent(self, event):
         super().resizeEvent(event)
-        self._adjust_height()
+        QTimer.singleShot(0, self._adjust_height)
 
     def showEvent(self, event):
         super().showEvent(event)

src/bite_size_notes/gui/export_dialog.py

@@ -1,4 +1,4 @@
-"""Export dialog for saving transcripts."""
+"""Export dialog for saving transcripts and output text."""
 
 from pathlib import Path
 
@@ -46,6 +46,29 @@ def export_transcript(session: TranscriptSession, parent: QWidget | None = None)
     content = exporters.get(ext, session.to_text)()
     path.write_text(content, encoding="utf-8")
 
-    QMessageBox.information(
-        parent, "Export", f"Transcript exported to:\n{path}"
+    QMessageBox.information(parent, "Export", f"Transcript exported to:\n{path}")
+
+
+def export_output(text: str, parent: QWidget | None = None):
+    """Show a file save dialog and export the summarization output as plain text."""
+    if not text.strip():
+        QMessageBox.information(parent, "Export", "No output to export.")
+        return
+
+    file_path, _ = QFileDialog.getSaveFileName(
+        parent,
+        "Export Output",
+        "output",
+        "Plain Text (*.txt)",
     )
+
+    if not file_path:
+        return
+
+    path = Path(file_path)
+    if not path.suffix:
+        path = path.with_suffix(".txt")
+
+    path.write_text(text, encoding="utf-8")
+
+    QMessageBox.information(parent, "Export", f"Output exported to:\n{path}")

src/bite_size_notes/gui/main_window.py

@@ -4,8 +4,9 @@
 import time
 
 from PySide6.QtCore import QThread, QTimer, Qt, Signal
-from PySide6.QtGui import QKeySequence, QShortcut
+from PySide6.QtGui import QFont, QKeySequence, QShortcut
 from PySide6.QtWidgets import (
+    QInputDialog,
     QLabel,
     QMainWindow,
     QMessageBox,
@@ -18,7 +19,7 @@
 from bite_size_notes.audio.capture import AudioCaptureThread
 from bite_size_notes.gui.themes import build_stylesheet, get_palette
 from bite_size_notes.audio.devices import get_default_mic, get_loopback_device
-from bite_size_notes.gui.export_dialog import export_transcript
+from bite_size_notes.gui.export_dialog import export_output, export_transcript
 from bite_size_notes.gui.notes_panel import NotesPanel
 from bite_size_notes.gui.output_panel import OutputPanel
 from bite_size_notes.gui.settings_dialog import SettingsDialog
@@ -147,20 +148,22 @@ def _setup_ui(self):
 
         layout.addWidget(self._splitter, 1)
 
-        # Connect sidebar collapse to redistribute splitter space
-        self._saved_splitter_sizes = self._splitter.sizes()
         self._saved_output_sizes = self._splitter.sizes()
-        self.sidebar.collapse_toggled.connect(self._on_sidebar_collapse_toggled)
 
         # Connect transcript view control signals
         self.transcript_view.record_clicked.connect(self._on_record_clicked)
         self.transcript_view.clear_clicked.connect(self._on_clear_clicked)
+        self.transcript_view.export_clicked.connect(self._on_export_clicked)
 
         # Connect output panel signals
         self.output_panel.notes_toggled.connect(self._toggle_notes)
-        self.output_panel.export_clicked.connect(self._on_export_clicked)
+        self.output_panel.export_output_clicked.connect(self._on_export_output_clicked)
         self.output_panel.collapse_toggled.connect(self._on_output_collapse_toggled)
         self.output_panel.bite_size_clicked.connect(self._on_bite_size_clicked)
+        self._splitter.splitterMoved.connect(self._on_splitter_moved)
+
+        # Connect sidebar rename signal
+        self.sidebar.rename_session_requested.connect(self._on_rename_session)
 
         self.setCentralWidget(central)
 
@@ -197,6 +200,10 @@ def _setup_status_bar(self):
         self.loopback_level.setTextVisible(True)
         self.statusBar().addPermanentWidget(self.loopback_level)
 
+        author_label = QLabel("By Philip Chung")
+        author_label.setFont(QFont("Segoe UI", 8))
+        self.statusBar().addPermanentWidget(author_label)
+
     def _setup_timers(self):
         # Timer to update duration and audio levels
         self.ui_timer = QTimer(self)
@@ -374,6 +381,9 @@ def _on_error(self, message: str):
     def _on_export_clicked(self):
         export_transcript(self.transcript_session, self)
 
+    def _on_export_output_clicked(self):
+        export_output(self.output_panel.text(), self)
+
     def _on_clear_clicked(self):
         self.transcript_session.clear()
         self.transcript_view.clear_transcript()
@@ -392,6 +402,9 @@ def _on_bite_size_clicked(self):
             return
 
         text = self.transcript_session.to_text()
+        notes = self.notes_panel.get_text().strip()
+        if notes:
+            text += f"\n\nNotes:\n{notes}"
         if not text.strip():
             QMessageBox.information(
                 self, "No Transcript", "There is no transcript to summarize."
@@ -410,6 +423,9 @@ def _on_bite_size_clicked(self):
     def _on_summarize_finished(self, result: str):
         self._summarize_thread = None
         self.output_panel.set_text(result)
+        self.transcript_session.summary = result
+        if self.transcript_session.segments:
+            self.session_store.save_session(self.transcript_session)
 
     def _on_summarize_error(self, message: str):
         self._summarize_thread = None
@@ -438,6 +454,10 @@ def _on_session_selected(self, session_id: str):
         for segment in session.segments:
             self.transcript_view.append_segment(segment)
         self.transcript_view.set_editable(True)
+        if session.summary:
+            self.output_panel.set_text(session.summary)
+        else:
+            self.output_panel.clear()
         self.sidebar.set_active_session(session_id)
 
     def _on_new_session(self):
@@ -447,9 +467,31 @@ def _on_new_session(self):
         if self.transcript_session.segments:
             self.session_store.save_session(self.transcript_session)
         self.transcript_session = TranscriptSession()
+        self.session_store.save_session(self.transcript_session)
         self.transcript_view.clear_transcript()
         self.transcript_view.set_editable(False)
         self.notes_panel.clear()
+        self.output_panel.clear()
+        self.sidebar.refresh_sessions(active_id=self.transcript_session.id)
+
+    def _on_rename_session(self, session_id: str):
+        """Prompt the user to rename a session."""
+        try:
+            session = self.session_store.load_session(session_id)
+        except Exception:
+            return
+        new_name, ok = QInputDialog.getText(
+            self,
+            "Rename Session",
+            "New name:",
+            text=session.title,
+        )
+        if not ok or not new_name.strip():
+            return
+        session.title = new_name.strip()
+        self.session_store.save_session(session)
+        if session_id == self.transcript_session.id:
+            self.transcript_session.title = session.title
         self.sidebar.refresh_sessions(active_id=self.transcript_session.id)
 
     def _on_delete_session(self, session_id: str):
@@ -478,16 +520,9 @@ def _toggle_notes(self):
             self.notes_panel.show()
             self.notes_panel.raise_()
 
-    def _on_sidebar_collapse_toggled(self, collapsed: bool):
-        """Redistribute splitter space when the sidebar collapses/expands."""
-        sizes = self._splitter.sizes()
-        total = sum(sizes)
-        if collapsed:
-            self._saved_splitter_sizes = sizes
-            self._splitter.setSizes([40, total - 40 - sizes[2], sizes[2]])
-        else:
-            # Restore previous proportions
-            self._splitter.setSizes(self._saved_splitter_sizes)
+    def _on_splitter_moved(self):
+        """Refresh bubble heights after the user drags the splitter."""
+        QTimer.singleShot(0, self.transcript_view._refresh_bubble_heights)
 
     def _on_output_collapse_toggled(self, collapsed: bool):
         """Redistribute splitter space when the output panel collapses/expands."""

src/bite_size_notes/gui/output_panel.py

@@ -16,7 +16,7 @@ class OutputPanel(QWidget):
 
     bite_size_clicked = Signal()
     notes_toggled = Signal()
-    export_clicked = Signal()
+    export_output_clicked = Signal()
     collapse_toggled = Signal(bool)  # emits True when collapsed
 
     def __init__(self, parent=None):
@@ -35,26 +35,23 @@ def _setup_ui(self):
         self._btn_row.setSpacing(4)
         self._btn_row.setAlignment(Qt.AlignmentFlag.AlignVCenter)
 
-        self._action_btns = []
-        for label in ("Bite Size It", "Tone", "Model"):
-            btn = QPushButton(label)
-            btn.setFixedHeight(28)
-            self._btn_row.addWidget(btn)
-            self._action_btns.append(btn)
-
-        self._action_btns[0].clicked.connect(self.bite_size_clicked.emit)
+        self._bite_size_btn = QPushButton("Bite Size It")
+        self._bite_size_btn.setFixedHeight(28)
+        self._btn_row.addWidget(self._bite_size_btn)
+        self._bite_size_btn.clicked.connect(self.bite_size_clicked.emit)
 
         self._copy_btn = QPushButton("\U0001f4cb")
         self._copy_btn.setObjectName("iconBtn")
         self._copy_btn.setFixedSize(28, 28)
         self._copy_btn.setToolTip("Copy")
+        self._copy_btn.clicked.connect(self._copy_to_clipboard)
         self._btn_row.addWidget(self._copy_btn)
 
         self._export_btn = QPushButton("\U0001f4e5")
         self._export_btn.setObjectName("iconBtn")
         self._export_btn.setFixedSize(28, 28)
-        self._export_btn.setToolTip("Export")
-        self._export_btn.clicked.connect(self.export_clicked.emit)
+        self._export_btn.setToolTip("Export output")
+        self._export_btn.clicked.connect(self.export_output_clicked.emit)
         self._btn_row.addWidget(self._export_btn)
 
         self._btn_row.addStretch()
@@ -90,8 +87,7 @@ def _setup_ui(self):
     def toggle_collapse(self):
         self._collapsed = not self._collapsed
         if self._collapsed:
-            for btn in self._action_btns:
-                btn.hide()
+            self._bite_size_btn.hide()
             self._copy_btn.hide()
             self._export_btn.hide()
             self._text_edit.hide()
@@ -101,8 +97,7 @@ def toggle_collapse(self):
             self.setMinimumWidth(40)
             self.setMaximumWidth(40)
         else:
-            for btn in self._action_btns:
-                btn.show()
+            self._bite_size_btn.show()
             self._copy_btn.show()
             self._export_btn.show()
             self._text_edit.show()
@@ -123,5 +118,14 @@ def append_text(self, text: str):
     def set_text(self, text: str):
         self._text_edit.setPlainText(text)
 
+    def text(self) -> str:
+        return self._text_edit.toPlainText()
+
+    def _copy_to_clipboard(self):
+        from PySide6.QtWidgets import QApplication
+
+        clipboard = QApplication.clipboard()
+        clipboard.setText(self._text_edit.toPlainText())
+
     def clear(self):
         self._text_edit.clear()