Update parser.py

parser.py

@@ -29,7 +29,7 @@ def get_category(node):
         return 'other'
 
 def create_vector(category, level, location, total_lines, parent_path):
-    """Create an optimized vector representation for a code part."""
+    """Create a vector optimized for role similarity."""
     category_map = {
         'import': 1, 'assignment': 2, 'function': 3, 'async_function': 4, 'class': 5,
         'if': 6, 'while': 7, 'for': 8, 'try': 9, 'expression': 10, 'spacer': 11,
@@ -48,7 +48,6 @@ def create_vector(category, level, location, total_lines, parent_path):
     return [category_id, level, center_pos, span, parent_depth, parent_weight]
 
 def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=None, counters=None):
-    """Recursively parse an AST node with full hierarchy tracking."""
     if total_lines is None:
         total_lines = len(lines)
     if parent_path is None:
@@ -61,11 +60,9 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
     end_line = getattr(node, 'end_lineno', start_line)
     category = get_category(node)
 
-    # Assign a node_id to every part
     counters[category] += 1
     node_id = f"{category.capitalize()}[{counters[category]}]"
 
-    # Spacer before the node
     if start_line > prev_end + 1:
         spacer_lines = lines[prev_end:start_line - 1]
         counters['spacer'] += 1
@@ -81,7 +78,6 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
             'node_id': spacer_node_id
         })
 
-    # Main node
     stmt_lines = lines[start_line - 1:end_line]
     current_path = parent_path + [node_id]
     node_vector = create_vector(category, level, (start_line, end_line), total_lines, current_path)
@@ -95,7 +91,6 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
         'node_id': node_id
     })
 
-    # Process nested bodies
     nested_prev_end = end_line - 1
     for attr in ('body', 'orelse', 'handlers', 'finalbody'):
         if hasattr(node, attr) and getattr(node, attr):
@@ -113,7 +108,7 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
                         'level': level,
                         'vector': sub_vector,
                         'parent_path': ' -> '.join(parent_path) if parent_path else 'Top-Level',
-                        'node_id': node_id  # Inherits parent's node_id for context
+                        'node_id': node_id
                     })
                     child_parts = parse_node(child, lines, child.lineno - 1, level + 1, total_lines, current_path, counters)
                     sub_parts.extend(child_parts)
@@ -156,7 +151,6 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
     return parts
 
 def parse_python_code(code):
-    """Parse Python code string and return parts with hierarchy and vectors."""
     lines = code.splitlines(keepends=True)
     total_lines = len(lines)
     try: