allow user to remove file if it's not processed yet

Author: daltonnyx

AgentCrew/modules/chat/message/command_processor.py

@@ -105,6 +105,10 @@ async def process_command(self, user_input: str) -> CommandResult:
         elif user_input.startswith("/file "):
             return self._handle_file_command(user_input)
 
+        # Drop command
+        elif user_input.startswith("/drop "):
+            return self._handle_drop_command(user_input)
+
         # Not a command
         return CommandResult(handled=False)
 
@@ -507,3 +511,50 @@ def _handle_file_command(self, user_input: str) -> CommandResult:
             )
 
         return CommandResult(handled=True, clear_flag=True)
+
+    def _handle_drop_command(self, user_input: str) -> CommandResult:
+        """Handle drop command to remove queued files."""
+        # Extract file identifier from user input
+        file_path = user_input[6:].strip()  # Remove "/drop " prefix
+
+        if not file_path:
+            # Show available files if no argument provided
+            if not self.message_handler._queued_attached_files:
+                self.message_handler._notify(
+                    "error", "No files are currently queued for processing"
+                )
+                return CommandResult(handled=True, clear_flag=True)
+
+            self.message_handler._notify(
+                "system_message",
+                "📋 Queued files:\n"
+                + "\n".join(self.message_handler._queued_attached_files)
+                + "\nUsage: /drop <file_id>",
+            )
+            return CommandResult(handled=True, clear_flag=True)
+
+        try:
+            # Get the file command and parse its file paths
+            try:
+                self.message_handler._queued_attached_files.remove(file_path)
+            except Exception:
+                self.message_handler._notify(
+                    "error", f"Cannot unqueue file: {file_path}"
+                )
+
+            # Update or remove the command
+            self.message_handler._notify(
+                "system_message", f"🗑️ Removed file from queue: {file_path}"
+            )
+
+            # Notify about the file being removed for UI updates
+            self.message_handler._notify("file_dropped", {"file_path": file_path})
+
+            return CommandResult(handled=True, clear_flag=True)
+
+        except ValueError as e:
+            self.message_handler._notify("error", f"Invalid file ID format: {str(e)}")
+            return CommandResult(handled=True, clear_flag=True)
+        except Exception as e:
+            self.message_handler._notify("error", f"Error removing file: {str(e)}")
+            return CommandResult(handled=True, clear_flag=True)

AgentCrew/modules/chat/message/handler.py

@@ -79,7 +79,6 @@ def _messages_append(self, message):
         self.streamline_messages.extend(std_msg)
 
     def _prepare_files_processing(self, file_command):
-        self._queued_attached_files.append(file_command)
         file_paths_str: str = file_command[6:].strip()
         file_paths: List[str] = [
             os.path.expanduser(path.strip())
@@ -88,6 +87,7 @@ def _prepare_files_processing(self, file_command):
         ]
 
         for file_path in file_paths:
+            self._queued_attached_files.append(file_path)
             self._notify("file_processing", {"file_path": file_path})
 
     async def process_user_input(
@@ -146,7 +146,7 @@ async def process_user_input(
 
         while len(self._queued_attached_files) > 0:
             file_command = self._queued_attached_files.pop(0)
-            await self.command_processor.process_command(file_command)
+            await self.command_processor.process_command(f"/file {file_command}")
 
         # Add regular text message
         self._messages_append(

AgentCrew/modules/console/__init__.py

@@ -14,4 +14,4 @@
     "UIEffects",
     "ConfirmationHandler",
     "ConversationHandler",
-]
+]

AgentCrew/modules/console/completers.py

@@ -5,7 +5,7 @@
 import os
 import re
 
-COMPLETER_PATTERN = re.compile(r"[A-z0-9-_.]+")
+COMPLETER_PATTERN = re.compile(r"[a-zA-Z0-9-_.]+")
 
 
 class JumpCompleter(Completer):
@@ -120,6 +120,7 @@ def __init__(self, message_handler=None):
         self.agent_completer = AgentCompleter()
         self.jump_completer = JumpCompleter(message_handler)
         self.mcp_completer = MCPCompleter(message_handler)
+        self.drop_completer = DropCompleter(message_handler)
 
     def get_completions(self, document, complete_event):
         text = document.text
@@ -137,6 +138,8 @@ def get_completions(self, document, complete_event):
             yield from self.mcp_completer.get_completions(document, complete_event)
         elif text.startswith("/file "):
             yield from self.file_completer.get_completions(document, complete_event)
+        elif text.startswith("/drop "):
+            yield from self.drop_completer.get_completions(document, complete_event)
         elif text.startswith("/"):
             yield from self.get_command_completions(document)
 
@@ -169,6 +172,7 @@ def get_command_completions(self, document):
                 "List MCP prompts or fetch specific prompt (usage: /mcp [server_id/prompt_name])",
             ),
             ("/file", "Process a file (usage: /file <path>)"),
+            ("/drop", "Remove a queued file from processing (usage: /drop <file_id>)"),
             ("/list", "List available conversations"),
             ("/load", "Load a conversation (usage: /load <conversation_id>)"),
             ("/help", "Show help message"),
@@ -219,6 +223,43 @@ def get_completions(self, document, complete_event):
                             )
 
 
+class DropCompleter(Completer):
+    """Completer that shows available queued files when typing /drop command."""
+
+    def __init__(self, message_handler=None):
+        self.message_handler = message_handler
+
+    def get_completions(self, document, complete_event):
+        text = document.text
+
+        # Only provide completions for the /drop command
+        if text.startswith("/drop "):
+            word_before_cursor = document.get_word_before_cursor(
+                pattern=COMPLETER_PATTERN
+            )
+
+            # Get all queued attached files
+            queued_files = (
+                self.message_handler._queued_attached_files
+                if self.message_handler
+                else []
+            )
+
+            # Extract file paths from queued files and create completions
+            for file_path in queued_files:
+                if file_path.startswith(word_before_cursor):
+                    yield Completion(
+                        file_path,
+                        start_position=-len(word_before_cursor),
+                        display=file_path,
+                    )
+                elif word_before_cursor.startswith("drop"):
+                    yield Completion(
+                        file_path,
+                        display=file_path,
+                    )
+
+
 class DirectoryListingCompleter(Completer):
     def __init__(self):
         # Use PathCompleter for the heavy lifting
@@ -230,7 +271,7 @@ def get_completions(self, document, complete_event):
             return
         # Look for patterns that might indicate a path
         # This regex searches for a potential directory path
-        path_match = re.search(r"((~|\.{1,2})?/[^\s]*|~)$", text)
+        path_match = re.search(r"((~|\.{1,2})?/[^\s/]*|~)$", text)
 
         if path_match:
             path = path_match.group(0)

AgentCrew/modules/console/console_ui.py

@@ -184,6 +184,8 @@ def listen(self, event: str, data: Any = None):
         elif event == "file_processing":
             self.ui_effects.stop_loading_animation()  # Stop loading on first chunk
             self.display_handlers.add_file(data["file_path"])
+        elif event == "file_dropped":
+            self.display_handlers._added_files.remove(data["file_path"])
         elif event == "consolidation_completed":
             self.display_handlers.display_consolidation_result(data)
         elif event == "conversations_listed":

AgentCrew/modules/gui/components/chat_components.py

@@ -98,6 +98,12 @@ def append_file(self, file_path, is_user=False, is_base64=False):
         else:
             message_bubble.display_file(file_path)
 
+        # Connect remove button if it exists
+        if message_bubble.remove_button:
+            message_bubble.remove_button.clicked.connect(
+                lambda: self._handle_file_remove(message_bubble)
+            )
+
         if is_user:
             container_layout.addStretch(1)  # Push to left
             container_layout.addWidget(message_bubble, 1)
@@ -278,3 +284,48 @@ def remove_messages_after(self, message_bubble):
         self.chat_window.current_response_bubble = None
         self.chat_window.current_response_container = None
         self.chat_window.expecting_response = False
+
+    def _handle_file_remove(self, message_bubble):
+        """Handle removal of a file from the processing queue."""
+        try:
+            if message_bubble.file_path:
+                # Use the chat window's worker to process the drop command
+                self.chat_window.llm_worker.process_request.emit(
+                    f"/drop {message_bubble.file_path}"
+                )
+
+                # Remove the message bubble from the UI immediately
+                self._remove_file_bubble(message_bubble)
+
+        except Exception as e:
+            print(f"Error removing file: {e}")
+
+    def _remove_file_bubble(self, message_bubble):
+        """Remove a specific file bubble from the chat UI."""
+        # Find the container that holds this message bubble
+        for i in range(self.chat_window.chat_layout.count()):
+            item = self.chat_window.chat_layout.itemAt(i)
+            if item and item.widget():
+                # Check if this widget contains our message bubble
+                from AgentCrew.modules.gui.widgets.message_bubble import MessageBubble
+
+                if message_bubble in item.widget().findChildren(MessageBubble):
+                    # Remove the container
+                    container = self.chat_window.chat_layout.takeAt(i)
+                    if container.widget():
+                        container.widget().deleteLater()
+                    break
+
+    def mark_file_processed(self, file_path):
+        """Mark a file as processed in all relevant file bubbles."""
+        # Find all file bubbles with matching file path and mark them as processed
+        for i in range(self.chat_window.chat_layout.count()):
+            item = self.chat_window.chat_layout.itemAt(i)
+            if item and item.widget():
+                # Find message bubbles in this container
+                from AgentCrew.modules.gui.widgets.message_bubble import MessageBubble
+
+                message_bubbles = item.widget().findChildren(MessageBubble)
+                for bubble in message_bubbles:
+                    if hasattr(bubble, "file_path") and bubble.file_path == file_path:
+                        bubble.mark_file_processed()

AgentCrew/modules/gui/qt_ui.py

@@ -510,6 +510,10 @@ def handle_event(self, event: str, data: Any):
             if not self.loading_conversation:
                 self.ui_state_manager.set_input_controls_enabled(True)
         elif event == "file_processed":
+            # Mark the file as processed in the chat components
+            file_path = data.get("file_path")
+            if file_path:
+                self.chat_components.mark_file_processed(file_path)
             self.current_file_bubble = None
         elif event == "image_generated":
             self.chat_components.append_file(data, False, True)