scitex 2.17.0__py3-none-any.whl → 2.17.4__py3-none-any.whl

scitex/plt/__init__.py

@@ -75,8 +75,8 @@ if _FIGRECIPE_AVAILABLE:
     from figrecipe import (
         compose,
         crop,
-        edit,
         extract_data,
+        gui,
         info,
         list_presets,
         load_style,
@@ -87,11 +87,17 @@ if _FIGRECIPE_AVAILABLE:
         validate,
     )
 
+    # Backward compatibility alias
+    edit = gui
+
     # Internal imports (not part of figrecipe public API)
     from figrecipe._api._notebook import enable_svg
     from figrecipe._api._seaborn_proxy import sns
     from figrecipe._api._style_manager import STYLE, apply_style
-    from figrecipe._composition import align_panels, distribute_panels, smart_align
+    from figrecipe._composition import align_panels, align_smart, distribute_panels
+
+    # Backward compatibility alias
+    smart_align = align_smart
     from figrecipe._graph_presets import get_preset as get_graph_preset
     from figrecipe._graph_presets import list_presets as list_graph_presets
     from figrecipe._graph_presets import register_preset as register_graph_preset
@@ -120,11 +126,13 @@ else:
     validate = _not_available
     extract_data = _not_available
     info = _not_available
-    edit = _not_available
+    gui = _not_available
+    edit = _not_available  # Backward compatibility alias
     compose = _not_available
     align_panels = _not_available
     distribute_panels = _not_available
-    smart_align = _not_available
+    align_smart = _not_available
+    smart_align = _not_available  # Backward compatibility alias
     sns = None
     enable_svg = _not_available
     get_graph_preset = _not_available
@@ -438,7 +446,8 @@ __all__ = [
     "validate",
     "extract_data",
     "info",
-    "edit",
+    "gui",
+    "edit",  # Backward compatibility alias for gui
     # Style management
     "STYLE",
     "load_style",
@@ -449,7 +458,8 @@ __all__ = [
     "compose",
     "align_panels",
     "distribute_panels",
-    "smart_align",
+    "align_smart",
+    "smart_align",  # Backward compatibility alias for align_smart
     # Graph visualization
     "draw_graph",
     "get_graph_preset",

scitex/scholar/_mcp/crossref_handlers.py

@@ -40,6 +40,8 @@ async def crossref_search_handler(
     year_min: int | None = None,
     year_max: int | None = None,
     enrich: bool = False,
+    save_path: str | None = None,
+    save_format: str = "json",
 ) -> dict:
     """Search CrossRef database (167M+ papers) via crossref-local.
 
@@ -50,6 +52,8 @@ async def crossref_search_handler(
         year_min: Minimum publication year filter
         year_max: Maximum publication year filter
         enrich: If True, add citation counts and references
+        save_path: Optional file path to save results (e.g., "results.json", "papers.bib")
+        save_format: Output format for save_path: "text", "json", or "bibtex" (default: "json")
     """
     try:
         crossref = _ensure_crossref()
@@ -91,11 +95,21 @@ async def crossref_search_handler(
                 if len(papers) >= limit:
                     break
 
-            return papers, results.total
+            return papers, results.total, results
 
-        papers, total = await loop.run_in_executor(None, do_search)
+        papers, total, search_results = await loop.run_in_executor(None, do_search)
 
-        return {
+        # Save to file if requested
+        saved_path = None
+        if save_path:
+            try:
+                from crossref_local import save as cr_save
+
+                saved_path = cr_save(search_results, save_path, format=save_format)
+            except Exception as e:
+                return {"success": False, "error": f"Failed to save: {e}"}
+
+        result = {
             "success": True,
             "query": query,
             "total": total,
@@ -107,6 +121,11 @@ async def crossref_search_handler(
             "timestamp": datetime.now().isoformat(),
         }
 
+        if saved_path:
+            result["saved_to"] = saved_path
+
+        return result
+
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -115,6 +134,8 @@ async def crossref_get_handler(
     doi: str,
     include_citations: bool = False,
     include_references: bool = False,
+    save_path: str | None = None,
+    save_format: str = "json",
 ) -> dict:
     """Get a paper by DOI from CrossRef database.
 
@@ -122,6 +143,8 @@ async def crossref_get_handler(
         doi: DOI of the paper
         include_citations: Include list of citing DOIs
         include_references: Include list of referenced DOIs
+        save_path: Optional file path to save result (e.g., "paper.json", "paper.bib")
+        save_format: Output format for save_path: "text", "json", or "bibtex" (default: "json")
     """
     try:
         crossref = _ensure_crossref()
@@ -130,7 +153,7 @@ async def crossref_get_handler(
         def do_get():
             work = crossref.get(doi)
             if not work:
-                return None
+                return None, None
 
             result = {
                 "doi": work.doi,
@@ -152,9 +175,9 @@ async def crossref_get_handler(
             if include_references:
                 result["referenced_dois"] = crossref.get_cited(doi)
 
-            return result
+            return result, work
 
-        result = await loop.run_in_executor(None, do_get)
+        result, work_obj = await loop.run_in_executor(None, do_get)
 
         if result is None:
             return {
@@ -163,13 +186,28 @@ async def crossref_get_handler(
                 "doi": doi,
             }
 
-        return {
+        # Save to file if requested
+        saved_path = None
+        if save_path and work_obj:
+            try:
+                from crossref_local import save as cr_save
+
+                saved_path = cr_save(work_obj, save_path, format=save_format)
+            except Exception as e:
+                return {"success": False, "error": f"Failed to save: {e}"}
+
+        response = {
             "success": True,
             "paper": result,
             "source": "crossref_local",
             "timestamp": datetime.now().isoformat(),
         }
 
+        if saved_path:
+            response["saved_to"] = saved_path
+
+        return response
+
     except Exception as e:
         return {"success": False, "error": str(e)}
 

scitex/scholar/_mcp/openalex_handlers.py

@@ -38,6 +38,8 @@ async def openalex_search_handler(
     offset: int = 0,
     year_min: int | None = None,
     year_max: int | None = None,
+    save_path: str | None = None,
+    save_format: str = "json",
 ) -> dict:
     """Search OpenAlex database (250M+ papers) via openalex-local.
 
@@ -47,6 +49,8 @@ async def openalex_search_handler(
         offset: Number of results to skip for pagination
         year_min: Minimum publication year filter
         year_max: Maximum publication year filter
+        save_path: Optional file path to save results (e.g., "results.json", "papers.bib")
+        save_format: Output format for save_path: "text", "json", or "bibtex" (default: "json")
     """
     try:
         openalex = _ensure_openalex()
@@ -86,11 +90,21 @@ async def openalex_search_handler(
                 if len(papers) >= limit:
                     break
 
-            return papers, getattr(results, "total", len(papers))
+            return papers, getattr(results, "total", len(papers)), results
 
-        papers, total = await loop.run_in_executor(None, do_search)
+        papers, total, search_results = await loop.run_in_executor(None, do_search)
 
-        return {
+        # Save to file if requested
+        saved_path = None
+        if save_path:
+            try:
+                from openalex_local import save as oa_save
+
+                saved_path = oa_save(search_results, save_path, format=save_format)
+            except Exception as e:
+                return {"success": False, "error": f"Failed to save: {e}"}
+
+        result = {
             "success": True,
             "query": query,
             "total": total,
@@ -102,6 +116,11 @@ async def openalex_search_handler(
             "timestamp": datetime.now().isoformat(),
         }
 
+        if saved_path:
+            result["saved_to"] = saved_path
+
+        return result
+
     except Exception as e:
         return {"success": False, "error": str(e)}
 
@@ -109,12 +128,16 @@ async def openalex_search_handler(
 async def openalex_get_handler(
     doi: str = None,
     openalex_id: str = None,
+    save_path: str | None = None,
+    save_format: str = "json",
 ) -> dict:
     """Get a paper by DOI or OpenAlex ID from OpenAlex database.
 
     Args:
         doi: DOI of the paper (e.g., '10.1038/nature12373')
         openalex_id: OpenAlex ID (e.g., 'W2100837269')
+        save_path: Optional file path to save result (e.g., "paper.json", "paper.bib")
+        save_format: Output format for save_path: "text", "json", or "bibtex" (default: "json")
     """
     if not doi and not openalex_id:
         return {"success": False, "error": "Must provide either doi or openalex_id"}
@@ -127,7 +150,7 @@ async def openalex_get_handler(
             identifier = doi or openalex_id
             work = openalex.get(identifier)
             if not work:
-                return None
+                return None, None
 
             result = {
                 "doi": work.doi,
@@ -144,9 +167,9 @@ async def openalex_get_handler(
                 "url": getattr(work, "url", None),
             }
 
-            return result
+            return result, work
 
-        result = await loop.run_in_executor(None, do_get)
+        result, work_obj = await loop.run_in_executor(None, do_get)
 
         if result is None:
             identifier = doi or openalex_id
@@ -156,13 +179,28 @@ async def openalex_get_handler(
                 "identifier": identifier,
             }
 
-        return {
+        # Save to file if requested
+        saved_path = None
+        if save_path and work_obj:
+            try:
+                from openalex_local import save as oa_save
+
+                saved_path = oa_save(work_obj, save_path, format=save_format)
+            except Exception as e:
+                return {"success": False, "error": f"Failed to save: {e}"}
+
+        response = {
             "success": True,
             "paper": result,
             "source": "openalex_local",
             "timestamp": datetime.now().isoformat(),
         }
 
+        if saved_path:
+            response["saved_to"] = saved_path
+
+        return response
+
     except Exception as e:
         return {"success": False, "error": str(e)}
 

scitex/scholar/config/default.yaml

@@ -16,6 +16,7 @@ enable_auto_download: ${SCITEX_SCHOLAR_AUTO_DOWNLOAD:-false}
 engines:
   - URL
   - CrossRefLocal
+  - OpenAlexLocal
   - Semantic_Scholar
   - CrossRef
   - OpenAlex
@@ -110,6 +111,7 @@ unpaywall_email: ${SCITEX_SCHOLAR_UNPAYWALL_EMAIL:-"research@scitex.io"}
 semantic_scholar_api_key: ${SCITEX_SCHOLAR_SEMANTIC_SCHOLAR_API_KEY:-null}
 crossref_api_key: ${SCITEX_SCHOLAR_CROSSREF_API_KEY:-null}
 crossref_api_url: ${SCITEX_SCHOLAR_CROSSREF_API_URL:-"http://127.0.0.1:3333"}
+openalex_api_url: ${SCITEX_SCHOLAR_OPENALEX_API_URL:-"http://127.0.0.1:31292"}
 pubmed_api_key: ${SCITEX_SCHOLAR_PUBMED_API_KEY:-null}
 twocaptcha_api_key: ${SCITEX_SCHOLAR_2CAPTCHA_API_KEY:-null}
 

scitex/scholar/local_dbs/__init__.py

@@ -20,11 +20,15 @@ Usage:
 
 from __future__ import annotations
 
-from . import crossref_scitex, openalex_scitex
+from . import crossref_scitex, openalex_scitex, unified
+from .export import SUPPORTED_FORMATS, save
 
 __all__ = [
     "crossref_scitex",
     "openalex_scitex",
+    "unified",
+    "save",
+    "SUPPORTED_FORMATS",
 ]
 
 

scitex/scholar/local_dbs/export.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+# Timestamp: 2026-02-04
+# File: src/scitex/scholar/local_dbs/export.py
+"""Export functionality for unified local database results.
+
+Supports multiple output formats:
+- text: Human-readable formatted text
+- json: JSON format with all fields
+- bibtex: BibTeX bibliography format
+"""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import TYPE_CHECKING, List, Union
+
+if TYPE_CHECKING:
+    from .unified import UnifiedSearchResult, UnifiedWork
+
+__all__ = [
+    "save",
+    "SUPPORTED_FORMATS",
+]
+
+SUPPORTED_FORMATS = ["text", "json", "bibtex"]
+
+
+def save(
+    data: Union[UnifiedWork, UnifiedSearchResult, List[UnifiedWork]],
+    path: str,
+    format: str = "json",
+) -> str:
+    """Save UnifiedWork(s) or UnifiedSearchResult to a file.
+
+    Args:
+        data: UnifiedWork, UnifiedSearchResult, or list of UnifiedWorks
+        path: Output file path
+        format: Output format ("text", "json", "bibtex")
+
+    Returns
+    -------
+        Path to saved file
+
+    Raises
+    ------
+        ValueError: If format is not supported
+
+    Examples
+    --------
+        >>> from scitex.scholar.local_dbs import search, save
+        >>> results = search("machine learning", limit=10)
+        >>> save(results, "results.json")
+        >>> save(results, "results.bib", format="bibtex")
+        >>> save(results, "results.txt", format="text")
+    """
+    from .unified import UnifiedSearchResult, UnifiedWork, to_bibtex, to_json, to_text
+
+    if format not in SUPPORTED_FORMATS:
+        raise ValueError(
+            f"Unsupported format: {format}. "
+            f"Supported formats: {', '.join(SUPPORTED_FORMATS)}"
+        )
+
+    path = Path(path)
+
+    # Extract works
+    if isinstance(data, UnifiedWork):
+        works = [data]
+    elif isinstance(data, UnifiedSearchResult):
+        works = data.works
+    elif isinstance(data, list):
+        works = data
+    else:
+        raise TypeError(f"Unsupported data type: {type(data)}")
+
+    # Generate content
+    if format == "text":
+        content = to_text(works)
+    elif format == "json":
+        content = to_json(works)
+    elif format == "bibtex":
+        content = to_bibtex(works)
+    else:
+        raise ValueError(f"Unsupported format: {format}")
+
+    # Write to file
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(content, encoding="utf-8")
+
+    return str(path)
+
+
+# EOF