supervertaler 1.9.163__py3-none-any.whl

modules/project_home_panel.py

@@ -0,0 +1,209 @@
+"""
+Project Home Panel - Collapsible sidebar like memoQ's Project Home
+"""
+
+from PyQt6.QtWidgets import (
+    QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel,
+    QFrame, QListWidget, QListWidgetItem, QScrollArea
+)
+from PyQt6.QtCore import Qt, pyqtSignal, QPropertyAnimation, QRect, QEasingCurve
+from PyQt6.QtGui import QColor, QFont, QIcon
+
+
+class ProjectHomeItem(QWidget):
+    """Individual item in the Project Home panel"""
+    
+    clicked = pyqtSignal(str)
+    
+    def __init__(self, icon: str, text: str, item_id: str, parent=None):
+        super().__init__(parent)
+        self.item_id = item_id
+        
+        layout = QHBoxLayout(self)
+        layout.setContentsMargins(12, 8, 12, 8)
+        layout.setSpacing(10)
+        
+        # Icon/emoji
+        icon_label = QLabel(icon)
+        icon_label.setStyleSheet("font-size: 18px; background: transparent;")
+        icon_label.setMaximumWidth(30)
+        layout.addWidget(icon_label)
+        
+        # Text
+        text_label = QLabel(text)
+        text_label.setStyleSheet("""
+            QLabel {
+                color: #2c3e50;
+                background: transparent;
+                font-size: 11px;
+                font-weight: 500;
+            }
+        """)
+        layout.addWidget(text_label, stretch=1)
+        
+        # Make the whole widget clickable
+        self.setCursor(Qt.CursorShape.PointingHandCursor)
+        self.setStyleSheet("""
+            ProjectHomeItem {
+                background: transparent;
+                border-radius: 4px;
+            }
+            ProjectHomeItem:hover {
+                background: rgba(0, 0, 0, 0.05);
+            }
+        """)
+    
+    def mousePressEvent(self, event):
+        """Emit signal when clicked"""
+        self.clicked.emit(self.item_id)
+
+
+class ProjectHomePanel(QWidget):
+    """Collapsible Project Home panel similar to memoQ"""
+    
+    # Signals
+    item_selected = pyqtSignal(str)  # Emits item ID
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.is_expanded = False
+        self.setup_ui()
+    
+    def setup_ui(self):
+        """Setup the UI"""
+        main_layout = QVBoxLayout(self)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+        main_layout.setSpacing(0)
+        
+        # Header with tab
+        header = QFrame()
+        header.setStyleSheet("""
+            QFrame {
+                background: #f0f0f0;
+                border-right: 1px solid #cccccc;
+            }
+        """)
+        header_layout = QVBoxLayout(header)
+        header_layout.setContentsMargins(0, 0, 0, 0)
+        header_layout.setSpacing(0)
+        
+        # Tab button (left edge)
+        tab_btn = QPushButton("PROJECT\nHOME")
+        tab_btn.setMaximumWidth(60)
+        tab_btn.setMinimumHeight(80)
+        tab_btn.setStyleSheet("""
+            QPushButton {
+                background: #e8e8e8;
+                border: 1px solid #cccccc;
+                border-right: none;
+                color: #333333;
+                font-size: 9px;
+                font-weight: bold;
+                padding: 8px 4px;
+            }
+            QPushButton:hover {
+                background: #f5f5f5;
+            }
+            QPushButton:pressed {
+                background: #e0e0e0;
+            }
+        """)
+        tab_btn.clicked.connect(self.toggle_panel)
+        header_layout.addWidget(tab_btn)
+        header_layout.addStretch()
+        
+        main_layout.addWidget(header)
+        
+        # Panel content (initially hidden)
+        content = QFrame()
+        content.setMaximumWidth(0)  # Start collapsed
+        content.setStyleSheet("""
+            QFrame {
+                background: white;
+                border-right: 1px solid #cccccc;
+            }
+        """)
+        self.content = content
+        
+        content_layout = QVBoxLayout(content)
+        content_layout.setContentsMargins(0, 0, 0, 0)
+        content_layout.setSpacing(0)
+        
+        # Scroll area for items
+        scroll = QScrollArea()
+        scroll.setWidgetResizable(True)
+        scroll.setStyleSheet("""
+            QScrollArea {
+                background: white;
+                border: none;
+            }
+            QScrollBar:vertical {
+                background: white;
+                width: 12px;
+            }
+            QScrollBar::handle:vertical {
+                background: #cccccc;
+                border-radius: 6px;
+                min-height: 20px;
+            }
+            QScrollBar::handle:vertical:hover {
+                background: #999999;
+            }
+        """)
+        
+        # Container for items
+        items_container = QWidget()
+        items_layout = QVBoxLayout(items_container)
+        items_layout.setContentsMargins(0, 0, 0, 0)
+        items_layout.setSpacing(1)
+        
+        # Add menu items
+        menu_items = [
+            ("📋", "Overview", "overview"),
+            ("📁", "Translations", "translations"),
+            ("📚", "Live Docs", "live_docs"),
+            ("🔄", "Translation Memories", "memories"),
+            ("🏷️", "Termbases", "termbases"),
+            ("🤖", "Muses", "muses"),
+            ("⚙️", "Settings", "settings"),
+        ]
+        
+        self.items_map = {}
+        for icon, text, item_id in menu_items:
+            item = ProjectHomeItem(icon, text, item_id)
+            item.clicked.connect(self.on_item_clicked)
+            self.items_map[item_id] = item
+            items_layout.addWidget(item)
+        
+        items_layout.addStretch()
+        scroll.setWidget(items_container)
+        content_layout.addWidget(scroll)
+        
+        main_layout.addWidget(content)
+        
+        # Animation for expand/collapse
+        self.animation = QPropertyAnimation(self.content, b"maximumWidth")
+        self.animation.setDuration(300)
+        self.animation.setEasingCurve(QEasingCurve.Type.InOutQuad)
+    
+    def toggle_panel(self):
+        """Toggle panel expanded/collapsed"""
+        self.is_expanded = not self.is_expanded
+        
+        if self.is_expanded:
+            # Expand
+            self.animation.setEndValue(250)  # Width when expanded
+        else:
+            # Collapse
+            self.animation.setEndValue(0)  # Width when collapsed
+        
+        self.animation.start()
+    
+    def on_item_clicked(self, item_id: str):
+        """Handle item click"""
+        self.item_selected.emit(item_id)
+    
+    def setMaximumWidth(self, width: int):
+        """Override to allow animation"""
+        super().setMaximumWidth(width)
+        self.content.setMaximumWidth(width - 60)  # Account for tab width

modules/prompt_assistant.py

@@ -0,0 +1,357 @@
+"""
+AI Prompt Assistant Module
+
+Provides AI-powered prompt modification through natural language conversation.
+Part of Phase 1 implementation for v4.0.0-beta.
+
+Features:
+- Conversational prompt modification
+- Visual diff generation
+- Prompt versioning
+- Chat history tracking
+"""
+
+import difflib
+import json
+from datetime import datetime
+from typing import Dict, List, Optional, Tuple
+
+
+class PromptAssistant:
+    """AI-powered prompt modification and learning system"""
+    
+    def __init__(self, llm_client=None):
+        """
+        Initialize the Prompt Assistant.
+        
+        Args:
+            llm_client: LLM client instance (OpenAI, Anthropic, or Google)
+        """
+        self.llm = llm_client
+        self.chat_history = []
+        self.modification_history = []  # Track all prompt versions
+        
+        # System prompt for the AI prompt engineer
+        self.system_prompt = """You are an expert prompt engineer specialising in translation and localisation prompts.
+
+Your role is to help users refine and optimise their translation prompts based on their specific needs.
+
+When a user requests a modification to a prompt:
+1. Analyze the current prompt structure and content
+2. Understand the user's specific request
+3. Make targeted, meaningful changes that improve the prompt
+4. Explain your modifications clearly
+5. Preserve the overall structure unless changes are needed
+
+Guidelines:
+- Be specific and precise in your modifications
+- Maintain professional translation terminology
+- Consider linguistic and cultural aspects
+- Keep prompts concise but comprehensive
+- Use clear, actionable language
+
+Always respond with:
+1. An explanation of what you're changing and why
+2. The complete modified prompt text
+3. Key improvements made
+
+Format your response as JSON:
+{
+    "explanation": "Brief explanation of changes",
+    "modified_prompt": "Complete new prompt text",
+    "changes_summary": ["Change 1", "Change 2", ...]
+}"""
+    
+    def set_llm_client(self, llm_client):
+        """Set or update the LLM client"""
+        self.llm = llm_client
+    
+    def send_message(self, system_prompt: str, user_message: str, callback=None) -> Optional[str]:
+        """
+        Send a message to the LLM and get a response (for Style Guides chat).
+        
+        Args:
+            system_prompt: System prompt to use for context
+            user_message: User's message
+            callback: Optional callback function to handle response (called with response text)
+        
+        Returns:
+            Response text or None if LLM not available
+        """
+        if not self.llm:
+            response = """I'm currently running in offline mode without LLM access. 
+For now, you can use these commands:
+- 'add to all: [text]' to add rules to all languages
+- 'add to [Language]: [text]' to add rules to a specific language
+- 'show' to list available languages
+
+When you have your LLM configured (OpenAI, Claude, or Gemini), 
+I'll be able to provide AI-powered suggestions for your style guides."""
+            if callback:
+                callback(response)
+            return response
+        
+        try:
+            # Build messages
+            messages = [
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_message}
+            ]
+            
+            # Call LLM through self.llm (which should be an LLM client)
+            # This is designed to work with OpenAI, Anthropic, or Google clients
+            response_text = None
+            
+            # Try different LLM client interfaces
+            if hasattr(self.llm, 'chat'):
+                # OpenAI-style interface
+                response_text = self.llm.chat(messages)
+            elif hasattr(self.llm, 'generate_content'):
+                # Google Gemini-style interface
+                response_text = self.llm.generate_content(f"{system_prompt}\n\nUser: {user_message}").text
+            elif hasattr(self.llm, 'messages'):
+                # Anthropic-style interface
+                response = self.llm.messages.create(
+                    model="claude-opus-4-1",
+                    max_tokens=1024,
+                    system=system_prompt,
+                    messages=[{"role": "user", "content": user_message}]
+                )
+                response_text = response.content[0].text
+            else:
+                response_text = "LLM client not properly configured"
+            
+            # Record in chat history
+            self.chat_history.append({
+                "role": "user",
+                "content": user_message,
+                "timestamp": datetime.now().isoformat()
+            })
+            self.chat_history.append({
+                "role": "assistant",
+                "content": response_text,
+                "timestamp": datetime.now().isoformat()
+            })
+            
+            # Call callback if provided
+            if callback:
+                callback(response_text)
+            
+            return response_text
+            
+        except Exception as e:
+            error_msg = f"Error communicating with LLM: {str(e)}"
+            if callback:
+                callback(error_msg)
+            return error_msg
+
+    
+    def suggest_modification(self, prompt_name: str, current_prompt: str, user_request: str) -> Dict:
+        """
+        AI suggests changes to a prompt based on user request.
+        
+        Args:
+            prompt_name: Name of the prompt being modified
+            current_prompt: Current content of the prompt
+            user_request: User's natural language request for modification
+        
+        Returns:
+            Dictionary containing:
+            - explanation: Why the changes were made
+            - modified_prompt: New version of the prompt
+            - changes_summary: List of key changes
+            - diff: Unified diff showing changes
+            - success: Boolean indicating if modification succeeded
+        """
+        if not self.llm:
+            return {
+                "success": False,
+                "error": "No LLM client configured"
+            }
+        
+        try:
+            # Build conversation context
+            messages = [
+                {"role": "system", "content": self.system_prompt},
+                {"role": "user", "content": f"""Please modify the following translation prompt:
+
+PROMPT NAME: {prompt_name}
+
+CURRENT PROMPT:
+{current_prompt}
+
+USER REQUEST: {user_request}
+
+Provide your response in JSON format with explanation, modified_prompt, and changes_summary."""}
+            ]
+            
+            # Get AI response
+            response = self.llm.chat(messages)
+            
+            # Parse JSON response
+            try:
+                result = json.loads(response)
+            except json.JSONDecodeError:
+                # Try to extract JSON from markdown code blocks
+                if "```json" in response:
+                    json_text = response.split("```json")[1].split("```")[0].strip()
+                    result = json.loads(json_text)
+                elif "```" in response:
+                    json_text = response.split("```")[1].split("```")[0].strip()
+                    result = json.loads(json_text)
+                else:
+                    # Fallback: treat entire response as modified prompt
+                    result = {
+                        "explanation": "Modified based on your request",
+                        "modified_prompt": response,
+                        "changes_summary": ["Prompt updated"]
+                    }
+            
+            # Generate diff
+            diff = self.generate_diff(current_prompt, result["modified_prompt"])
+            
+            # Record in modification history
+            self.modification_history.append({
+                "timestamp": datetime.now().isoformat(),
+                "prompt_name": prompt_name,
+                "original": current_prompt,
+                "modified": result["modified_prompt"],
+                "request": user_request,
+                "explanation": result.get("explanation", "")
+            })
+            
+            # Record in chat history
+            self.chat_history.append({
+                "role": "user",
+                "content": user_request,
+                "timestamp": datetime.now().isoformat()
+            })
+            self.chat_history.append({
+                "role": "assistant",
+                "content": result.get("explanation", "Prompt modified"),
+                "timestamp": datetime.now().isoformat(),
+                "has_suggestion": True
+            })
+            
+            return {
+                "success": True,
+                "explanation": result.get("explanation", ""),
+                "modified_prompt": result["modified_prompt"],
+                "changes_summary": result.get("changes_summary", []),
+                "diff": diff,
+                "diff_html": self.generate_diff_html(current_prompt, result["modified_prompt"])
+            }
+            
+        except Exception as e:
+            return {
+                "success": False,
+                "error": f"Failed to generate modification: {str(e)}"
+            }
+    
+    def generate_diff(self, original: str, modified: str) -> str:
+        """
+        Generate unified diff between two prompts.
+        
+        Args:
+            original: Original prompt text
+            modified: Modified prompt text
+        
+        Returns:
+            Unified diff string
+        """
+        original_lines = original.splitlines(keepends=True)
+        modified_lines = modified.splitlines(keepends=True)
+        
+        diff = difflib.unified_diff(
+            original_lines,
+            modified_lines,
+            fromfile="Original",
+            tofile="Modified",
+            lineterm=""
+        )
+        
+        return ''.join(diff)
+    
+    def generate_diff_html(self, original: str, modified: str) -> List[Tuple[str, str]]:
+        """
+        Generate line-by-line diff suitable for colored display.
+        
+        Args:
+            original: Original prompt text
+            modified: Modified prompt text
+        
+        Returns:
+            List of tuples (change_type, line) where change_type is:
+            - 'equal': unchanged line
+            - 'delete': removed line
+            - 'insert': added line
+            - 'replace': modified line
+        """
+        original_lines = original.splitlines()
+        modified_lines = modified.splitlines()
+        
+        diff = []
+        matcher = difflib.SequenceMatcher(None, original_lines, modified_lines)
+        
+        for tag, i1, i2, j1, j2 in matcher.get_opcodes():
+            if tag == 'equal':
+                for line in original_lines[i1:i2]:
+                    diff.append(('equal', line))
+            elif tag == 'delete':
+                for line in original_lines[i1:i2]:
+                    diff.append(('delete', line))
+            elif tag == 'insert':
+                for line in modified_lines[j1:j2]:
+                    diff.append(('insert', line))
+            elif tag == 'replace':
+                for line in original_lines[i1:i2]:
+                    diff.append(('delete', line))
+                for line in modified_lines[j1:j2]:
+                    diff.append(('insert', line))
+        
+        return diff
+    
+    def get_chat_history(self) -> List[Dict]:
+        """Get the chat history"""
+        return self.chat_history
+    
+    def clear_chat_history(self):
+        """Clear the chat history"""
+        self.chat_history = []
+    
+    def get_modification_history(self) -> List[Dict]:
+        """Get the modification history"""
+        return self.modification_history
+    
+    def undo_last_modification(self) -> Optional[Dict]:
+        """
+        Get the previous version of the prompt for undo functionality.
+        
+        Returns:
+            Dictionary with original prompt info, or None if no history
+        """
+        if not self.modification_history:
+            return None
+        
+        last_mod = self.modification_history[-1]
+        return {
+            "prompt_name": last_mod["prompt_name"],
+            "original_prompt": last_mod["original"],
+            "timestamp": last_mod["timestamp"]
+        }
+    
+    def export_chat_session(self, filepath: str):
+        """
+        Export chat history and modifications to a file.
+        
+        Args:
+            filepath: Path to save the session data
+        """
+        session_data = {
+            "chat_history": self.chat_history,
+            "modification_history": self.modification_history,
+            "exported_at": datetime.now().isoformat()
+        }
+        
+        with open(filepath, 'w', encoding='utf-8') as f:
+            json.dump(session_data, f, indent=2, ensure_ascii=False)