supervertaler 1.9.116__py3-none-any.whl → 1.9.172__py3-none-any.whl

modules/unified_prompt_manager_qt.py

@@ -18,7 +18,7 @@ from PyQt6.QtWidgets import (
     QTextEdit, QPlainTextEdit, QSplitter, QGroupBox, QMessageBox, QFileDialog,
     QInputDialog, QLineEdit, QFrame, QMenu, QCheckBox, QSizePolicy, QScrollArea, QTabWidget,
     QListWidget, QListWidgetItem, QStyledItemDelegate, QStyleOptionViewItem, QApplication, QDialog,
-    QAbstractItemView
+    QAbstractItemView, QTableWidget, QTableWidgetItem, QHeaderView
 )
 from PyQt6.QtCore import Qt, QSettings, pyqtSignal, QThread, QSize, QRect, QRectF
 from PyQt6.QtGui import QFont, QColor, QAction, QIcon, QPainter, QPen, QBrush, QPainterPath, QLinearGradient
@@ -31,6 +31,89 @@ from modules.ai_file_viewer_dialog import FileViewerDialog, FileRemoveConfirmDia
 from modules.ai_actions import AIActionSystem
 
 
+class CheckmarkCheckBox(QCheckBox):
+    """Custom checkbox with green background and white checkmark when checked"""
+    
+    def __init__(self, text="", parent=None):
+        super().__init__(text, parent)
+        self.setCheckable(True)
+        self.setEnabled(True)
+        self.setStyleSheet("""
+            QCheckBox {
+                font-size: 9pt;
+                spacing: 6px;
+            }
+            QCheckBox::indicator {
+                width: 16px;
+                height: 16px;
+                border: 2px solid #999;
+                border-radius: 3px;
+                background-color: white;
+            }
+            QCheckBox::indicator:checked {
+                background-color: #4CAF50;
+                border-color: #4CAF50;
+            }
+            QCheckBox::indicator:hover {
+                border-color: #666;
+            }
+            QCheckBox::indicator:checked:hover {
+                background-color: #45a049;
+                border-color: #45a049;
+            }
+        """)
+    
+    def paintEvent(self, event):
+        """Override paint event to draw white checkmark when checked"""
+        super().paintEvent(event)
+        
+        if self.isChecked():
+            # Get the indicator rectangle using QStyle
+            from PyQt6.QtWidgets import QStyleOptionButton
+            from PyQt6.QtCore import QPointF
+            
+            opt = QStyleOptionButton()
+            self.initStyleOption(opt)
+            indicator_rect = self.style().subElementRect(
+                self.style().SubElement.SE_CheckBoxIndicator,
+                opt,
+                self
+            )
+            
+            if indicator_rect.isValid():
+                # Draw white checkmark
+                painter = QPainter(self)
+                painter.setRenderHint(QPainter.RenderHint.Antialiasing)
+                pen_width = max(2.0, min(indicator_rect.width(), indicator_rect.height()) * 0.12)
+                painter.setPen(QPen(QColor(255, 255, 255), pen_width, Qt.PenStyle.SolidLine, Qt.PenCapStyle.RoundCap, Qt.PenJoinStyle.RoundJoin))
+                painter.setBrush(QColor(255, 255, 255))
+                
+                # Draw checkmark (✓ shape) - coordinates relative to indicator
+                x = indicator_rect.x()
+                y = indicator_rect.y()
+                w = indicator_rect.width()
+                h = indicator_rect.height()
+                
+                # Add padding (15% on all sides)
+                padding = min(w, h) * 0.15
+                x += padding
+                y += padding
+                w -= padding * 2
+                h -= padding * 2
+                
+                # Checkmark path
+                check_x1 = x + w * 0.10
+                check_y1 = y + h * 0.50
+                check_x2 = x + w * 0.35
+                check_y2 = y + h * 0.70
+                check_x3 = x + w * 0.90
+                check_y3 = y + h * 0.25
+                
+                # Draw two lines forming the checkmark
+                painter.drawLine(QPointF(check_x2, check_y2), QPointF(check_x3, check_y3))
+                painter.drawLine(QPointF(check_x1, check_y1), QPointF(check_x2, check_y2))
+
+
 class PromptLibraryTreeWidget(QTreeWidget):
     """Tree widget that supports drag-and-drop moves for prompt files."""
 
@@ -461,8 +544,8 @@ class UnifiedPromptManagerQt:
         self.log = parent_app.log if hasattr(parent_app, 'log') else print
         
         # Paths
-        self.prompt_library_dir = self.user_data_path / "Prompt_Library"
-        # Use Prompt_Library directly, not Prompt_Library/Library
+        self.prompt_library_dir = self.user_data_path / "prompt_library"
+        # Use prompt_library directly, not prompt_library/Library
         self.unified_library_dir = self.prompt_library_dir
 
         # Run migration if needed
@@ -496,7 +579,7 @@ class UnifiedPromptManagerQt:
         self._cached_document_markdown: Optional[str] = None  # Cached markdown conversion of current document
 
         # Initialize Attachment Manager
-        ai_assistant_dir = self.user_data_path / "AI_Assistant"
+        ai_assistant_dir = self.user_data_path / "ai_assistant"
         self.attachment_manager = AttachmentManager(
             base_dir=str(ai_assistant_dir),
             log_callback=self.log_message
@@ -563,6 +646,10 @@ class UnifiedPromptManagerQt:
         assistant_tab = self._create_ai_assistant_tab()
         self.sub_tabs.addTab(assistant_tab, "✨ AI Assistant")
 
+        # Tab 3: Placeholders Reference
+        placeholders_tab = self._create_placeholders_tab()
+        self.sub_tabs.addTab(placeholders_tab, "📝 Placeholders")
+
         # Connect tab change signal to update context
         self.sub_tabs.currentChanged.connect(self._on_tab_changed)
 
@@ -607,7 +694,7 @@ class UnifiedPromptManagerQt:
         layout.setSpacing(5)
         
         # Title
-        title = QLabel("🤖 Prompt Manager")
+        title = QLabel("📝 Prompt Manager")
         title.setStyleSheet("font-size: 16pt; font-weight: bold; color: #1976D2;")
         layout.addWidget(title, 0)
         
@@ -764,6 +851,153 @@ class UnifiedPromptManagerQt:
         
         return tab
     
+    def _create_placeholders_tab(self) -> QWidget:
+        """Create the Placeholders Reference sub-tab"""
+        tab = QWidget()
+        layout = QVBoxLayout(tab)
+        layout.setContentsMargins(10, 10, 10, 10)
+        layout.setSpacing(5)
+        
+        # Header (matches standard tool style: Superbench, AutoFingers, TMX Editor)
+        header = QLabel("📝 Available Placeholders")
+        header.setStyleSheet("font-size: 16pt; font-weight: bold; color: #1976D2;")
+        layout.addWidget(header, 0)
+        
+        # Description box (matches standard tool style)
+        description = QLabel(
+            "Use these placeholders in your prompts. They will be replaced with actual values when the prompt runs."
+        )
+        description.setWordWrap(True)
+        description.setStyleSheet("color: #666; padding: 5px; background-color: #E3F2FD; border-radius: 3px;")
+        layout.addWidget(description, 0)
+        
+        # Horizontal splitter for table and tips
+        splitter = QSplitter(Qt.Orientation.Horizontal)
+        splitter.setHandleWidth(3)
+        
+        # Left: Table with placeholders
+        table = QTableWidget()
+        table.setColumnCount(3)
+        table.setHorizontalHeaderLabels(["Placeholder", "Description", "Example"])
+        table.horizontalHeader().setStretchLastSection(True)
+        table.verticalHeader().setVisible(False)
+        table.setEditTriggers(QTableWidget.EditTrigger.NoEditTriggers)
+        table.setSelectionBehavior(QTableWidget.SelectionBehavior.SelectRows)
+        table.setAlternatingRowColors(True)
+        
+        # Placeholder data
+        placeholders = [
+            (
+                "{{SELECTION}}",
+                "Currently selected text in the grid (source or target cell)",
+                "If you select 'translation memory' in the grid, this will contain that text"
+            ),
+            (
+                "{{SOURCE_TEXT}}",
+                "Full text of the current source segment",
+                "The complete source sentence/paragraph from the active segment"
+            ),
+            (
+                "{{SOURCE_LANGUAGE}}",
+                "Project's source language",
+                "Dutch, English, German, French, etc."
+            ),
+            (
+                "{{TARGET_LANGUAGE}}",
+                "Project's target language",
+                "English, Spanish, Portuguese, etc."
+            ),
+            (
+                "{{SOURCE+TARGET_CONTEXT}}",
+                "Project segments with BOTH source and target text. Use for proofreading prompts.",
+                "[1] Source text\\n    → Target text\\n\\n[2] Source text\\n    → Target text\\n\\n..."
+            ),
+            (
+                "{{SOURCE_CONTEXT}}",
+                "Project segments with SOURCE ONLY. Use for translation/terminology questions.",
+                "[1] Source text\\n\\n[2] Source text\\n\\n[3] Source text\\n\\n..."
+            ),
+            (
+                "{{TARGET_CONTEXT}}",
+                "Project segments with TARGET ONLY. Use for consistency/style analysis.",
+                "[1] Target text\\n\\n[2] Target text\\n\\n[3] Target text\\n\\n..."
+            )
+        ]
+        
+        table.setRowCount(len(placeholders))
+        for row, (placeholder, description, example) in enumerate(placeholders):
+            # Placeholder column (monospace, bold)
+            item_placeholder = QTableWidgetItem(placeholder)
+            item_placeholder.setFont(QFont("Courier New", 10, QFont.Weight.Bold))
+            table.setItem(row, 0, item_placeholder)
+            
+            # Description column
+            item_desc = QTableWidgetItem(description)
+            item_desc.setToolTip(description)
+            table.setItem(row, 1, item_desc)
+            
+            # Example column (monospace, italic)
+            item_example = QTableWidgetItem(example)
+            item_example.setFont(QFont("Courier New", 9))
+            item_example.setToolTip(example)
+            table.setItem(row, 2, item_example)
+        
+        # Set column widths
+        table.setColumnWidth(0, 200)
+        table.setColumnWidth(1, 300)
+        header = table.horizontalHeader()
+        header.setSectionResizeMode(0, QHeaderView.ResizeMode.Fixed)
+        header.setSectionResizeMode(1, QHeaderView.ResizeMode.Interactive)
+        header.setSectionResizeMode(2, QHeaderView.ResizeMode.Stretch)
+        
+        # Adjust row heights for readability
+        for row in range(table.rowCount()):
+            table.setRowHeight(row, 60)
+        
+        splitter.addWidget(table)
+        
+        # Right: Usage tips panel
+        tips_panel = QWidget()
+        tips_layout = QVBoxLayout(tips_panel)
+        tips_layout.setContentsMargins(10, 0, 0, 0)
+        
+        tips_header = QLabel("💡 Usage Tips")
+        tips_header.setStyleSheet("font-weight: bold; font-size: 11pt; color: #2196F3; margin-bottom: 8px;")
+        tips_layout.addWidget(tips_header)
+        
+        tips_intro = QLabel(
+            "Use these placeholders in your prompts. They will be replaced with actual values when the prompt runs."
+        )
+        tips_intro.setWordWrap(True)
+        tips_intro.setStyleSheet("color: #666; margin-bottom: 15px; font-style: italic;")
+        tips_layout.addWidget(tips_intro)
+        
+        tips_text = QLabel(
+            "• Placeholders are case-sensitive (use UPPERCASE)\n\n"
+            "• Surround placeholders with double curly braces: {{ }}\n\n"
+            "• You can combine multiple placeholders in one prompt\n\n"
+            "• Use {{DOCUMENT_CONTEXT}} for context-aware translations\n\n"
+            "• Configure {{DOCUMENT_CONTEXT}} percentage in Settings → AI Settings"
+        )
+        tips_text.setWordWrap(True)
+        tips_text.setStyleSheet("color: #666; line-height: 1.6;")
+        tips_layout.addWidget(tips_text)
+        
+        tips_layout.addStretch()
+        
+        tips_panel.setMinimumWidth(280)
+        tips_panel.setMaximumWidth(400)
+        splitter.addWidget(tips_panel)
+        
+        # Set splitter proportions (75% table, 25% tips)
+        splitter.setSizes([750, 250])
+        splitter.setStretchFactor(0, 1)  # Table expands
+        splitter.setStretchFactor(1, 0)  # Tips panel fixed-ish
+        
+        layout.addWidget(splitter, 1)  # 1 = stretch to fill all available space
+        
+        return tab
+    
     def _create_context_sidebar(self) -> QWidget:
         """Create context sidebar showing available resources"""
         panel = QWidget()
@@ -1312,7 +1546,7 @@ class UnifiedPromptManagerQt:
     
     def _create_active_config_panel(self) -> QGroupBox:
         """Create active prompt configuration panel"""
-        group = QGroupBox("Active Configuration")
+        group = QGroupBox("Active Prompt")
         layout = QVBoxLayout()
         
         # Mode info (read-only, auto-selected)
@@ -1470,10 +1704,10 @@ class UnifiedPromptManagerQt:
         self.editor_quickmenu_label_input.setPlaceholderText("Label shown in QuickMenu")
         quickmenu_layout.addWidget(self.editor_quickmenu_label_input, 2)
 
-        self.editor_quickmenu_in_grid_cb = QCheckBox("Show in Grid right-click QuickMenu")
+        self.editor_quickmenu_in_grid_cb = CheckmarkCheckBox("Show in QuickMenu (in-app)")
         quickmenu_layout.addWidget(self.editor_quickmenu_in_grid_cb, 2)
 
-        self.editor_quickmenu_in_quickmenu_cb = QCheckBox("Show in QuickMenu")
+        self.editor_quickmenu_in_quickmenu_cb = CheckmarkCheckBox("Show in QuickMenu (global)")
         quickmenu_layout.addWidget(self.editor_quickmenu_in_quickmenu_cb, 1)
 
         layout.addLayout(quickmenu_layout)
@@ -1526,26 +1760,7 @@ class UnifiedPromptManagerQt:
             item.setFlags(item.flags() | Qt.ItemFlag.ItemIsDragEnabled)
             favorites_root.addChild(item)
         
-        # QuickMenu section (legacy kind name: quick_run)
-        quick_run_root = QTreeWidgetItem(["⚡ QuickMenu"])
-        # Special node: not draggable/droppable
-        quick_run_root.setData(0, Qt.ItemDataRole.UserRole, {'type': 'special', 'kind': 'quick_run'})
-        quick_run_root.setExpanded(False)
-        font = quick_run_root.font(0)
-        font.setBold(True)
-        quick_run_root.setFont(0, font)
-        self.tree_widget.addTopLevelItem(quick_run_root)
-        
-        quickmenu_items = self.library.get_quickmenu_prompts() if hasattr(self.library, 'get_quickmenu_prompts') else self.library.get_quick_run_prompts()
-        self.log_message(f"🔍 DEBUG: QuickMenu count: {len(quickmenu_items)}")
-        for path, label in quickmenu_items:
-            item = QTreeWidgetItem([label])
-            item.setData(0, Qt.ItemDataRole.UserRole, {'type': 'prompt', 'path': path})
-            # Quick Run entries are shortcuts, but allow dragging to move the actual prompt file.
-            item.setFlags(item.flags() | Qt.ItemFlag.ItemIsDragEnabled)
-            quick_run_root.addChild(item)
-        
-        # Library folders
+        # Library folders (QuickMenu parent folder removed - folder hierarchy now defines menu structure)
         self.log_message(f"🔍 DEBUG: Building tree from {self.unified_library_dir}")
         self._build_tree_recursive(None, self.unified_library_dir, "")
         
@@ -1950,7 +2165,7 @@ class UnifiedPromptManagerQt:
                     indicators = []
                     if prompt_data.get('favorite'):
                         indicators.append("⭐")
-                    if prompt_data.get('quickmenu_quickmenu', prompt_data.get('quick_run', False)):
+                    if prompt_data.get('sv_quickmenu', prompt_data.get('quick_run', False)):
                         indicators.append("⚡")
                     if prompt_data.get('quickmenu_grid', False):
                         indicators.append("🖱️")
@@ -2018,7 +2233,7 @@ class UnifiedPromptManagerQt:
             action_fav.triggered.connect(lambda: self._toggle_favorite(path))
             
             # Toggle QuickMenu (legacy: quick_run)
-            if prompt_data.get('quickmenu_quickmenu', prompt_data.get('quick_run', False)):
+            if prompt_data.get('sv_quickmenu', prompt_data.get('quick_run', False)):
                 action_qr = menu.addAction("⚡ Remove from QuickMenu")
             else:
                 action_qr = menu.addAction("⚡ Add to QuickMenu")
@@ -2070,7 +2285,7 @@ class UnifiedPromptManagerQt:
         if hasattr(self, 'editor_quickmenu_in_grid_cb'):
             self.editor_quickmenu_in_grid_cb.setChecked(bool(prompt_data.get('quickmenu_grid', False)))
         if hasattr(self, 'editor_quickmenu_in_quickmenu_cb'):
-            self.editor_quickmenu_in_quickmenu_cb.setChecked(bool(prompt_data.get('quickmenu_quickmenu', prompt_data.get('quick_run', False))))
+            self.editor_quickmenu_in_quickmenu_cb.setChecked(bool(prompt_data.get('sv_quickmenu', prompt_data.get('quick_run', False))))
         self.editor_content.setPlainText(prompt_data.get('content', ''))
         
         # Store current path for saving
@@ -2083,119 +2298,127 @@ class UnifiedPromptManagerQt:
     
     def _save_current_prompt(self):
         """Save currently edited prompt"""
-        name = self.editor_name_input.text().strip()
-        description = self.editor_desc_input.text().strip()
-        content = self.editor_content.toPlainText().strip()
-        
-        # Name field now represents the complete filename with extension
-        # No stripping needed - user sees and edits the full filename
-
-        quickmenu_label = ''
-        quickmenu_grid = False
-        quickmenu_quickmenu = False
-        if hasattr(self, 'editor_quickmenu_label_input'):
-            quickmenu_label = self.editor_quickmenu_label_input.text().strip()
-        if hasattr(self, 'editor_quickmenu_in_grid_cb'):
-            quickmenu_grid = bool(self.editor_quickmenu_in_grid_cb.isChecked())
-        if hasattr(self, 'editor_quickmenu_in_quickmenu_cb'):
-            quickmenu_quickmenu = bool(self.editor_quickmenu_in_quickmenu_cb.isChecked())
-
-        if not name or not content:
-            QMessageBox.warning(self.main_widget, "Error", "Name and content are required")
-            return
-
-        # Check if this is a new prompt or editing existing
-        if hasattr(self, 'editor_current_path') and self.editor_current_path:
-            path = self.editor_current_path
-            
-            # Handle external prompts (save back to external file)
-            if path.startswith("[EXTERNAL] "):
-                external_file_path = path[11:]  # Remove "[EXTERNAL] " prefix
-                self._save_external_prompt(external_file_path, name, description, content)
-                return
+        try:
+            name = self.editor_name_input.text().strip()
+            description = self.editor_desc_input.text().strip()
+            content = self.editor_content.toPlainText().strip()
             
-            # Editing existing library prompt
-            if path not in self.library.prompts:
-                QMessageBox.warning(self.main_widget, "Error", "Prompt no longer exists")
+            # Name field now represents the complete filename with extension
+            # No stripping needed - user sees and edits the full filename
+
+            quickmenu_label = ''
+            quickmenu_grid = False
+            sv_quickmenu = False
+            if hasattr(self, 'editor_quickmenu_label_input'):
+                quickmenu_label = self.editor_quickmenu_label_input.text().strip()
+            if hasattr(self, 'editor_quickmenu_in_grid_cb'):
+                quickmenu_grid = bool(self.editor_quickmenu_in_grid_cb.isChecked())
+            if hasattr(self, 'editor_quickmenu_in_quickmenu_cb'):
+                sv_quickmenu = bool(self.editor_quickmenu_in_quickmenu_cb.isChecked())
+
+            if not name or not content:
+                QMessageBox.warning(self.main_widget, "Error", "Name and content are required")
                 return
 
-            prompt_data = self.library.prompts[path].copy()
-            old_filename = Path(path).name
-            
-            # Extract name without extension for metadata
-            name_without_ext = Path(name).stem
-            
-            prompt_data['name'] = name_without_ext
-            prompt_data['description'] = description
-            prompt_data['content'] = content
-            prompt_data['quickmenu_label'] = quickmenu_label or name_without_ext
-            prompt_data['quickmenu_grid'] = quickmenu_grid
-            prompt_data['quickmenu_quickmenu'] = quickmenu_quickmenu
-            # Keep legacy field in sync
-            prompt_data['quick_run'] = quickmenu_quickmenu
-            
-            # Check if filename changed - need to rename file
-            if old_filename != name:
-                from pathlib import Path
-                old_path = Path(path)
-                folder = str(old_path.parent) if old_path.parent != Path('.') else ''
-                new_path = f"{folder}/{name}" if folder else name
+            # Check if this is a new prompt or editing existing
+            if hasattr(self, 'editor_current_path') and self.editor_current_path:
+                path = self.editor_current_path
                 
-                # Delete old file and save to new location
-                if self.library.delete_prompt(path):
-                    if self.library.save_prompt(new_path, prompt_data):
-                        self.library.load_all_prompts()
-                        self._refresh_tree()
-                        self._select_and_reveal_prompt(new_path)
-                        self.editor_current_path = new_path  # Update to new path
-                        QMessageBox.information(self.main_widget, "Saved", f"Prompt renamed to '{name}' successfully!")
-                        self.log_message(f"✓ Renamed prompt: {old_filename} → {name}")
+                # Handle external prompts (save back to external file)
+                if path.startswith("[EXTERNAL] "):
+                    external_file_path = path[11:]  # Remove "[EXTERNAL] " prefix
+                    self._save_external_prompt(external_file_path, name, description, content)
+                    return
+                
+                # Editing existing library prompt
+                if path not in self.library.prompts:
+                    QMessageBox.warning(self.main_widget, "Error", "Prompt no longer exists")
+                    return
+
+                prompt_data = self.library.prompts[path].copy()
+                old_filename = Path(path).name
+                
+                # Extract name without extension for metadata
+                name_without_ext = Path(name).stem
+                
+                prompt_data['name'] = name_without_ext
+                prompt_data['description'] = description
+                prompt_data['content'] = content
+                prompt_data['quickmenu_label'] = quickmenu_label or name_without_ext
+                prompt_data['quickmenu_grid'] = quickmenu_grid
+                prompt_data['sv_quickmenu'] = sv_quickmenu
+                # Keep legacy field in sync
+                prompt_data['quick_run'] = sv_quickmenu
+                
+                # Check if filename changed - need to rename file
+                if old_filename != name:
+                    old_path = Path(path)
+                    folder = str(old_path.parent) if old_path.parent != Path('.') else ''
+                    new_path = f"{folder}/{name}" if folder else name
+                    
+                    # Delete old file and save to new location
+                    if self.library.delete_prompt(path):
+                        if self.library.save_prompt(new_path, prompt_data):
+                            self.library.load_all_prompts()
+                            self._refresh_tree()
+                            self._select_and_reveal_prompt(new_path)
+                            self.editor_current_path = new_path  # Update to new path
+                            QMessageBox.information(self.main_widget, "Saved", f"Prompt renamed to '{name}' successfully!")
+                            self.log_message(f"✓ Renamed prompt: {old_filename} → {name}")
+                        else:
+                            QMessageBox.warning(self.main_widget, "Error", "Failed to rename prompt")
                     else:
-                        QMessageBox.warning(self.main_widget, "Error", "Failed to rename prompt")
+                        QMessageBox.warning(self.main_widget, "Error", "Failed to delete old prompt file")
                 else:
-                    QMessageBox.warning(self.main_widget, "Error", "Failed to delete old prompt file")
+                    # Name unchanged, just update in place
+                    if self.library.save_prompt(path, prompt_data):
+                        QMessageBox.information(self.main_widget, "Saved", "Prompt updated successfully!")
+                        self._refresh_tree()
+                    else:
+                        QMessageBox.warning(self.main_widget, "Error", "Failed to save prompt")
             else:
-                # Name unchanged, just update in place
-                if self.library.save_prompt(path, prompt_data):
-                    QMessageBox.information(self.main_widget, "Saved", "Prompt updated successfully!")
+                # Creating new prompt
+                folder = getattr(self, 'editor_target_folder', 'Project Prompts')
+
+                # Create new prompt data
+                from datetime import datetime
+                prompt_data = {
+                    'name': name,
+                    'description': description,
+                    'content': content,
+                    'domain': '',
+                    'version': '1.0',
+                    'task_type': 'Translation',
+                    'favorite': False,
+                    # QuickMenu
+                    'quickmenu_label': quickmenu_label or name,
+                    'quickmenu_grid': quickmenu_grid,
+                    'sv_quickmenu': sv_quickmenu,
+                    # Legacy
+                    'quick_run': sv_quickmenu,
+                    'folder': folder,
+                    'tags': [],
+                    'created': datetime.now().strftime('%Y-%m-%d'),
+                    'modified': datetime.now().strftime('%Y-%m-%d')
+                }
+
+                # Create the prompt file (save_prompt creates new file if it doesn't exist)
+                relative_path = f"{folder}/{name}.svprompt"
+                if self.library.save_prompt(relative_path, prompt_data):
+                    QMessageBox.information(self.main_widget, "Created", f"Prompt '{name}' created successfully!")
+                    self.library.load_all_prompts()  # Reload to get new prompt in memory
                     self._refresh_tree()
+                    self.editor_current_path = relative_path  # Now editing this prompt
                 else:
-                    QMessageBox.warning(self.main_widget, "Error", "Failed to save prompt")
-        else:
-            # Creating new prompt
-            folder = getattr(self, 'editor_target_folder', 'Project Prompts')
-
-            # Create new prompt data
-            from datetime import datetime
-            prompt_data = {
-                'name': name,
-                'description': description,
-                'content': content,
-                'domain': '',
-                'version': '1.0',
-                'task_type': 'Translation',
-                'favorite': False,
-                # QuickMenu
-                'quickmenu_label': quickmenu_label or name,
-                'quickmenu_grid': quickmenu_grid,
-                'quickmenu_quickmenu': quickmenu_quickmenu,
-                # Legacy
-                'quick_run': quickmenu_quickmenu,
-                'folder': folder,
-                'tags': [],
-                'created': datetime.now().strftime('%Y-%m-%d'),
-                'modified': datetime.now().strftime('%Y-%m-%d')
-            }
-
-            # Create the prompt file (save_prompt creates new file if it doesn't exist)
-            relative_path = f"{folder}/{name}.svprompt"
-            if self.library.save_prompt(relative_path, prompt_data):
-                QMessageBox.information(self.main_widget, "Created", f"Prompt '{name}' created successfully!")
-                self.library.load_all_prompts()  # Reload to get new prompt in memory
-                self._refresh_tree()
-                self.editor_current_path = relative_path  # Now editing this prompt
-            else:
-                QMessageBox.warning(self.main_widget, "Error", "Failed to create prompt")
+                    QMessageBox.warning(self.main_widget, "Error", "Failed to create prompt")
+        
+        except Exception as e:
+            import traceback
+            error_msg = f"Prompt save error: {str(e)}\n{traceback.format_exc()}"
+            print(f"[ERROR] {error_msg}")
+            self.log_message(f"❌ Prompt save error: {str(e)}")
+            QMessageBox.critical(self.main_widget, "Save Error", f"Failed to save prompt:\n\n{str(e)}")
+            return
     
     def _save_external_prompt(self, file_path: str, name: str, description: str, content: str):
         """Save changes to an external prompt file"""
@@ -2421,7 +2644,7 @@ class UnifiedPromptManagerQt:
             'favorite': False,
             'quickmenu_label': name_without_ext,
             'quickmenu_grid': False,
-            'quickmenu_quickmenu': False,
+            'sv_quickmenu': False,
             'quick_run': False,
             'folder': folder_path,
             'tags': [],
@@ -2437,6 +2660,7 @@ class UnifiedPromptManagerQt:
             self._refresh_tree()
             self._select_and_reveal_prompt(relative_path)
             self._load_prompt_in_editor(relative_path)
+            self.btn_save_prompt.setEnabled(True)  # Ensure Save button is enabled for new prompt
             self.log_message(f"✓ Created new prompt '{name}' in folder: {folder_path}")
         else:
             QMessageBox.warning(self.main_widget, "Error", "Failed to create prompt")
@@ -2469,7 +2693,7 @@ class UnifiedPromptManagerQt:
         src_data['favorite'] = False
         src_data['quick_run'] = False
         src_data['quickmenu_grid'] = False
-        src_data['quickmenu_quickmenu'] = False
+        src_data['sv_quickmenu'] = False
         src_data['folder'] = folder
         src_data['created'] = datetime.now().strftime('%Y-%m-%d')
         src_data['modified'] = datetime.now().strftime('%Y-%m-%d')
@@ -3305,7 +3529,7 @@ Output complete ACTION."""
         """
         Save the markdown conversion of the current document.
 
-        Saves to: user_data_private/AI_Assistant/current_document/
+        Saves to: user_data_private/ai_assistant/current_document/
 
         Args:
             original_path: Original document file path
@@ -3313,7 +3537,7 @@ Output complete ACTION."""
         """
         try:
             # Create directory for current document markdown
-            doc_dir = self.user_data_path / "AI_Assistant" / "current_document"
+            doc_dir = self.user_data_path / "ai_assistant" / "current_document"
             doc_dir.mkdir(parents=True, exist_ok=True)
 
             # Create filename based on original