Update parser.py

parser.py

@@ -130,23 +130,25 @@ def parse_node(node, lines, prev_end, level=0, total_lines=None, parent_path=Non
         processed_lines.add(start_line)
 
     # Handle variables in function definitions (input variables)
-    if isinstance(node, (ast.FunctionDef, ast.AsyncFunctionDef)) and node.args.args:
-        for arg in node.args.args:
-            var_start = start_line  # Assume args are on the same line as function def for simplicity
-            if var_start not in processed_lines:
-                counters['input_variable'] += 1
-                var_node_id = f"InputVariable[{counters['input_variable']}]"
-                var_info = {'role': 'input_variable', 'name': arg.arg}
-                parts.append({
-                    'category': 'input_variable',
-                    'source': f"    {arg.arg},",  # Indented as part of function
-                    'location': (var_start, var_start),
-                    'level': level + 1,
-                    'vector': create_vector('input_variable', level + 1, (var_start, var_start), total_lines, current_path, var_info),
-                    'parent_path': f"{current_path[0]} -> {var_node_id}",
-                    'node_id': var_node_id
-                })
-                processed_lines.add(var_start)
+    if isinstance(child, ast.Assign) or isinstance(child, ast.AnnAssign) or isinstance(child, ast.AugAssign):
+        targets = child.targets if isinstance(child, ast.Assign) else [child.target]
+        for target in targets:
+            if isinstance(target, ast.Name):
+                var_start = child.lineno
+                if var_start not in processed_lines and not is_blank_or_comment(lines[var_start - 1]):
+                    counters['assigned_variable'] += 1
+                    var_node_id = f"AssignedVariable[{counters['assigned_variable']}]"
+                    var_info = {'role': 'assigned_variable', 'name': target.id}
+                    parts.append({
+                        'category': 'assigned_variable',
+                        'source': lines[var_start - 1],
+                        'location': (var_start, var_start),
+                        'level': level + 1,
+                        'vector': create_vector('assigned_variable', level + 1, (var_start, var_start), total_lines, current_path, var_info),
+                        'parent_path': f"{current_path[0]} -> {var_node_id}",
+                        'node_id': var_node_id
+                    })
+                    processed_lines.add(var_start)
 
     # Process nested bodies
     nested_prev_end = start_line