axtract symbol key dict

Author: lsarra-ax

src/axiomatic/axtract/interactive_table.py

@@ -408,7 +408,7 @@ def _create_variable_dict(equation_response: EquationExtractionResponse) -> dict
     for equation in equation_response.equations:
         for symbol in equation.latex_symbols:
             # Only add if not already present (avoid duplicates)
-            if symbol.key not in variable_dict:
-                variable_dict[symbol.key] = {"name": symbol.value}
+            if symbol['key'] not in variable_dict:
+                variable_dict[symbol['key']] = {"name": symbol['value']}
 
     return variable_dict

src/axiomatic/axtract/relation_graph.py

@@ -58,7 +58,7 @@ def generate_relation_graph(equations: List[EquationExtraction]) -> str:
         # Add variable nodes and edges
         for symbol in eq.latex_symbols:
             # Normalize the variable name
-            var_name = normalize_latex_symbol(symbol.key)
+            var_name = normalize_latex_symbol(symbol["key"])
             if var_name not in all_variables:
                 net.add_node(
                     var_name,
@@ -67,7 +67,7 @@ def generate_relation_graph(equations: List[EquationExtraction]) -> str:
                     shape="dot",
                     size=20,
                     font=dict(size=16, color="#000000"),
-                    title=f"{var_name}: {symbol.value}",
+                    title=f"{var_name}: {symbol['value']}",
                 )
                 all_variables.add(var_name)