ultimate-gemini-mcp 1.0.15__py3-none-any.whl → 1.0.17__py3-none-any.whl

src/__init__.py

@@ -7,7 +7,7 @@ A unified MCP server that combines the best features from:
 - Advanced features: batch processing, editing, templates, and more
 """
 
-__version__ = "1.0.15"
+__version__ = "1.0.17"
 __author__ = "Ultimate Gemini MCP"
 
 from .config import get_settings

src/server.py

@@ -14,7 +14,11 @@ import sys
 from fastmcp import FastMCP
 
 from .config import ALL_MODELS, get_settings
-from .tools import register_batch_generate_tool, register_generate_image_tool
+from .tools import (
+    register_batch_generate_tool,
+    register_generate_image_tool,
+    register_get_image_tool,
+)
 
 # Set up logging
 logging.basicConfig(
@@ -51,6 +55,7 @@ def create_app() -> FastMCP:
         # Register tools
         register_generate_image_tool(mcp)
         register_batch_generate_tool(mcp)
+        register_get_image_tool(mcp)
 
         # Add resources
         @mcp.resource("models://list")
@@ -133,38 +138,6 @@ def create_app() -> FastMCP:
 
             return json.dumps(config, indent=2)
 
-        @mcp.tool()
-        async def get_image(filename: str) -> str:
-            """
-            Retrieve a generated image by filename and return as base64.
-
-            Args:
-                filename: The filename of the image (e.g., "gemini-2.5-flash-image_20251026_055415_a cute orange cat sleeping on a sunny windowsill.png")
-
-            Returns:
-                JSON string with base64-encoded image data
-            """
-            import base64
-            import json
-            from pathlib import Path
-
-            image_path = settings.output_dir / filename
-
-            if not image_path.exists():
-                return json.dumps({"success": False, "error": f"Image not found: {filename}"})
-
-            if not image_path.is_relative_to(settings.output_dir):
-                return json.dumps({"success": False, "error": "Access denied: path outside output directory"})
-
-            image_data = base64.b64encode(image_path.read_bytes()).decode()
-
-            return json.dumps({
-                "success": True,
-                "filename": filename,
-                "image_base64": image_data,
-                "size": len(image_path.read_bytes())
-            }, indent=2)
-
         logger.info("Ultimate Gemini MCP Server initialized successfully")
         return mcp
 

src/tools/__init__.py

@@ -1,11 +1,16 @@
 """Tools module for Ultimate Gemini MCP."""
 
 from .batch_generate import batch_generate_images, register_batch_generate_tool
-from .generate_image import generate_image_tool, register_generate_image_tool
+from .generate_image import (
+    generate_image_tool,
+    register_generate_image_tool,
+    register_get_image_tool,
+)
 
 __all__ = [
     "generate_image_tool",
     "register_generate_image_tool",
+    "register_get_image_tool",
     "batch_generate_images",
     "register_batch_generate_tool",
 ]

src/tools/generate_image.py

@@ -179,6 +179,47 @@ async def generate_image_tool(
         await image_service.close()
 
 
+def register_get_image_tool(mcp_server: Any) -> None:
+    """Register get_image tool with MCP server."""
+
+    @mcp_server.tool()
+    async def get_image(filename: str) -> str:
+        """
+        Retrieve a generated image by filename and return as base64.
+
+        Args:
+            filename: The filename of the image (e.g., "gemini-2.5-flash-image_20251026_055415_a cute orange cat sleeping on a sunny windowsill.png")
+
+        Returns:
+            JSON string with base64-encoded image data
+        """
+        import base64
+        import json
+
+        from ..config import get_settings
+
+        settings = get_settings()
+        image_path = settings.output_dir / filename
+
+        if not image_path.exists():
+            return json.dumps({"success": False, "error": f"Image not found: {filename}"})
+
+        try:
+            image_data = base64.b64encode(image_path.read_bytes()).decode()
+
+            return json.dumps(
+                {
+                    "success": True,
+                    "filename": filename,
+                    "image_base64": image_data,
+                    "size": image_path.stat().st_size,
+                },
+                indent=2,
+            )
+        except Exception as e:
+            return json.dumps({"success": False, "error": str(e)})
+
+
 def register_generate_image_tool(mcp_server: Any) -> None:
     """Register generate_image tool with MCP server."""
 

{ultimate_gemini_mcp-1.0.15.dist-info → ultimate_gemini_mcp-1.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultimate-gemini-mcp
-Version: 1.0.15
+Version: 1.0.17
 Summary: Ultimate image generation MCP server unifying Gemini 2.5 Flash Image and Imagen 4/Fast/Ultra with advanced features
 Project-URL: Homepage, https://github.com/anand-92/ultimate-image-gen-mcp
 Project-URL: Repository, https://github.com/anand-92/ultimate-image-gen-mcp

{ultimate_gemini_mcp-1.0.15.dist-info → ultimate_gemini_mcp-1.0.17.dist-info}/RECORD

@@ -1,5 +1,5 @@
-src/__init__.py,sha256=37W1Cj8KSa632L8H4PZLI55Hy2cndCTsAzv0lRAw-nE,436
-src/server.py,sha256=T_zx1x6rFVnonFgiS2LN1W3porjpl1KzPM-X6JHewAQ,7055
+src/__init__.py,sha256=9rQLRqhsB1_h6jbGeL4y3ShIhH1i4bnb9sE1tKaDtN8,436
+src/server.py,sha256=nKO88PU-8dmu1rzPmRFrTrd3xYxV6ab4ln-9Bb1g-L8,5941
 src/config/__init__.py,sha256=hL0recV_ycXBEGCym7BqwyaPCnQHy8o429pBirnBeiA,704
 src/config/constants.py,sha256=uFQJMk03vLb5YjkYMimeIz06WAVnqW3tuWQlJLYjHtc,1875
 src/config/settings.py,sha256=uTxxblnyqmglz1COd3QxjC2w3o6l51S9MTanP4HBrgE,4257
@@ -13,12 +13,12 @@ src/services/gemini_client.py,sha256=VPuBPDHhAag_8xBqzMuVIEWVG0BINXOHkwKIALzcFAk
 src/services/image_service.py,sha256=ovYsxDMwqsSiG1YbwBcTBrgmGTA7bdhDF6_jPs9XV_k,7095
 src/services/imagen_client.py,sha256=mvwbmtYQHqpKKM36tp8JPZTjxav5eeeist37lRJsv4k,6122
 src/services/prompt_enhancer.py,sha256=J_0s1EVmXdPAhjZPM4hL1vk9YiawqfH_ogF89VrFkW8,4776
-src/tools/__init__.py,sha256=zBfAjFT51LvvD7WXTfDYiyJstRdphr2ChddAmGMZxkI,346
+src/tools/__init__.py,sha256=idmgH7aLtnwROyqOf2-KlTutb4mmJsN-xC2ClLB1Z-Q,419
 src/tools/batch_generate.py,sha256=ESAUkHX5TOKPlHCzgY2jMLyAvSmInN2Jwm5-51EeWwg,5406
-src/tools/generate_image.py,sha256=M8jvqEdcPQpKgnMQqIhlCy7i9yO3hMVdsWR6l1pd8G8,9438
+src/tools/generate_image.py,sha256=vTwLY1v8lTZlwb9T7TKNcuYxgNpDDJKaB_NMrM_C0C0,10726
 src/utils/__init__.py,sha256=T2UShY0j2kiu63YPOsTACrzOfWTTq4PmC0flzcgeCl0,55
-ultimate_gemini_mcp-1.0.15.dist-info/METADATA,sha256=w7nd2jPIVtwvCaH5KgDyOt4iogJZ7hKN99Fe2Oaeols,16131
-ultimate_gemini_mcp-1.0.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ultimate_gemini_mcp-1.0.15.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
-ultimate_gemini_mcp-1.0.15.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
-ultimate_gemini_mcp-1.0.15.dist-info/RECORD,,
+ultimate_gemini_mcp-1.0.17.dist-info/METADATA,sha256=AD7ym2HwURwfP07yPdwBL9e_XGYtRHojADfuROwFjVY,16131
+ultimate_gemini_mcp-1.0.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ultimate_gemini_mcp-1.0.17.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
+ultimate_gemini_mcp-1.0.17.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
+ultimate_gemini_mcp-1.0.17.dist-info/RECORD,,