feat(gui): also add diffview to gui

Author: daltonnyx

AgentCrew/modules/gui/components/tool_handlers.py

@@ -7,10 +7,12 @@
     QVBoxLayout,
     QPushButton,
     QLabel,
+    QScrollArea,
 )
 from PySide6.QtGui import QKeySequence, QShortcut
 from PySide6.QtCore import Qt
 from AgentCrew.modules.gui.widgets.tool_widget import ToolWidget
+from AgentCrew.modules.gui.widgets.diff_widget import DiffWidget
 
 
 class ToolEventHandler:
@@ -158,17 +160,18 @@ def handle_tool_confirmation_required(self, tool_info):
         tool_use = tool_info.copy()
         confirmation_id = tool_use.pop("confirmation_id")
 
-        # Special handling for 'ask' tool
         if tool_use["name"] == "ask":
             self._handle_ask_tool_confirmation(tool_use, confirmation_id)
             return
 
-        # Create dialog
+        if tool_use["name"] == "write_or_edit_file":
+            self._handle_write_or_edit_file_confirmation(tool_use, confirmation_id)
+            return
+
         dialog = QMessageBox(self.chat_window)
         dialog.setWindowTitle("Tool Execution Confirmation")
         dialog.setIcon(QMessageBox.Icon.Question)
 
-        # Format tool information for display
         tool_description = f"The assistant wants to use the '{tool_use['name']}' tool."
         params_text = ""
 
@@ -189,7 +192,6 @@ def handle_tool_confirmation_required(self, tool_info):
         text_edit.setReadOnly(True)
         text_edit.setText(params_text)
 
-        # Style the text edit to match the main theme
         text_edit.setStyleSheet(
             self.chat_window.style_provider.get_tool_dialog_text_edit_style()
         )
@@ -293,6 +295,167 @@ def handle_tool_denied(self, data):
         self.chat_window.current_response_bubble = None
         self.chat_window.current_response_container = None
 
+    def _handle_write_or_edit_file_confirmation(self, tool_use, confirmation_id):
+        """Handle write_or_edit_file tool confirmation with diff view."""
+        from PySide6.QtWidgets import QWidget, QHBoxLayout
+
+        tool_input = tool_use.get("input", {})
+        file_path = tool_input.get("file_path", "")
+        text_or_blocks = tool_input.get("text_or_search_replace_blocks", "")
+        percentage = tool_input.get("percentage_to_change", 0)
+
+        has_diff = DiffWidget.has_search_replace_blocks(text_or_blocks)
+
+        dialog = QDialog(self.chat_window)
+        dialog.setWindowTitle("File Edit Confirmation")
+        dialog.setMinimumWidth(800 if has_diff else 600)
+        dialog.setMinimumHeight(600 if has_diff else 400)
+
+        layout = QVBoxLayout()
+
+        diff_colors = self.chat_window.style_provider.get_diff_colors()
+        header_label = QLabel(f"📝 <b>Edit File:</b> {file_path}")
+        header_label.setStyleSheet(
+            f"font-size: 14px; padding: 10px; color: {diff_colors.get('header_text', '#89b4fa')};"
+        )
+        layout.addWidget(header_label)
+
+        info_label = QLabel(
+            f"Change percentage: {percentage}% | "
+            f"Mode: {'Search/Replace Blocks' if has_diff else 'Full Content'}"
+        )
+        info_label.setStyleSheet(
+            f"font-size: 11px; color: {diff_colors.get('line_number_text', '#6c7086')}; padding: 0 10px;"
+        )
+        layout.addWidget(info_label)
+
+        scroll_area = QScrollArea()
+        scroll_area.setWidgetResizable(True)
+        scroll_area.setMinimumHeight(350)
+
+        if has_diff:
+            diff_widget = DiffWidget(style_provider=self.chat_window.style_provider)
+            diff_widget.set_diff_content(text_or_blocks, file_path)
+            scroll_area.setWidget(diff_widget)
+        else:
+            content_widget = QTextEdit()
+            content_widget.setReadOnly(True)
+            content_widget.setText(text_or_blocks)
+            content_widget.setStyleSheet(
+                self.chat_window.style_provider.get_tool_dialog_text_edit_style()
+            )
+            scroll_area.setWidget(content_widget)
+
+        layout.addWidget(scroll_area)
+
+        buttons_container = QWidget()
+        buttons_layout = QHBoxLayout(buttons_container)
+        buttons_layout.setContentsMargins(0, 10, 0, 0)
+
+        yes_button = QPushButton("✓ Approve")
+        yes_button.setStyleSheet(
+            self.chat_window.style_provider.get_tool_dialog_yes_button_style()
+        )
+
+        all_button = QPushButton("✓✓ Yes to All")
+        all_button.setStyleSheet(
+            self.chat_window.style_provider.get_tool_dialog_all_button_style()
+        )
+
+        forever_button = QPushButton("∞ Forever")
+        forever_button.setStyleSheet(
+            self.chat_window.style_provider.get_tool_dialog_all_button_style()
+        )
+
+        no_button = QPushButton("✗ Deny")
+        no_button.setStyleSheet(
+            self.chat_window.style_provider.get_tool_dialog_no_button_style()
+        )
+
+        buttons_layout.addWidget(yes_button)
+        buttons_layout.addWidget(all_button)
+        buttons_layout.addWidget(forever_button)
+        buttons_layout.addStretch()
+        buttons_layout.addWidget(no_button)
+
+        layout.addWidget(buttons_container)
+
+        dialog.setLayout(layout)
+        dialog.setStyleSheet(self.chat_window.style_provider.get_config_window_style())
+
+        result = {"action": ""}
+
+        def on_yes():
+            result["action"] = "approve"
+            dialog.accept()
+
+        def on_all():
+            result["action"] = "approve_all"
+            dialog.accept()
+
+        def on_forever():
+            result["action"] = "approve_forever"
+            dialog.accept()
+
+        def on_no():
+            result["action"] = "deny"
+            dialog.reject()
+
+        yes_button.clicked.connect(on_yes)
+        all_button.clicked.connect(on_all)
+        forever_button.clicked.connect(on_forever)
+        no_button.clicked.connect(on_no)
+
+        approve_shortcut = QShortcut(QKeySequence("Ctrl+Return"), dialog)
+        approve_shortcut.activated.connect(on_yes)
+
+        dialog.exec()
+
+        if result["action"] == "approve":
+            self.chat_window.message_handler.resolve_tool_confirmation(
+                confirmation_id, {"action": "approve"}
+            )
+            self.chat_window.display_status_message(f"Approved file edit: {file_path}")
+
+        elif result["action"] == "approve_all":
+            self.chat_window.message_handler.resolve_tool_confirmation(
+                confirmation_id, {"action": "approve_all"}
+            )
+            self.chat_window.display_status_message(
+                "Approved all future write_or_edit_file calls"
+            )
+
+        elif result["action"] == "approve_forever":
+            from AgentCrew.modules.config import ConfigManagement
+
+            config_manager = ConfigManagement()
+            config_manager.write_auto_approval_tools("write_or_edit_file", add=True)
+
+            self.chat_window.message_handler.resolve_tool_confirmation(
+                confirmation_id, {"action": "approve_all"}
+            )
+            self.chat_window.display_status_message(
+                "write_or_edit_file will be auto-approved forever"
+            )
+
+        else:
+            denial_reason = self.show_denial_reason_dialog("write_or_edit_file")
+
+            if denial_reason:
+                self.chat_window.message_handler.resolve_tool_confirmation(
+                    confirmation_id, {"action": "deny", "reason": denial_reason}
+                )
+                self.chat_window.display_status_message(
+                    f"Denied file edit - Reason: {denial_reason[:50]}..."
+                    if len(denial_reason) > 50
+                    else f"Denied file edit - Reason: {denial_reason}"
+                )
+            else:
+                self.chat_window.message_handler.resolve_tool_confirmation(
+                    confirmation_id, {"action": "deny"}
+                )
+                self.chat_window.display_status_message("Denied file edit")
+
     def _handle_ask_tool_confirmation(self, tool_use, confirmation_id):
         """Handle the ask tool - display question and guided answers in GUI."""
         question = tool_use["input"].get("question", "")

AgentCrew/modules/gui/themes/atom_light.py

@@ -1279,6 +1279,26 @@ class AtomLightTheme:
         "default": "🔧",  # Default icon for unspecified tools
     }
 
+    # Diff Widget colors
+    DIFF_COLORS = {
+        "background": "#fafafa",  # Light background
+        "panel_bg": "#ffffff",  # White
+        "header_bg": "#e5e5e6",  # Light gray
+        "header_text": "#383a42",  # Dark text
+        "line_number_bg": "#f0f0f0",  # Very light gray
+        "line_number_text": "#9d9d9f",  # Medium gray
+        "removed_bg": "#ffeef0",  # Light red background
+        "removed_text": "#d73a49",  # Red text
+        "removed_highlight": "#d73a49",  # Red for character highlight
+        "added_bg": "#e6ffec",  # Light green background
+        "added_text": "#22863a",  # Green text
+        "added_highlight": "#22863a",  # Green for character highlight
+        "unchanged_text": "#9d9d9f",  # Gray
+        "border": "#e5e5e6",  # Light border
+        "block_header_bg": "#ddf4ff",  # Light blue
+        "block_header_text": "#0969da",  # Blue text
+    }
+
     # JSON Editor styles
     JSON_EDITOR_COLORS = {
         "background": "#fafafa",

AgentCrew/modules/gui/themes/catppuccin.py

@@ -1318,6 +1318,26 @@ class CatppuccinTheme:
         "default": "🔧",  # Default icon for unspecified tools
     }
 
+    # Diff Widget colors
+    DIFF_COLORS = {
+        "background": "#1e1e2e",  # Base
+        "panel_bg": "#313244",  # Surface0
+        "header_bg": "#45475a",  # Surface1
+        "header_text": "#cdd6f4",  # Text
+        "line_number_bg": "#181825",  # Mantle
+        "line_number_text": "#6c7086",  # Overlay0
+        "removed_bg": "#3b2d33",  # Subtle red background
+        "removed_text": "#f38ba8",  # Red
+        "removed_highlight": "#f38ba8",  # Red for character highlight
+        "added_bg": "#2d3b33",  # Subtle green background
+        "added_text": "#a6e3a1",  # Green
+        "added_highlight": "#a6e3a1",  # Green for character highlight
+        "unchanged_text": "#6c7086",  # Overlay0
+        "border": "#45475a",  # Surface1
+        "block_header_bg": "#585b70",  # Surface2
+        "block_header_text": "#b4befe",  # Lavender
+    }
+
     # JSON Editor styles
     JSON_EDITOR_COLORS = {
         "background": "#313244",  # Surface0

AgentCrew/modules/gui/themes/dracula.py

@@ -1286,6 +1286,26 @@ class DraculaTheme:
         "default": "🔧",  # Default icon for unspecified tools
     }
 
+    # Diff Widget colors
+    DIFF_COLORS = {
+        "background": "#282a36",  # Dracula Background
+        "panel_bg": "#21222c",  # Darker
+        "header_bg": "#44475a",  # Comment
+        "header_text": "#f8f8f2",  # Foreground
+        "line_number_bg": "#1e1f29",  # Very dark
+        "line_number_text": "#6272a4",  # Comment
+        "removed_bg": "#3d2a36",  # Subtle red background
+        "removed_text": "#ff5555",  # Red
+        "removed_highlight": "#ff5555",  # Red
+        "added_bg": "#2a3d36",  # Subtle green background
+        "added_text": "#50fa7b",  # Green
+        "added_highlight": "#50fa7b",  # Green
+        "unchanged_text": "#6272a4",  # Comment
+        "border": "#44475a",  # Border
+        "block_header_bg": "#bd93f9",  # Purple
+        "block_header_text": "#282a36",  # Background
+    }
+
     # JSON Editor styles
     JSON_EDITOR_COLORS = {
         "background": "#282a36",

AgentCrew/modules/gui/themes/nord.py

@@ -1279,6 +1279,26 @@ class NordTheme:
         "default": "🔧",  # Default icon for unspecified tools
     }
 
+    # Diff Widget colors
+    DIFF_COLORS = {
+        "background": "#2e3440",  # Polar Night
+        "panel_bg": "#3b4252",  # Polar Night lighter
+        "header_bg": "#434c5e",  # Polar Night
+        "header_text": "#d8dee9",  # Snow Storm
+        "line_number_bg": "#2e3440",  # Polar Night
+        "line_number_text": "#4c566a",  # Polar Night
+        "removed_bg": "#3d2f3a",  # Subtle red background
+        "removed_text": "#bf616a",  # Aurora Red
+        "removed_highlight": "#bf616a",  # Aurora Red
+        "added_bg": "#2f3d38",  # Subtle green background
+        "added_text": "#a3be8c",  # Aurora Green
+        "added_highlight": "#a3be8c",  # Aurora Green
+        "unchanged_text": "#4c566a",  # Polar Night
+        "border": "#4c566a",  # Border
+        "block_header_bg": "#5e81ac",  # Frost
+        "block_header_text": "#eceff4",  # Snow Storm
+    }
+
     # JSON Editor styles
     JSON_EDITOR_COLORS = {
         "background": "#2e3440",

AgentCrew/modules/gui/themes/saigontech.py

@@ -1273,6 +1273,26 @@ class SaigonTechTheme:
         "default": "🔧",  # Default icon for unspecified tools
     }
 
+    # Diff Widget colors
+    DIFF_COLORS = {
+        "background": "#334155",  # Slate 700
+        "panel_bg": "#475569",  # Slate 600
+        "header_bg": "#1e293b",  # Slate 800
+        "header_text": "#f8fafc",  # Slate 50
+        "line_number_bg": "#1e293b",  # Slate 800
+        "line_number_text": "#64748b",  # Slate 500
+        "removed_bg": "#4c2c3a",  # Subtle red
+        "removed_text": "#fca5a5",  # Red 300
+        "removed_highlight": "#fca5a5",  # Red 300
+        "added_bg": "#2c4c3a",  # Subtle green
+        "added_text": "#7fb239",  # Primary Green
+        "added_highlight": "#7fb239",  # Primary Green
+        "unchanged_text": "#64748b",  # Slate 500
+        "border": "#475569",  # Border
+        "block_header_bg": "#638b2c",  # Secondary Green
+        "block_header_text": "#f8fafc",  # Slate 50
+    }
+
     # JSON Editor styles
     JSON_EDITOR_COLORS = {
         "background": "#475569",  # Input Fields - Slate 600

AgentCrew/modules/gui/themes/style_provider.py

@@ -395,3 +395,28 @@ def get_markdown_editor_style(self):
 }
 """,
         )
+
+    def get_diff_colors(self):
+        """Get color scheme for diff widget from current theme."""
+        return getattr(
+            self.theme_class,
+            "DIFF_COLORS",
+            {
+                "background": "#1e1e2e",
+                "panel_bg": "#313244",
+                "header_bg": "#45475a",
+                "header_text": "#cdd6f4",
+                "line_number_bg": "#181825",
+                "line_number_text": "#6c7086",
+                "removed_bg": "#3b2d33",
+                "removed_text": "#f38ba8",
+                "removed_highlight": "#f38ba8",
+                "added_bg": "#2d3b33",
+                "added_text": "#a6e3a1",
+                "added_highlight": "#a6e3a1",
+                "unchanged_text": "#6c7086",
+                "border": "#45475a",
+                "block_header_bg": "#585b70",
+                "block_header_text": "#b4befe",
+            },
+        )