titan-cli 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

titan_plugin_git/steps/diff_summary_step.py

@@ -0,0 +1,182 @@
+# plugins/titan-plugin-git/titan_plugin_git/steps/diff_summary_step.py
+from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error, Skip
+from titan_plugin_git.messages import msg
+
+
+def show_uncommitted_diff_summary(ctx: WorkflowContext) -> WorkflowResult:
+    """
+    Show summary of uncommitted changes (git diff --stat).
+
+    Provides a visual overview of files changed and lines modified
+    before generating commit messages.
+
+    Returns:
+        Success: Always (even if no changes, for workflow continuity)
+    """
+    if not ctx.textual:
+        return Error("Textual UI context is not available for this step.")
+
+    if not ctx.git:
+        return Error(msg.Steps.Push.GIT_CLIENT_NOT_AVAILABLE)
+
+    # Begin step container
+    ctx.textual.begin_step("Show Changes Summary")
+
+    try:
+        # Get diff stat for uncommitted changes
+        stat_output = ctx.git._run_command(["git", "diff", "--stat", "HEAD"])
+
+        if not stat_output or not stat_output.strip():
+            ctx.textual.text("No uncommitted changes to show", markup="dim")
+            ctx.textual.end_step("success")
+            return Success("No changes")
+
+        # Show the stat summary with colors
+        ctx.textual.text("")  # spacing
+        ctx.textual.text("Changes summary:", markup="bold")
+        ctx.textual.text("")  # spacing
+
+        # Parse lines to find max filename length for alignment
+        file_lines = []
+        summary_lines = []
+        max_filename_len = 0
+
+        for line in stat_output.split('\n'):
+            if not line.strip():
+                continue
+
+            if '|' in line:
+                parts = line.split('|')
+                filename = parts[0].strip()
+                stats = '|'.join(parts[1:]) if len(parts) > 1 else ''
+                file_lines.append((filename, stats))
+                max_filename_len = max(max_filename_len, len(filename))
+            else:
+                summary_lines.append(line)
+
+        # Display aligned file changes
+        for filename, stats in file_lines:
+            # Pad filename to align pipes
+            padded_filename = filename.ljust(max_filename_len)
+
+            # Replace + with green and - with red
+            stats = stats.replace('+', '[green]+[/green]')
+            stats = stats.replace('-', '[red]-[/red]')
+
+            ctx.textual.text(f"  {padded_filename} | {stats}")
+
+        # Display summary lines
+        for line in summary_lines:
+            colored_line = line.replace('(+)', '[green](+)[/green]')
+            colored_line = colored_line.replace('(-)', '[red](-)[/red]')
+            ctx.textual.text(f"  {colored_line}", markup="dim")
+
+        ctx.textual.text("")  # spacing
+
+        # End step container with success
+        ctx.textual.end_step("success")
+
+        return Success("Diff summary displayed")
+
+    except Exception as e:
+        # Don't fail the workflow, just skip
+        ctx.textual.end_step("skip")
+        return Skip(f"Could not show diff summary: {e}")
+
+
+def show_branch_diff_summary(ctx: WorkflowContext) -> WorkflowResult:
+    """
+    Show summary of branch changes (git diff base...head --stat).
+
+    Provides a visual overview of files changed between branches
+    before generating PR descriptions.
+
+    Inputs (from ctx.data):
+        pr_head_branch (str): Head branch name
+
+    Returns:
+        Success: Always (even if no changes, for workflow continuity)
+    """
+    if not ctx.textual:
+        return Error("Textual UI context is not available for this step.")
+
+    # Begin step container
+    ctx.textual.begin_step("Show Branch Changes Summary")
+
+    if not ctx.git:
+        ctx.textual.text(msg.Steps.Push.GIT_CLIENT_NOT_AVAILABLE, markup="red")
+        ctx.textual.end_step("error")
+        return Error(msg.Steps.Push.GIT_CLIENT_NOT_AVAILABLE)
+
+    head_branch = ctx.get("pr_head_branch")
+    if not head_branch:
+        ctx.textual.text("No head branch specified", markup="dim")
+        ctx.textual.end_step("skip")
+        return Skip("No head branch specified")
+
+    base_branch = ctx.git.main_branch
+
+    try:
+        # Get diff stat between branches
+        stat_output = ctx.git._run_command([
+            "git", "diff", "--stat", f"{base_branch}...{head_branch}"
+        ])
+
+        if not stat_output or not stat_output.strip():
+            ctx.textual.text(f"No changes between {base_branch} and {head_branch}", markup="dim")
+            ctx.textual.end_step("success")
+            return Success("No changes")
+
+        # Show the stat summary with colors
+        ctx.textual.text("")  # spacing
+        ctx.textual.text(f"Changes in {head_branch} vs {base_branch}:", markup="bold")
+        ctx.textual.text("")  # spacing
+
+        # Parse lines to find max filename length for alignment
+        file_lines = []
+        summary_lines = []
+        max_filename_len = 0
+
+        for line in stat_output.split('\n'):
+            if not line.strip():
+                continue
+
+            if '|' in line:
+                parts = line.split('|')
+                filename = parts[0].strip()
+                stats = '|'.join(parts[1:]) if len(parts) > 1 else ''
+                file_lines.append((filename, stats))
+                max_filename_len = max(max_filename_len, len(filename))
+            else:
+                summary_lines.append(line)
+
+        # Display aligned file changes
+        for filename, stats in file_lines:
+            # Pad filename to align pipes
+            padded_filename = filename.ljust(max_filename_len)
+
+            # Replace + with green and - with red
+            stats = stats.replace('+', '[green]+[/green]')
+            stats = stats.replace('-', '[red]-[/red]')
+
+            ctx.textual.text(f"  {padded_filename} | {stats}")
+
+        # Display summary lines
+        for line in summary_lines:
+            colored_line = line.replace('(+)', '[green](+)[/green]')
+            colored_line = colored_line.replace('(-)', '[red](-)[/red]')
+            ctx.textual.text(f"  {colored_line}", markup="dim")
+
+        ctx.textual.text("")  # spacing
+
+        ctx.textual.end_step("success")
+        return Success("Branch diff summary displayed")
+
+    except Exception as e:
+        # Don't fail the workflow, just skip
+        ctx.textual.text(f"Could not show branch diff summary: {e}", markup="yellow")
+        ctx.textual.end_step("skip")
+        return Skip(f"Could not show branch diff summary: {e}")
+
+
+__all__ = ["show_uncommitted_diff_summary", "show_branch_diff_summary"]

titan_plugin_git/steps/push_step.py

@@ -2,7 +2,6 @@
 from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error
 from titan_plugin_git.exceptions import GitCommandError
 from titan_plugin_git.messages import msg
-from titan_cli.ui.tui.widgets import Panel
 
 def create_git_push_step(ctx: WorkflowContext) -> WorkflowResult:
     """
@@ -12,6 +11,7 @@ def create_git_push_step(ctx: WorkflowContext) -> WorkflowResult:
         remote (str, optional): The name of the remote to push to. Defaults to the client's default remote.
         branch (str, optional): The name of the branch to push. Defaults to the current branch.
         set_upstream (bool, optional): Whether to set the upstream tracking branch. Defaults to False.
+        push_tags (bool, optional): Whether to push tags along with the branch. Defaults to False.
 
     Requires:
         ctx.git: An initialized GitClient.
@@ -29,10 +29,14 @@ def create_git_push_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.git:
         return Error(msg.Steps.Push.GIT_CLIENT_NOT_AVAILABLE)
 
+    # Begin step container
+    ctx.textual.begin_step("Push changes to remote")
+
     # Get params from context
     remote = ctx.get('remote')
     branch = ctx.get('branch')
     set_upstream = ctx.get('set_upstream', False)
+    push_tags = ctx.get('push_tags', False)
 
     # Use defaults from the GitClient if not provided in the context
     remote_to_use = remote or ctx.git.default_remote
@@ -43,21 +47,33 @@ def create_git_push_step(ctx: WorkflowContext) -> WorkflowResult:
         if not ctx.git.branch_exists_on_remote(branch=branch_to_use, remote=remote_to_use):
             set_upstream = True
 
-        ctx.git.push(remote=remote_to_use, branch=branch_to_use, set_upstream=set_upstream)
-
-        # Show success panel
-        ctx.textual.mount(
-            Panel(
-                text=f"Pushed to {remote_to_use}/{branch_to_use}",
-                panel_type="success"
-            )
+        # Push branch (and tags if requested)
+        ctx.git.push(
+            remote=remote_to_use,
+            branch=branch_to_use,
+            set_upstream=set_upstream,
+            tags=push_tags
         )
 
+        # Show success message
+        success_msg = f"Pushed to {remote_to_use}/{branch_to_use}"
+        if push_tags:
+            success_msg += " (with tags)"
+
+        ctx.textual.text(success_msg, markup="green")
+
+        ctx.textual.end_step("success")
         return Success(
             message=msg.Git.PUSH_SUCCESS.format(remote=remote_to_use, branch=branch_to_use),
             metadata={"pr_head_branch": branch_to_use}
         )
     except GitCommandError as e:
-        return Error(msg.Steps.Push.PUSH_FAILED.format(e=e))
+        error_msg = msg.Steps.Push.PUSH_FAILED.format(e=e)
+        ctx.textual.text(error_msg, markup="red")
+        ctx.textual.end_step("error")
+        return Error(error_msg)
     except Exception as e:
-        return Error(msg.Git.UNEXPECTED_ERROR.format(e=e))
+        error_msg = msg.Git.UNEXPECTED_ERROR.format(e=e)
+        ctx.textual.text(error_msg, markup="red")
+        ctx.textual.end_step("error")
+        return Error(error_msg)

titan_plugin_git/steps/status_step.py

@@ -7,7 +7,6 @@ from titan_cli.engine import (
 )
 from titan_cli.messages import msg as global_msg
 from ..messages import msg
-from titan_cli.ui.tui.widgets import Panel
 
 def get_git_status_step(ctx: WorkflowContext) -> WorkflowResult:
     """
@@ -23,37 +22,35 @@ def get_git_status_step(ctx: WorkflowContext) -> WorkflowResult:
         Success: If the status was retrieved successfully.
         Error: If the GitClient is not available or the git command fails.
     """
+    if not ctx.textual:
+        return Error("Textual UI context is not available for this step.")
+
     if not ctx.git:
         return Error(msg.Steps.Status.GIT_CLIENT_NOT_AVAILABLE)
 
+    # Begin step container
+    ctx.textual.begin_step("Check Git Status")
+
     try:
         status = ctx.git.get_status()
 
-        if not ctx.textual:
-            return Error("Textual UI context is not available for this step.")
-
-        # If there are uncommitted changes, show warning panel
+        # If there are uncommitted changes, show text (border will be green on success)
         if not status.is_clean:
-            ctx.textual.mount(
-                Panel(
-                    text=global_msg.Workflow.UNCOMMITTED_CHANGES_WARNING,
-                    panel_type="warning"
-                )
-            )
+            ctx.textual.text(global_msg.Workflow.UNCOMMITTED_CHANGES_WARNING, markup="yellow")
             message = msg.Steps.Status.STATUS_RETRIEVED_WITH_UNCOMMITTED
         else:
-            # Show success panel for clean working directory
-            ctx.textual.mount(
-                Panel(
-                    text=msg.Steps.Status.WORKING_DIRECTORY_IS_CLEAN,
-                    panel_type="success"
-                )
-            )
+            # Show text for clean working directory (border will be green)
+            ctx.textual.text(msg.Steps.Status.WORKING_DIRECTORY_IS_CLEAN, markup="green")
             message = msg.Steps.Status.WORKING_DIRECTORY_IS_CLEAN
 
+        # End step container with success
+        ctx.textual.end_step("success")
+
         return Success(
             message=message,
             metadata={"git_status": status}
         )
     except Exception as e:
+        # End step container with error
+        ctx.textual.end_step("error")
         return Error(msg.Steps.Status.FAILED_TO_GET_STATUS.format(e=e))

titan_plugin_git/workflows/commit-ai.yaml

@@ -12,6 +12,11 @@ steps:
     plugin: git
     step: get_status
 
+  - id: diff_summary
+    name: "Show Changes Summary"
+    plugin: git
+    step: show_uncommitted_diff_summary
+
   - id: ai_commit_message
     name: "AI Commit Message"
     plugin: git

titan_plugin_github/agents/pr_agent.py

@@ -420,8 +420,9 @@ COMMIT_MESSAGE: <conventional commit message>"""
 ```
 
 ## CRITICAL Instructions
-1. **Title**: Follow conventional commits (type(scope): description), be clear and descriptive
-   - Examples: "feat(auth): add OAuth2 integration with Google provider", "fix(api): resolve race condition in cache invalidation"
+1. **Title**: Follow conventional commits (type(scope): Description), be clear and descriptive
+   - Start description with CAPITAL letter (imperative mood)
+   - Examples: "feat(auth): Add OAuth2 integration with Google provider", "fix(api): Resolve race condition in cache invalidation"
 
 2. **Description**: MUST follow the template structure above but keep it under {max_chars} characters total
    - Fill in the template sections (Summary, Type of Change, Changes Made, etc.)
@@ -460,6 +461,18 @@ DESCRIPTION:
         # Clean up title
         title = title.strip('"').strip("'")
 
+        # Ensure title subject starts with capital letter (conventional commits requirement)
+        # Format: type(scope): Description
+        if ':' in title:
+            parts = title.split(':', 1)
+            if len(parts) == 2:
+                prefix = parts[0]  # type(scope)
+                subject = parts[1].strip()  # description
+                # Capitalize first letter of subject
+                if subject and subject[0].islower():
+                    subject = subject[0].upper() + subject[1:]
+                title = f"{prefix}: {subject}"
+
         # Truncate description if needed (but not title)
         if len(description) > max_chars:
             description = description[:max_chars - 3] + "..."

titan_plugin_github/steps/ai_pr_step.py

@@ -6,7 +6,6 @@ Uses PRAgent to analyze branch context and generate PR content.
 """
 
 from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error, Skip
-from titan_cli.ui.tui.widgets import Panel
 
 from ..agents import PRAgent
 from ..messages import msg
@@ -43,23 +42,24 @@ def ai_suggest_pr_description_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("AI PR Description")
+
     # Check if AI is configured
     if not ctx.ai or not ctx.ai.is_available():
-        ctx.textual.mount(
-            Panel(
-                text=msg.GitHub.AI.AI_NOT_CONFIGURED,
-                panel_type="info"
-            )
-        )
+        ctx.textual.text(msg.GitHub.AI.AI_NOT_CONFIGURED, markup="dim")
+        ctx.textual.end_step("skip")
         return Skip(msg.GitHub.AI.AI_NOT_CONFIGURED)
 
     # Get Git client
     if not ctx.git:
+        ctx.textual.end_step("error")
         return Error(msg.GitHub.AI.GIT_CLIENT_NOT_AVAILABLE)
 
     # Get branch info
     head_branch = ctx.get("pr_head_branch")
     if not head_branch:
+        ctx.textual.end_step("error")
         return Error(msg.GitHub.AI.MISSING_PR_HEAD_BRANCH)
 
     base_branch = ctx.git.main_branch
@@ -88,12 +88,8 @@ def ai_suggest_pr_description_step(ctx: WorkflowContext) -> WorkflowResult:
 
         # Check if PR content was generated (need commits in branch)
         if not analysis.pr_title or not analysis.pr_body:
-            ctx.textual.mount(
-                Panel(
-                    text="No commits found in branch to generate PR description.",
-                    panel_type="info"
-                )
-            )
+            ctx.textual.text("No commits found in branch to generate PR description.", markup="dim")
+            ctx.textual.end_step("skip")
             return Skip("No commits found for PR generation")
 
         # Show PR size info
@@ -137,16 +133,9 @@ def ai_suggest_pr_description_step(ctx: WorkflowContext) -> WorkflowResult:
 
         if not use_ai_pr:
             ctx.textual.text(msg.GitHub.AI.AI_SUGGESTION_REJECTED, markup="yellow")
+            ctx.textual.end_step("skip")
             return Skip("User rejected AI-generated PR")
 
-        # Show success panel
-        ctx.textual.mount(
-            Panel(
-                text="AI generated PR description successfully",
-                panel_type="success"
-            )
-        )
-
         # Success - save to context
         metadata = {
             "ai_generated": True,
@@ -155,6 +144,7 @@ def ai_suggest_pr_description_step(ctx: WorkflowContext) -> WorkflowResult:
             "pr_size": analysis.pr_size
         }
 
+        ctx.textual.end_step("success")
         return Success(
             msg.GitHub.AI.AI_GENERATED_PR_DESCRIPTION_SUCCESS,
             metadata=metadata
@@ -165,6 +155,7 @@ def ai_suggest_pr_description_step(ctx: WorkflowContext) -> WorkflowResult:
         ctx.textual.text(msg.GitHub.AI.AI_GENERATION_FAILED.format(e=e), markup="yellow")
         ctx.textual.text(msg.GitHub.AI.FALLBACK_TO_MANUAL, markup="dim")
 
+        ctx.textual.end_step("skip")
         return Skip(msg.GitHub.AI.AI_GENERATION_FAILED.format(e=e))
 
 

titan_plugin_github/steps/create_pr_step.py

@@ -1,6 +1,5 @@
 # plugins/titan-plugin-github/titan_plugin_github/steps/create_pr_step.py
 from titan_cli.engine import WorkflowContext, WorkflowResult, Success, Error
-from titan_cli.ui.tui.widgets import Panel
 from ..exceptions import GitHubAPIError
 from ..messages import msg
 
@@ -33,10 +32,17 @@ def create_pr_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Create Pull Request")
+
     # 1. Get GitHub client from context
     if not ctx.github:
+        ctx.textual.text("GitHub client is not available in the workflow context.", markup="red")
+        ctx.textual.end_step("error")
         return Error("GitHub client is not available in the workflow context.")
     if not ctx.git:
+        ctx.textual.text("Git client is not available in the workflow context.", markup="red")
+        ctx.textual.end_step("error")
         return Error("Git client is not available in the workflow context.")
 
     # 2. Get required data from context and client config
@@ -47,6 +53,8 @@ def create_pr_step(ctx: WorkflowContext) -> WorkflowResult:
     is_draft = ctx.get("pr_is_draft", False)  # Default to not a draft
 
     if not all([title, base, head]):
+        ctx.textual.text("Missing required context for creating a pull request: pr_title, pr_head_branch.", markup="red")
+        ctx.textual.end_step("error")
         return Error(
             "Missing required context for creating a pull request: pr_title, pr_head_branch."
         )
@@ -63,24 +71,26 @@ def create_pr_step(ctx: WorkflowContext) -> WorkflowResult:
 
     # 4. Call the client method
     try:
+        ctx.textual.text(f"Creating pull request '{title}' from {head} to {base}...", markup="dim")
         pr = ctx.github.create_pull_request(
             title=title, body=body, base=base, head=head, draft=is_draft, assignees=assignees
         )
-        ctx.textual.mount(
-            Panel(
-                text=msg.GitHub.PR_CREATED.format(number=pr["number"], url=pr["url"]),
-                panel_type="success"
-            )
-        )
+        ctx.textual.text("")  # spacing
+        ctx.textual.text(msg.GitHub.PR_CREATED.format(number=pr["number"], url=pr["url"]), markup="green")
 
         # 4. Return Success with PR info
+        ctx.textual.end_step("success")
         return Success(
             "Pull request created successfully.",
             metadata={"pr_number": pr["number"], "pr_url": pr["url"]},
         )
     except GitHubAPIError as e:
+        ctx.textual.text(f"Failed to create pull request: {e}", markup="red")
+        ctx.textual.end_step("error")
         return Error(f"Failed to create pull request: {e}")
     except Exception as e:
+        ctx.textual.text(f"An unexpected error occurred while creating the pull request: {e}", markup="red")
+        ctx.textual.end_step("error")
         return Error(
             f"An unexpected error occurred while creating the pull request: {e}"
         )

titan_plugin_github/steps/github_prompt_steps.py

@@ -23,18 +23,27 @@ def prompt_for_pr_title_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Prompt for PR Title")
+
     # Skip if title already exists (e.g., from AI generation)
     if ctx.get("pr_title"):
+        ctx.textual.text("PR title already provided, skipping manual prompt.", markup="dim")
+        ctx.textual.end_step("skip")
         return Skip("PR title already provided, skipping manual prompt.")
 
     try:
         title = ctx.textual.ask_text(msg.Prompts.ENTER_PR_TITLE)
         if not title:
+            ctx.textual.end_step("error")
             return Error("PR title cannot be empty.")
+        ctx.textual.end_step("success")
         return Success("PR title captured", metadata={"pr_title": title})
     except (KeyboardInterrupt, EOFError):
+        ctx.textual.end_step("error")
         return Error("User cancelled.")
     except Exception as e:
+        ctx.textual.end_step("error")
         return Error(f"Failed to prompt for PR title: {e}", exception=e)
 
 
@@ -57,17 +66,25 @@ def prompt_for_pr_body_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Prompt for PR Body")
+
     # Skip if body already exists (e.g., from AI generation)
     if ctx.get("pr_body"):
+        ctx.textual.text("PR body already provided, skipping manual prompt.", markup="dim")
+        ctx.textual.end_step("skip")
         return Skip("PR body already provided, skipping manual prompt.")
 
     try:
         body = ctx.textual.ask_multiline(msg.Prompts.ENTER_PR_BODY, default="")
         # Body can be empty
+        ctx.textual.end_step("success")
         return Success("PR body captured", metadata={"pr_body": body})
     except (KeyboardInterrupt, EOFError):
+        ctx.textual.end_step("error")
         return Error("User cancelled.")
     except Exception as e:
+        ctx.textual.end_step("error")
         return Error(f"Failed to prompt for PR body: {e}", exception=e)
 
 
@@ -90,17 +107,25 @@ def prompt_for_issue_body_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Prompt for Issue Body")
+
     # Skip if body already exists (e.g., from AI generation)
     if ctx.get("issue_body"):
+        ctx.textual.text("Issue body already provided, skipping manual prompt.", markup="dim")
+        ctx.textual.end_step("skip")
         return Skip("Issue body already provided, skipping manual prompt.")
 
     try:
         body = ctx.textual.ask_multiline(msg.Prompts.ENTER_ISSUE_BODY, default="")
         # Body can be empty
+        ctx.textual.end_step("success")
         return Success("Issue body captured", metadata={"issue_body": body})
     except (KeyboardInterrupt, EOFError):
+        ctx.textual.end_step("error")
         return Error("User cancelled.")
     except Exception as e:
+        ctx.textual.end_step("error")
         return Error(f"Failed to prompt for issue body: {e}", exception=e)
 
 
@@ -111,7 +136,12 @@ def prompt_for_self_assign_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Assign Issue")
+
     if not ctx.github:
+        ctx.textual.text("GitHub client not available", markup="red")
+        ctx.textual.end_step("error")
         return Error("GitHub client not available")
 
     try:
@@ -121,11 +151,18 @@ def prompt_for_self_assign_step(ctx: WorkflowContext) -> WorkflowResult:
             if current_user not in assignees:
                 assignees.append(current_user)
             ctx.set("assignees", assignees)
+            ctx.textual.text(f"Issue will be assigned to {current_user}", markup="green")
+            ctx.textual.end_step("success")
             return Success(f"Issue will be assigned to {current_user}")
+        ctx.textual.text("Issue will not be assigned to current user", markup="dim")
+        ctx.textual.end_step("success")
         return Success("Issue will not be assigned to current user")
     except (KeyboardInterrupt, EOFError):
+        ctx.textual.end_step("error")
         return Error("User cancelled.")
     except Exception as e:
+        ctx.textual.text(f"Failed to prompt for self-assign: {e}", markup="red")
+        ctx.textual.end_step("error")
         return Error(f"Failed to prompt for self-assign: {e}", exception=e)
 
 
@@ -136,12 +173,19 @@ def prompt_for_labels_step(ctx: WorkflowContext) -> WorkflowResult:
     if not ctx.textual:
         return Error("Textual UI context is not available for this step.")
 
+    # Begin step container
+    ctx.textual.begin_step("Select Labels")
+
     if not ctx.github:
+        ctx.textual.text("GitHub client not available", markup="red")
+        ctx.textual.end_step("error")
         return Error("GitHub client not available")
 
     try:
         available_labels = ctx.github.list_labels()
         if not available_labels:
+            ctx.textual.text("No labels found in the repository.", markup="dim")
+            ctx.textual.end_step("skip")
             return Skip("No labels found in the repository.")
 
         # Show available labels
@@ -164,8 +208,16 @@ def prompt_for_labels_step(ctx: WorkflowContext) -> WorkflowResult:
             selected_labels = []
 
         ctx.set("labels", selected_labels)
+        if selected_labels:
+            ctx.textual.text(f"Selected labels: {', '.join(selected_labels)}", markup="green")
+        else:
+            ctx.textual.text("No labels selected", markup="dim")
+        ctx.textual.end_step("success")
         return Success("Labels selected")
     except (KeyboardInterrupt, EOFError):
+        ctx.textual.end_step("error")
         return Error("User cancelled.")
     except Exception as e:
+        ctx.textual.text(f"Failed to prompt for labels: {e}", markup="red")
+        ctx.textual.end_step("error")
         return Error(f"Failed to prompt for labels: {e}", exception=e)