errors fixed

Author: tymekaxai

src/axiomatic/__init__.py

@@ -94,7 +94,7 @@
     requirements,
     tools,
 )
-from .client import AsyncAxiomatic, Axiomatic, AxtractHelper
+from .client import AsyncAxiomatic, Axiomatic
 from .environment import AxiomaticEnvironment
 from .version import __version__
 
@@ -103,7 +103,6 @@
     "AxesInfo",
     "Axiomatic",
     "AxiomaticEnvironment",
-    "AxtractHelper",
     "Bundle",
     "BundleSettingsValue",
     "Computation",

src/axiomatic/axtract/interactive_table.py

@@ -151,7 +151,7 @@ def save_custom_presets(custom_data, file_path):
     # 5) For the "Add Requirement" dropdown
     # ---------------------------------------------------------------
     variable_names = [details["name"] for details in variable_dict.values()]
-
+    print(variable_names)
     # This dict will store the final user inputs after pressing "Submit"
     result = {}
 

src/axiomatic/client.py

@@ -2,7 +2,7 @@
 import requests
 import os
 import time
-from typing import Dict, List, Union, Optional
+from typing import Dict, List, Optional
 
 from .base_client import BaseClient, AsyncBaseClient
 from . import ParseResponse
@@ -17,6 +17,7 @@ def __init__(self, *args, **kwargs):
 
         self.document_helper = DocumentHelper(self)
         self.tools_helper = ToolsHelper(self)
+        self.axtract_helper = AxtractHelper(self)
 
 
 class AxtractHelper:
@@ -32,26 +33,21 @@ def create_report(self, response: EquationExtractionResponse, path: str):
 
         create_report(response, path)
 
-    def analyze_equations(
-        self,
-        file_path: Optional[str] = None,
-        url_path: Optional[str] = None
-    ) -> Optional[EquationExtractionResponse]:
+    def analyze_equations(self, file_path: Optional[str] = None, url_path: Optional[str] = None):
         if file_path:
             with open(file_path, "rb") as file:
                 response = self._ax_client.document.equation.from_pdf(document=file)
 
         elif url_path:
             if "arxiv" in url_path and "abs" in url_path:
                 url_path = url_path.replace("abs", "pdf")
-            
-            response = self._ax_client.document.equation.from_pdf(document=url_path)
+            file = requests.get(url_path)
+            response = self._ax_client.document.equation.from_pdf(document=file.content)
 
         else:
             print("Please provide either a file path or a URL to analyze.")
             return None
-
-        return EquationExtractionResponse(equations=response.equations)
+        return response
 
     def validate_equations(
         self,
@@ -65,12 +61,9 @@ def validate_equations(
 
         api_requirements = [
             ApiVariableRequirement(
-                symbol=req.symbol,
-                name=req.name,
-                value=req.value,
-                units=req.units,
-                tolerance=req.tolerance
-            ) for req in requirements
+                symbol=req.symbol, name=req.name, value=req.value, units=req.units, tolerance=req.tolerance
+            )
+            for req in requirements
         ]
 
         variable_dict = _create_variable_dict(loaded_equations)