DictItem is not indexable

Author: lsarra-ax

src/axiomatic/axtract/axtract_report.py

@@ -247,8 +247,8 @@ def create_report(report_data: EquationExtractionResponse, report_path: str = ".
         for symbol in eq.latex_symbols:
             html_content += f"""
                         <tr>
-                            <td>\\({symbol["key"]}\\)</td>
-                            <td>{symbol["value"]}</td>
+                            <td>\\({symbol.key}\\)</td>
+                            <td>{symbol.key}</td>
                         </tr>
             """
 

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.key}
 
     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)