updated fixes

Author: tymekaxai

src/axiomatic/axtract/interactive_table.py

@@ -141,7 +141,6 @@ def save_custom_presets(custom_data, file_path):
     # 4) Load custom presets from JSON (if any) and integrate them
     # ---------------------------------------------------------------
     variable_dict = _create_variable_dict(loaded_equations)
-    print(variable_dict)
     custom_presets = load_custom_presets(file_path)
 
     for custom_name, values_dict in custom_presets.items():
@@ -151,7 +150,6 @@ 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_dict)
 
     # This dict will store the final user inputs after pressing "Submit"
     result = {}

src/axiomatic/client.py

@@ -42,19 +42,23 @@ def analyze_equations(
         file_path: Optional[str] = None,
         url_path: Optional[str] = None,
         parsed_paper: Optional[ParseResponse] = None,
-    ):        
+    ) -> Optional[EquationExtractionResponse]:        
         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")
-            file = requests.get(url_path)
-            response = self._ax_client.document.equation.from_pdf(document=file.content)
+            response = requests.get(url_path)
+            from io import BytesIO
+            pdf_content = BytesIO(response.content)
+            response = self._ax_client.document.equation.from_pdf(document=pdf_content)
 
         elif parsed_paper:
-            response = self._ax_client.document.equation.process(**parsed_paper.model_dump())
+            response = EquationExtractionResponse.model_validate(
+                self._ax_client.document.equation.process(**parsed_paper.model_dump()).model_dump()
+            )
 
         else:
             print("Please provide either a file path or a URL to analyze.")