number of vars

main.py

@@ -125,11 +125,11 @@ with gr.Blocks() as demo:
                 # print(fun_data)
 
                 addrs = [
-                    (f"{name} ({hex(int(addr))})", int(addr))
-                    for addr, (name, cf) in fun_data.items()
+                    (f"{name} ({hex(int(addr))}; {numvars} vars)", int(addr))
+                    for addr, (name, cf, numvars) in fun_data.items()
                 ]
 
-                cfs = {name: cf for (name, cf) in fun_data.values()}
+                cfs = {name: cf for (name, cf, _numvars) in fun_data.values()}
             except Exception as e:
                 raise gr.Error(f"Unable to analyze binary with Ghidra: {e}")
 

scripts/dump_functions.py

@@ -30,7 +30,7 @@ def dump_functions_to_json():
 
         # Add function name and address to the dictionary
         if example.is_valid_example:
-            functions_dict[func_address] = (func_name, cf.to_json())
+            functions_dict[func_address] = (func_name, cf.to_json(), len(example.source))
     
     # Convert the dictionary to a JSON object
     json_data = json.dumps(functions_dict, indent=4)