Upload 2 files

app.py

@@ -64,7 +64,7 @@ app.layout = html.Div([
             ),
             dcc.Checklist(
                 id="function-detail-toggle",
-                options=[{"label": "Show Function-Level Detail", "value": "show"}],
+                options=[{"label": "Show Function-Level Detail in Summaray tab", "value": "show"}],
                 value=[],
                 labelStyle={"display": "block", "fontWeight": "normal"},
                 style={"marginTop": "5px"}

utility.py

@@ -1,5 +1,17 @@
 import ast
 
+import ast
+
+class CallCollector(ast.NodeVisitor):
+    def __init__(self, defined_funcs):
+        self.calls = []
+        self.defined_funcs = defined_funcs
+
+    def visit_Call(self, node):
+        if isinstance(node.func, ast.Name) and node.func.id in self.defined_funcs:
+            self.calls.append(node.func.id)
+        self.generic_visit(node)
+
 def parse_functions_from_files(file_dict):
     functions = {}
     defined_funcs = set()
@@ -28,16 +40,20 @@ def parse_functions_from_files(file_dict):
         for node in ast.walk(tree):
             if isinstance(node, ast.FunctionDef):
                 func_name = node.name
-                args, returns, calls = [], [], []
+                args, returns = [], []
                 local_assignments = {}
                 reads_state, writes_state = set(), set()
+
                 for arg in node.args.args:
                     arg_type = ast.unparse(arg.annotation) if arg.annotation else "?"
                     args.append(f"{arg.arg}: {arg_type}")
+
+                collector = CallCollector(defined_funcs)
+                collector.visit(node)
+                calls = collector.calls
+
                 for sub in ast.walk(node):
-                    if isinstance(sub, ast.Call) and isinstance(sub.func, ast.Name):
-                        if sub.func.id in defined_funcs: calls.append(sub.func.id)
-                    elif isinstance(sub, ast.Assign):
+                    if isinstance(sub, ast.Assign):
                         for target in sub.targets:
                             if isinstance(target, ast.Name):
                                 local_assignments[target.id] = infer_type_from_value(sub.value)
@@ -50,6 +66,7 @@ def parse_functions_from_files(file_dict):
                         else:
                             label = ast.unparse(sub.value)
                             returns.append(f"{label}: {local_assignments.get(label, infer_type_from_value(sub.value))}")
+
                 functions[func_name] = {
                     "args": args,
                     "returns": returns,
@@ -58,8 +75,11 @@ def parse_functions_from_files(file_dict):
                     "reads_state": sorted(reads_state),
                     "writes_state": sorted(writes_state)
                 }
+
     return functions
 
+
+
 def get_reachable_functions(start, graph):
     visited, stack = set(), [start]
     while stack: