scitex 2.11.0__py3-none-any.whl → 2.13.0__py3-none-any.whl

scitex/plt/mcp_server.py

@@ -0,0 +1,205 @@
+#!/usr/bin/env python3
+# Timestamp: 2026-01-08
+# File: src/scitex/plt/mcp_server.py
+# ----------------------------------------
+
+"""
+MCP Server for SciTeX plt - Publication-quality plotting.
+
+Provides tools for:
+- Style configuration (get/set publication styles)
+- Style presets (journal-specific configurations)
+- Figure cropping (auto-crop whitespace)
+- DPI management
+- Color palette access
+"""
+
+from __future__ import annotations
+
+import asyncio
+
+# Graceful MCP dependency handling
+try:
+    import mcp.types as types
+    from mcp.server import NotificationOptions, Server
+    from mcp.server.models import InitializationOptions
+    from mcp.server.stdio import stdio_server
+
+    MCP_AVAILABLE = True
+except ImportError:
+    MCP_AVAILABLE = False
+    types = None  # type: ignore
+    Server = None  # type: ignore
+    NotificationOptions = None  # type: ignore
+    InitializationOptions = None  # type: ignore
+    stdio_server = None  # type: ignore
+
+__all__ = ["PltServer", "main", "MCP_AVAILABLE"]
+
+
+class PltServer:
+    """MCP Server for Publication-quality Plotting."""
+
+    def __init__(self):
+        self.server = Server("scitex-plt")
+        self.setup_handlers()
+
+    def setup_handlers(self):
+        """Set up MCP server handlers."""
+        from ._mcp_handlers import (
+            crop_figure_handler,
+            get_color_palette_handler,
+            get_dpi_settings_handler,
+            get_style_handler,
+            list_presets_handler,
+            set_style_handler,
+        )
+        from ._mcp_tool_schemas import get_tool_schemas
+
+        @self.server.list_tools()
+        async def handle_list_tools():
+            return get_tool_schemas()
+
+        @self.server.call_tool()
+        async def handle_call_tool(name: str, arguments: dict):
+            if name == "get_style":
+                return await self._wrap_result(get_style_handler())
+
+            elif name == "set_style":
+                return await self._wrap_result(set_style_handler(**arguments))
+
+            elif name == "list_presets":
+                return await self._wrap_result(list_presets_handler())
+
+            elif name == "crop_figure":
+                return await self._wrap_result(crop_figure_handler(**arguments))
+
+            elif name == "get_dpi_settings":
+                return await self._wrap_result(get_dpi_settings_handler())
+
+            elif name == "get_color_palette":
+                return await self._wrap_result(get_color_palette_handler(**arguments))
+
+            else:
+                raise ValueError(f"Unknown tool: {name}")
+
+        @self.server.list_resources()
+        async def handle_list_resources():
+            """List available plt resources."""
+            resources = [
+                types.Resource(
+                    uri="plt://style/current",
+                    name="Current Style",
+                    description="Current publication style configuration",
+                    mimeType="application/json",
+                ),
+                types.Resource(
+                    uri="plt://presets",
+                    name="Style Presets",
+                    description="Available journal style presets",
+                    mimeType="application/json",
+                ),
+                types.Resource(
+                    uri="plt://colors",
+                    name="Color Palette",
+                    description="SciTeX color palette",
+                    mimeType="application/json",
+                ),
+            ]
+            return resources
+
+        @self.server.read_resource()
+        async def handle_read_resource(uri: str):
+            """Read a plt resource."""
+            import json
+
+            if uri == "plt://style/current":
+                result = await get_style_handler()
+                return types.TextResourceContents(
+                    uri=uri,
+                    mimeType="application/json",
+                    text=json.dumps(result, indent=2),
+                )
+
+            elif uri == "plt://presets":
+                result = await list_presets_handler()
+                return types.TextResourceContents(
+                    uri=uri,
+                    mimeType="application/json",
+                    text=json.dumps(result, indent=2),
+                )
+
+            elif uri == "plt://colors":
+                result = await get_color_palette_handler()
+                return types.TextResourceContents(
+                    uri=uri,
+                    mimeType="application/json",
+                    text=json.dumps(result, indent=2),
+                )
+
+            else:
+                raise ValueError(f"Unknown resource URI: {uri}")
+
+    async def _wrap_result(self, coro):
+        """Wrap handler result as MCP TextContent."""
+        import json
+
+        try:
+            result = await coro
+            return [
+                types.TextContent(
+                    type="text",
+                    text=json.dumps(result, indent=2, default=str),
+                )
+            ]
+        except Exception as e:
+            return [
+                types.TextContent(
+                    type="text",
+                    text=json.dumps({"success": False, "error": str(e)}, indent=2),
+                )
+            ]
+
+
+async def _run_server():
+    """Run the MCP server (internal)."""
+    server = PltServer()
+    async with stdio_server() as (read_stream, write_stream):
+        await server.server.run(
+            read_stream,
+            write_stream,
+            InitializationOptions(
+                server_name="scitex-plt",
+                server_version="0.1.0",
+                capabilities=server.server.get_capabilities(
+                    notification_options=NotificationOptions(),
+                    experimental_capabilities={},
+                ),
+            ),
+        )
+
+
+def main():
+    """Main entry point for the MCP server."""
+    if not MCP_AVAILABLE:
+        import sys
+
+        print("=" * 60)
+        print("MCP Server 'scitex-plt' requires the 'mcp' package.")
+        print()
+        print("Install with:")
+        print("  pip install mcp")
+        print()
+        print("Or install scitex with MCP support:")
+        print("  pip install scitex[mcp]")
+        print("=" * 60)
+        sys.exit(1)
+
+    asyncio.run(_run_server())
+
+
+if __name__ == "__main__":
+    main()
+
+
+# EOF

scitex/repro/README_RandomStateManager.md

@@ -28,7 +28,7 @@ import torch
 import tensorflow as tf
 
 # Initialize once - fixes everything!
-rng_manager = stx.rng.RandomStateManager(seed=42)
+rng = stx.rng.RandomStateManager(seed=42)
 
 # Python random - automatically fixed
 python_val = random.random()  # Reproducible!
@@ -87,7 +87,7 @@ import sys
 import scitex as stx
 
 # Session.start returns RNG, fixing random seeds of the supported libraries automatically
-CONFIG, stdout, stderr, plt, CC, rng_manager = stx.session.start(
+CONFIG, stdout, stderr, plt, CC, rng = stx.session.start(
     sys=sys,
     seed=42
 )
@@ -208,4 +208,4 @@ When you create a RandomStateManager, it automatically:
 
 yusuke.watanabe@scitex.ai
 
-<!-- EOF -->
+<!-- EOF -->

scitex/repro/_RandomStateManager.py

@@ -14,7 +14,7 @@ __DIR__ = os.path.dirname(__FILE__)
 Clean, simple RandomStateManager for scientific reproducibility.
 
 Main API:
-    rng_manager = RandomStateManager(seed=42)   # Create instance
+    rng = RandomStateManager(seed=42)   # Create instance
     gen = rng("name")                   # Get named generator
     rng.verify(obj, "name")             # Verify reproducibility
 """
@@ -44,11 +44,11 @@ class RandomStateManager:
     >>> import scitex as stx
     >>>
     >>> # Method 1: Direct usage
-    >>> rng_manager = stx.rng.RandomStateManager(seed=42)
+    >>> rng = stx.rng.RandomStateManager(seed=42)
     >>> data = rng("data").random(100)
     >>>
     >>> # Method 2: From session.start
-    >>> CONFIG, stdout, stderr, plt, CC, rng_manager = stx.session.start(seed=42)
+    >>> CONFIG, stdout, stderr, plt, CC, rng = stx.session.start(seed=42)
     >>> model = rng("model").normal(size=(10, 10))
     >>>
     >>> # Verify reproducibility
@@ -98,7 +98,7 @@ class RandomStateManager:
             np.random.seed(self.seed)
             # Also set default_rng for new API
             self._np = np
-            self._np_default_rng_manager = np.random.default_rng(self.seed)
+            self._np_default_rng = np.random.default_rng(self.seed)
             fixed_modules.append("numpy")
         except ImportError:
             self._np = None
@@ -159,7 +159,7 @@ class RandomStateManager:
 
         Examples
         --------
-        >>> rng_manager = RandomStateManager(42)
+        >>> rng = RandomStateManager(42)
         >>> gen = rng.get_np_generator("data")
         >>> values = gen.random(100)
         >>> perm = gen.permutation(100)
@@ -427,7 +427,7 @@ class RandomStateManager:
 
         Examples
         --------
-        >>> rng_manager = RandomStateManager(42)
+        >>> rng = RandomStateManager(42)
         >>> from sklearn.model_selection import train_test_split
         >>> X_train, X_test = train_test_split(
         ...     X, test_size=0.2,
@@ -455,7 +455,7 @@ class RandomStateManager:
 
         Examples
         --------
-        >>> rng_manager = RandomStateManager(42)
+        >>> rng = RandomStateManager(42)
         >>> gen = rng.get_torch_generator("model")
         >>> torch.randn(5, 5, generator=gen)
         """
@@ -503,7 +503,7 @@ class RandomStateManager:
 
         Examples
         --------
-        >>> rng_manager = RandomStateManager(42)
+        >>> rng = RandomStateManager(42)
         >>> rng.clear_cache()  # Clear all
         >>> rng.clear_cache("old_data")  # Clear specific
         >>> rng.clear_cache(["test1", "test2"])  # Clear multiple
@@ -559,7 +559,7 @@ def get(verbose: bool = False) -> RandomStateManager:
     Examples
     --------
     >>> import scitex as stx
-    >>> rng_manager = stx.rng.get()
+    >>> rng = stx.rng.get()
     >>> data = rng("data").random(100)
     """
     global _GLOBAL_INSTANCE
@@ -589,7 +589,7 @@ def reset(seed: int = 42, verbose: bool = False) -> RandomStateManager:
     Examples
     --------
     >>> import scitex as stx
-    >>> rng_manager = stx.repro.reset(seed=123)
+    >>> rng = stx.repro.reset(seed=123)
     """
     global _GLOBAL_INSTANCE
     _GLOBAL_INSTANCE = RandomStateManager(seed, verbose=verbose)
@@ -626,8 +626,8 @@ def main(args):
     print(f"Seed: {args.seed}")
 
     # Get named generators
-    data_gen = rng_manager("data")
-    model_gen = rng_manager("model")
+    data_gen = rng("data")
+    model_gen = rng("model")
 
     # Generate data
     print(f"\n{'Data Generation':-^60}")
@@ -641,7 +641,7 @@ def main(args):
 
     # Verify reproducibility
     print(f"\n{'Verification':-^60}")
-    rng_manager.verify(data, "demo_data")
+    rng.verify(data, "demo_data")
     print("✓ Data reproducibility verified")
 
     print(f"\n{'=' * 60}")
@@ -660,7 +660,7 @@ if __name__ == "__main__":
 
     args = parse_args()
 
-    CONFIG, sys.stdout, sys.stderr, plt, CC, rng_manager = stx.session.start(
+    CONFIG, sys.stdout, sys.stderr, plt, CC, rng = stx.session.start(
         sys,
         plt,
         args=args,

scitex/repro/_gen_ID.py

@@ -114,7 +114,7 @@ if __name__ == "__main__":
 
     args = parse_args()
 
-    CONFIG, sys.stdout, sys.stderr, plt, CC, rng_manager = stx.session.start(
+    CONFIG, sys.stdout, sys.stderr, plt, CC, rng = stx.session.start(
         sys,
         plt,
         args=args,

scitex/repro/_gen_timestamp.py

@@ -91,7 +91,7 @@ if __name__ == "__main__":
 
     args = parse_args()
 
-    CONFIG, sys.stdout, sys.stderr, plt, CC, rng_manager = stx.session.start(
+    CONFIG, sys.stdout, sys.stderr, plt, CC, rng = stx.session.start(
         sys,
         plt,
         args=args,

scitex/repro/_hash_array.py

@@ -72,8 +72,8 @@ def main(args):
     print(f"Array size: {args.size}")
     print(f"Seed: {args.seed}")
 
-    # Generate arrays using rng_manager
-    gen = rng_manager("demo")
+    # Generate arrays using rng
+    gen = rng("demo")
 
     # Create array and hash it
     print(f"\n{'Hash Generation':-^60}")
@@ -95,7 +95,7 @@ def main(args):
 
     # Reset generator and create same data
     print(f"\n{'Reproducibility Check':-^60}")
-    gen_repro = rng_manager("demo")  # Same name = same seed
+    gen_repro = rng("demo")  # Same name = same seed
     data3 = gen_repro.random(args.size)
     hash3 = hash_array(data3)
     print(f"Array 3 hash (reproduced): {hash3}")
@@ -115,7 +115,7 @@ if __name__ == "__main__":
 
     args = parse_args()
 
-    CONFIG, sys.stdout, sys.stderr, plt, CC, rng_manager = stx.session.start(
+    CONFIG, sys.stdout, sys.stderr, plt, CC, rng = stx.session.start(
         sys,
         plt,
         args=args,

scitex/scholar/__main__.py

@@ -1,5 +1,4 @@
 #!/usr/bin/env python3
-# -*- coding: utf-8 -*-
 # File: /home/ywatanabe/proj/scitex_repo/src/scitex/scholar/__main__.py
 
 """Scholar CLI entry point - Subcommand-based interface.
@@ -8,6 +7,7 @@ Clean interface routing to battle-tested pipeline implementations:
 - single: Process single paper (DOI or title)
 - parallel: Process multiple papers in parallel
 - bibtex: Process papers from BibTeX file
+- mcp: Start MCP server for LLM integration
 """
 
 from __future__ import annotations
@@ -33,6 +33,7 @@ Clean subcommand interface to battle-tested pipelines:
   single   - Process a single paper (DOI or title)
   parallel - Process multiple papers in parallel
   bibtex   - Process papers from BibTeX file
+  mcp      - Start MCP server for LLM integration
 
 STORAGE: ~/.scitex/scholar/library/
   MASTER/{8DIGITID}/  - Centralized storage (no duplicates)
@@ -193,6 +194,16 @@ STORAGE: ~/.scitex/scholar/library/
         help="Base Chrome profile to sync from (default: system)",
     )
 
+    # ========================================
+    # Subcommand: mcp
+    # ========================================
+    subparsers.add_parser(
+        "mcp",
+        help="Start MCP server for LLM integration",
+        description="Start the MCP (Model Context Protocol) server for Claude/LLM integration",
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+    )
+
     return parser
 
 
@@ -220,7 +231,7 @@ async def run_single_pipeline(args):
         force=args.force,
     )
 
-    logger.success(f"Single paper pipeline completed")
+    logger.success("Single paper pipeline completed")
     return 0
 
 
@@ -288,6 +299,15 @@ async def run_bibtex_pipeline(args):
     return 0
 
 
+async def run_mcp_server():
+    """Run MCP server."""
+    from .mcp_server import main as mcp_main
+
+    logger.info("Starting Scholar MCP server...")
+    await mcp_main()
+    return 0
+
+
 async def main_async():
     """Main async entry point."""
     parser = create_parser()
@@ -300,6 +320,8 @@ async def main_async():
         return await run_parallel_pipeline(args)
     elif args.command == "bibtex":
         return await run_bibtex_pipeline(args)
+    elif args.command == "mcp":
+        return await run_mcp_server()
     else:
         logger.error(f"Unknown command: {args.command}")
         return 1