ultimate-gemini-mcp 1.0.16__py3-none-any.whl → 1.0.18__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.16"
+__version__ = "1.0.18"
 __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,8 +55,32 @@ 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("image://latest", mime_type="image/png")
+        def get_latest_image() -> bytes:
+            """
+            Get the most recently generated image.
+
+            Returns:
+                Binary PNG image data
+            """
+            if not settings.output_dir.exists():
+                raise FileNotFoundError("No images have been generated yet")
+
+            # Find most recent PNG file
+            images = sorted(
+                settings.output_dir.glob("*.png"),
+                key=lambda p: p.stat().st_mtime,
+                reverse=True
+            )
+
+            if not images:
+                raise FileNotFoundError("No images found in output directory")
+
+            return images[0].read_bytes()
+
         @mcp.resource("models://list")
         def list_models() -> str:
             """List all available image generation models."""
@@ -133,47 +161,6 @@ def create_app() -> FastMCP:
 
             return json.dumps(config, indent=2)
 
-        @mcp.resource("images://list")
-        def list_images() -> str:
-            """List all generated images in the output directory."""
-            import json
-            from pathlib import Path
-
-            image_files = []
-            if settings.output_dir.exists():
-                for img in sorted(settings.output_dir.glob("*.png"), key=lambda p: p.stat().st_mtime, reverse=True):
-                    image_files.append({
-                        "filename": img.name,
-                        "size": img.stat().st_size,
-                        "modified": img.stat().st_mtime,
-                        "uri": f"image://{img.name}"
-                    })
-
-            return json.dumps({"images": image_files}, indent=2)
-
-        @mcp.resource("image://{filename}")
-        def get_image(filename: str) -> bytes:
-            """
-            Get a generated image by filename.
-
-            Args:
-                filename: The filename of the image
-
-            Returns:
-                Image bytes (PNG format)
-            """
-            from pathlib import Path
-
-            image_path = settings.output_dir / filename
-
-            if not image_path.exists():
-                raise FileNotFoundError(f"Image not found: {filename}")
-
-            if not image_path.is_relative_to(settings.output_dir):
-                raise ValueError("Access denied: path outside output directory")
-
-            return image_path.read_bytes()
-
         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.16.dist-info → ultimate_gemini_mcp-1.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultimate-gemini-mcp
-Version: 1.0.16
+Version: 1.0.18
 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.16.dist-info → ultimate_gemini_mcp-1.0.18.dist-info}/RECORD

@@ -1,5 +1,5 @@
-src/__init__.py,sha256=tp0HzLV4X_y-gtvSrYfK9OPkED-ZLzFy6cNMwROpyFE,436
-src/server.py,sha256=Nw-OCalYEPo0EmWJpDot8IKux5p50LbFS0hEYjWlCyY,7301
+src/__init__.py,sha256=rh3gu_KdsPjCNLWjsGKuZSr20I1-vugKmz1xv-wVZiM,436
+src/server.py,sha256=mpzCQBE3yUqPc9gDJNvoxR2XKrN56w3yKm2l0S-dBB0,6678
 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.16.dist-info/METADATA,sha256=wcjXuoJtH06tV9_1qrBewfDdOWVU92CdeLyHZlHqOys,16131
-ultimate_gemini_mcp-1.0.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ultimate_gemini_mcp-1.0.16.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
-ultimate_gemini_mcp-1.0.16.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
-ultimate_gemini_mcp-1.0.16.dist-info/RECORD,,
+ultimate_gemini_mcp-1.0.18.dist-info/METADATA,sha256=uoAjdHpu3zNfUfbqpV3JCMt5t7zofS1ffrT9OVy4KW8,16131
+ultimate_gemini_mcp-1.0.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ultimate_gemini_mcp-1.0.18.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
+ultimate_gemini_mcp-1.0.18.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
+ultimate_gemini_mcp-1.0.18.dist-info/RECORD,,