ultimate-gemini-mcp 1.0.18__py3-none-any.whl → 1.5.1__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.18"
+__version__ = "1.5.1"
 __author__ = "Ultimate Gemini MCP"
 
 from .config import get_settings

src/config/constants.py

@@ -74,4 +74,4 @@ ENHANCEMENT_TIMEOUT = 30
 BATCH_TIMEOUT = 120
 
 # Output settings
-DEFAULT_OUTPUT_DIR = "/tmp/generated_images"
+DEFAULT_OUTPUT_DIR = "generated_images"

src/server.py

@@ -14,11 +14,7 @@ import sys
 from fastmcp import FastMCP
 
 from .config import ALL_MODELS, get_settings
-from .tools import (
-    register_batch_generate_tool,
-    register_generate_image_tool,
-    register_get_image_tool,
-)
+from .tools import register_batch_generate_tool, register_generate_image_tool
 
 # Set up logging
 logging.basicConfig(
@@ -49,38 +45,14 @@ def create_app() -> FastMCP:
         # Create FastMCP server
         mcp = FastMCP(
             "Ultimate Gemini MCP",
-            version="1.0.0",
+            version="1.5.0",
         )
 
         # 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."""

src/tools/__init__.py

@@ -1,16 +1,11 @@
 """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,
-    register_get_image_tool,
-)
+from .generate_image import generate_image_tool, register_generate_image_tool
 
 __all__ = [
     "generate_image_tool",
     "register_generate_image_tool",
-    "register_get_image_tool",
     "batch_generate_images",
     "register_batch_generate_tool",
 ]

src/tools/batch_generate.py

@@ -115,7 +115,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
         output_format: str = "png",
         batch_size: int | None = None,
         negative_prompt: str | None = None,
-        save_to_disk: bool = True,
     ) -> str:
         """
         Generate multiple images from a list of prompts efficiently.
@@ -131,13 +130,9 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
             output_format: Image format for all images (default: png)
             batch_size: Parallel batch size (default: from config)
             negative_prompt: Negative prompt for Imagen models (optional)
-            save_to_disk: Save images to disk and return file paths (default: True).
-                         When True: Returns file paths only (efficient for MCP clients).
-                         When False: Returns base64 image data (for cloud/serverless).
 
         Returns:
-            JSON string with batch results and either file paths (save_to_disk=True)
-            or base64 image data (save_to_disk=False)
+            JSON string with batch results including individual image paths
         """
         try:
             result = await batch_generate_images(
@@ -148,7 +143,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
                 output_format=output_format,
                 batch_size=batch_size,
                 negative_prompt=negative_prompt,
-                save_to_disk=save_to_disk,
             )
 
             return json.dumps(result, indent=2)

src/tools/generate_image.py

@@ -153,19 +153,10 @@ async def generate_image_tool(
             }
 
             if save_to_disk:
-                # Try to save to output directory, but don't fail if it errors
-                try:
-                    file_path = result.save(settings.output_dir)
-                    image_info["path"] = str(file_path)
-                    image_info["filename"] = file_path.name
-                    logger.info(f"Saved image to {file_path}")
-                except Exception as e:
-                    logger.warning(f"Failed to save image to disk: {e}")
-                    image_info["save_error"] = str(e)
-            else:
-                # Only include base64 data when not saving to disk (cloud/serverless scenarios)
-                # This prevents token overflow for MCP clients with file system access
-                image_info["image_base64"] = result.image_data
+                # Save to output directory
+                file_path = result.save(settings.output_dir)
+                image_info["path"] = str(file_path)
+                image_info["filename"] = file_path.name
 
             # Add enhanced prompt info
             if "enhanced_prompt" in result.metadata:
@@ -179,47 +170,6 @@ 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."""
 
@@ -237,7 +187,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
         use_world_knowledge: bool = False,
         negative_prompt: str | None = None,
         seed: int | None = None,
-        save_to_disk: bool = True,
     ) -> str:
         """
         Generate images using Google's Gemini or Imagen models.
@@ -259,9 +208,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
             use_world_knowledge: Use real-world knowledge (Gemini only)
             negative_prompt: What to avoid in the image (Imagen only)
             seed: Random seed for reproducibility (NOT SUPPORTED - will be ignored)
-            save_to_disk: Save images to disk and return file paths (default: True).
-                         When True: Returns file paths only (efficient for MCP clients).
-                         When False: Returns base64 image data (for cloud/serverless).
 
         Available models:
         - gemini-2.5-flash-image (default)
@@ -270,8 +216,7 @@ def register_generate_image_tool(mcp_server: Any) -> None:
         - imagen-4-ultra
 
         Returns:
-            JSON string with generation results and either file paths (save_to_disk=True)
-            or base64 image data (save_to_disk=False)
+            JSON string with generation results and file paths
         """
         try:
             result = await generate_image_tool(
@@ -287,7 +232,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
                 use_world_knowledge=use_world_knowledge,
                 negative_prompt=negative_prompt,
                 seed=seed,
-                save_to_disk=save_to_disk,
             )
 
             return json.dumps(result, indent=2)

{ultimate_gemini_mcp-1.0.18.dist-info → ultimate_gemini_mcp-1.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ultimate-gemini-mcp
-Version: 1.0.18
+Version: 1.5.1
 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.5.1.dist-info/RECORD

@@ -0,0 +1,21 @@
+src/__init__.py,sha256=nPaNFXizUWdTOnarUMPZPxfvBrhRHj7ifYKXoN8cvn8,435
+src/server.py,sha256=hyXuhS1h6DN2EiGle8giEPSV1UB3ba7t4do_2olTK3A,5862
+src/config/__init__.py,sha256=hL0recV_ycXBEGCym7BqwyaPCnQHy8o429pBirnBeiA,704
+src/config/constants.py,sha256=ue4dT6wFwCzAgDWvSt8RnbdaoaGHY8c7SViNxI-P73w,1870
+src/config/settings.py,sha256=uTxxblnyqmglz1COd3QxjC2w3o6l51S9MTanP4HBrgE,4257
+src/core/__init__.py,sha256=PQKMAY5eYIEO1oS_hrzeuDgRopl76Wn6NLfhtXBX92I,1200
+src/core/exceptions.py,sha256=NVb4tpwoX_DR-Wp8r73brXvZhEP35SUMyuEis8llFR0,1211
+src/core/validation.py,sha256=JTB6_ft4nN6NHRVD5vOv_q12aPupBrlaNl32jwR2TuE,4758
+src/services/__init__.py,sha256=n6FVDj052zvTerDFJSjaiLIycWGePm8Wr6dabbvd9o0,395
+src/services/gemini_client.py,sha256=VPuBPDHhAag_8xBqzMuVIEWVG0BINXOHkwKIALzcFAk,8324
+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/batch_generate.py,sha256=LLrIwVvx-0kZBCiR0tcYnmKi5iYIdV-93ekIQbMrabo,5004
+src/tools/generate_image.py,sha256=BGmplbkqaFvLa8mbJWs-a8v1t5i-Pdep4IZhXHIbisk,8456
+ultimate_gemini_mcp-1.5.1.dist-info/METADATA,sha256=KQptxAop21rrwPfCNAPTM0402UaZjuZLvyC93-75lrA,16130
+ultimate_gemini_mcp-1.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ultimate_gemini_mcp-1.5.1.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
+ultimate_gemini_mcp-1.5.1.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
+ultimate_gemini_mcp-1.5.1.dist-info/RECORD,,

src/prompts/__init__.py

@@ -1 +0,0 @@
-"""Prompts module for Ultimate Gemini MCP Server."""

src/resources/__init__.py

@@ -1 +0,0 @@
-"""Resources module for Ultimate Gemini MCP Server."""

src/utils/__init__.py

@@ -1 +0,0 @@
-"""Utilities module for Ultimate Gemini MCP Server."""

ultimate_gemini_mcp-1.0.18.dist-info/RECORD

@@ -1,24 +0,0 @@
-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
-src/core/__init__.py,sha256=PQKMAY5eYIEO1oS_hrzeuDgRopl76Wn6NLfhtXBX92I,1200
-src/core/exceptions.py,sha256=NVb4tpwoX_DR-Wp8r73brXvZhEP35SUMyuEis8llFR0,1211
-src/core/validation.py,sha256=JTB6_ft4nN6NHRVD5vOv_q12aPupBrlaNl32jwR2TuE,4758
-src/prompts/__init__.py,sha256=86jl0abA84m7ugnBVKMmw3-LYTkzIR1A0zpBJCwIX1s,53
-src/resources/__init__.py,sha256=LEm1Gy-IyekCbJ4bPiyzj0__-UxyXkM_PI3wgGofJsM,55
-src/services/__init__.py,sha256=n6FVDj052zvTerDFJSjaiLIycWGePm8Wr6dabbvd9o0,395
-src/services/gemini_client.py,sha256=VPuBPDHhAag_8xBqzMuVIEWVG0BINXOHkwKIALzcFAk,8324
-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=idmgH7aLtnwROyqOf2-KlTutb4mmJsN-xC2ClLB1Z-Q,419
-src/tools/batch_generate.py,sha256=ESAUkHX5TOKPlHCzgY2jMLyAvSmInN2Jwm5-51EeWwg,5406
-src/tools/generate_image.py,sha256=vTwLY1v8lTZlwb9T7TKNcuYxgNpDDJKaB_NMrM_C0C0,10726
-src/utils/__init__.py,sha256=T2UShY0j2kiu63YPOsTACrzOfWTTq4PmC0flzcgeCl0,55
-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,,