Fix plan display: filter out section headers like Title/Description

chainlit_app.py

@@ -22,8 +22,8 @@ from concurrent.futures import ThreadPoolExecutor
 # ============================================================
 # STARTUP VERSION CHECK - Change this to detect if rebuild worked
 # ============================================================
-APP_VERSION = "3.6.0-detailed-progress"
-BUILD_ID = "2024-12-20-v5"
+APP_VERSION = "3.6.1-plan-fix"
+BUILD_ID = "2024-12-20-v6"
 print("=" * 60)
 print(f"[STARTUP] CodePilot Chainlit App")
 print(f"[STARTUP] APP_VERSION: {APP_VERSION}")
@@ -372,6 +372,27 @@ def format_plan_display(plan: str) -> str:
     plan_lines = plan.split('\n')
     steps = []
 
+    # Section headers to skip (template sections, not actual steps)
+    skip_patterns = [
+        'title:', 'description:', 'overview:', 'summary:', 'objective:',
+        'installation:', 'running', 'testing', 'use case:', 'example:',
+        'note:', 'warning:', 'important:', 'files:', 'dependencies:',
+        'prerequisites:', 'requirements:', 'setup:', 'configuration:',
+        'readme', 'docstring', 'documentation'
+    ]
+
+    def is_section_header(text: str) -> bool:
+        """Check if text is a section header, not an action step."""
+        text_lower = text.lower().strip()
+        # Skip if starts with common header patterns
+        for pattern in skip_patterns:
+            if text_lower.startswith(pattern):
+                return True
+        # Skip if it's just a label ending with colon and nothing else useful
+        if text_lower.endswith(':') and len(text_lower) < 30:
+            return True
+        return False
+
     # Strategy 1: Look for existing numbered steps (1., 2., etc.)
     for line in plan_lines:
         stripped = line.strip()
@@ -381,8 +402,10 @@ def format_plan_display(plan: str) -> str:
             match = re.match(r'^(\d+)[.)\]:]\s*(.+)', stripped)
             if match:
                 step_text = match.group(2).strip()
-                # Skip if it's just a file path or too short
-                if len(step_text) > 10 and not step_text.startswith('/'):
+                # Skip section headers, file paths, and too short items
+                if (len(step_text) > 10 and
+                    not step_text.startswith('/') and
+                    not is_section_header(step_text)):
                     steps.append(step_text)
 
     # Strategy 2: Look for bullet points if no numbered steps found
@@ -393,11 +416,11 @@ def format_plan_display(plan: str) -> str:
             # Match bullet points
             if stripped.startswith(('-', '*', '•')) and len(stripped) > 5:
                 step_text = stripped.lstrip('-*• ').strip()
-                # Skip headers, file paths, and very short items
+                # Skip headers, file paths, section headers
                 if (len(step_text) > 15 and
                     not step_text.startswith('#') and
                     not step_text.startswith('/') and
-                    ':' not in step_text[:5]):  # Skip "Note:" etc.
+                    not is_section_header(step_text)):
                     steps.append(step_text)
 
     # Strategy 3: Extract key sentences with action verbs
@@ -406,12 +429,16 @@ def format_plan_display(plan: str) -> str:
         action_verbs = ['create', 'add', 'implement', 'write', 'update', 'modify',
                         'define', 'set up', 'configure', 'import', 'export', 'build']
         for line in plan_lines:
-            stripped = line.strip().lower()
+            stripped = line.strip()
+            stripped_lower = stripped.lower()
+            # Skip section headers
+            if is_section_header(stripped):
+                continue
             for verb in action_verbs:
-                if verb in stripped and len(line.strip()) > 20:
+                if verb in stripped_lower and len(stripped) > 20:
                     # Clean up the line
-                    clean = line.strip().lstrip('-*• 0123456789.):]').strip()
-                    if clean and clean not in steps:
+                    clean = stripped.lstrip('-*• 0123456789.):]').strip()
+                    if clean and clean not in steps and not is_section_header(clean):
                         steps.append(clean)
                         break
 
@@ -435,7 +462,7 @@ def format_plan_display(plan: str) -> str:
     else:
         # Fallback: show first meaningful line
         for line in plan_lines:
-            if line.strip() and not line.startswith('#'):
+            if line.strip() and not line.startswith('#') and not is_section_header(line):
                 lines.append(f"1. {line.strip()[:80]}")
                 break
         if len(lines) == 1: