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

src/config/constants.py

@@ -31,13 +31,13 @@ DEFAULT_ENHANCEMENT_MODEL = "gemini-flash-latest"
 
 # Aspect ratios
 ASPECT_RATIOS = [
-    "1:1",   # Square
-    "2:3",   # Portrait
-    "3:2",   # Landscape
-    "3:4",   # Portrait
-    "4:3",   # Standard landscape
-    "4:5",   # Portrait
-    "5:4",   # Landscape
+    "1:1",  # Square
+    "2:3",  # Portrait
+    "3:2",  # Landscape
+    "3:4",  # Portrait
+    "4:3",  # Standard landscape
+    "4:5",  # Portrait
+    "5:4",  # Landscape
     "9:16",  # Vertical mobile
     "16:9",  # Widescreen
     "21:9",  # Ultrawide

src/config/settings.py

@@ -4,7 +4,6 @@ Configuration settings for the Ultimate Gemini MCP server.
 
 import os
 from pathlib import Path
-from typing import Optional
 
 from pydantic import Field
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -81,9 +80,7 @@ class APIConfig(BaseSettings):
     enable_prompt_enhancement: bool = Field(
         default=True, description="Enable automatic prompt enhancement"
     )
-    enable_batch_processing: bool = Field(
-        default=True, description="Enable batch processing"
-    )
+    enable_batch_processing: bool = Field(default=True, description="Enable batch processing")
 
     # Request settings
     request_timeout: int = Field(default=DEFAULT_TIMEOUT, description="API request timeout")
@@ -104,9 +101,7 @@ class APIConfig(BaseSettings):
             self.gemini_api_key = os.getenv("GOOGLE_API_KEY", "")
 
         if not self.gemini_api_key:
-            raise ValueError(
-                "GEMINI_API_KEY or GOOGLE_API_KEY environment variable is required"
-            )
+            raise ValueError("GEMINI_API_KEY or GOOGLE_API_KEY environment variable is required")
 
     @classmethod
     def from_env(cls) -> "APIConfig":
@@ -132,7 +127,7 @@ class Settings:
 
 
 # Global settings instance (lazy initialization)
-_settings: Optional[Settings] = None
+_settings: Settings | None = None
 
 
 def get_settings() -> Settings:

src/core/__init__.py

@@ -21,7 +21,6 @@ from .validation import (
     validate_model,
     validate_negative_prompt,
     validate_number_of_images,
-    validate_person_generation,
     validate_prompt,
     validate_prompts_list,
     validate_seed,
@@ -45,7 +44,6 @@ __all__ = [
     "validate_aspect_ratio",
     "validate_number_of_images",
     "validate_image_format",
-    "validate_person_generation",
     "validate_seed",
     "validate_file_path",
     "validate_base64_image",

src/core/exceptions.py

@@ -24,7 +24,9 @@ class ValidationError(UltimateGeminiError):
 class APIError(UltimateGeminiError):
     """Raised when an API request fails."""
 
-    def __init__(self, message: str, status_code: int | None = None, response_data: dict | None = None):
+    def __init__(
+        self, message: str, status_code: int | None = None, response_data: dict | None = None
+    ):
         super().__init__(message)
         self.status_code = status_code
         self.response_data = response_data

src/core/validation.py

@@ -5,7 +5,6 @@ Input validation utilities.
 import base64
 import re
 from pathlib import Path
-from typing import Any
 
 from ..config.constants import (
     ALL_MODELS,
@@ -14,7 +13,6 @@ from ..config.constants import (
     MAX_IMAGES_PER_REQUEST,
     MAX_NEGATIVE_PROMPT_LENGTH,
     MAX_PROMPT_LENGTH,
-    PERSON_GENERATION_OPTIONS,
 )
 from .exceptions import ValidationError
 
@@ -50,9 +48,7 @@ def validate_aspect_ratio(aspect_ratio: str) -> None:
     """Validate aspect ratio."""
     if aspect_ratio not in ASPECT_RATIOS:
         available = ", ".join(ASPECT_RATIOS)
-        raise ValidationError(
-            f"Invalid aspect ratio '{aspect_ratio}'. Available: {available}"
-        )
+        raise ValidationError(f"Invalid aspect ratio '{aspect_ratio}'. Available: {available}")
 
 
 def validate_number_of_images(num: int) -> None:
@@ -61,27 +57,14 @@ def validate_number_of_images(num: int) -> None:
         raise ValidationError(f"Number of images must be at least 1, got {num}")
 
     if num > MAX_IMAGES_PER_REQUEST:
-        raise ValidationError(
-            f"Number of images exceeds maximum: {num} > {MAX_IMAGES_PER_REQUEST}"
-        )
+        raise ValidationError(f"Number of images exceeds maximum: {num} > {MAX_IMAGES_PER_REQUEST}")
 
 
 def validate_image_format(format_str: str) -> None:
     """Validate image format."""
     if format_str.lower() not in IMAGE_FORMATS:
         available = ", ".join(IMAGE_FORMATS.keys())
-        raise ValidationError(
-            f"Invalid image format '{format_str}'. Available: {available}"
-        )
-
-
-def validate_person_generation(option: str) -> None:
-    """Validate person generation option."""
-    if option not in PERSON_GENERATION_OPTIONS:
-        available = ", ".join(PERSON_GENERATION_OPTIONS)
-        raise ValidationError(
-            f"Invalid person generation option '{option}'. Available: {available}"
-        )
+        raise ValidationError(f"Invalid image format '{format_str}'. Available: {available}")
 
 
 def validate_seed(seed: int | None) -> None:
@@ -98,7 +81,7 @@ def validate_file_path(path: str) -> Path:
     try:
         file_path = Path(path).resolve()
     except Exception as e:
-        raise ValidationError(f"Invalid file path '{path}': {e}")
+        raise ValidationError(f"Invalid file path '{path}': {e}") from e
 
     if not file_path.exists():
         raise ValidationError(f"File does not exist: {file_path}")
@@ -120,7 +103,7 @@ def validate_base64_image(data: str) -> None:
         if len(decoded) == 0:
             raise ValidationError("Decoded image data is empty")
     except Exception as e:
-        raise ValidationError(f"Invalid base64 image data: {e}")
+        raise ValidationError(f"Invalid base64 image data: {e}") from e
 
 
 def validate_prompts_list(prompts: list[str]) -> None:
@@ -137,7 +120,7 @@ def validate_prompts_list(prompts: list[str]) -> None:
         try:
             validate_prompt(prompt)
         except ValidationError as e:
-            raise ValidationError(f"Invalid prompt at index {i}: {e}")
+            raise ValidationError(f"Invalid prompt at index {i}: {e}") from e
 
 
 def sanitize_filename(filename: str) -> str:

src/server.py

@@ -10,7 +10,6 @@ Unified MCP server supporting:
 
 import logging
 import sys
-from pathlib import Path
 
 from fastmcp import FastMCP
 
@@ -69,9 +68,9 @@ def create_app() -> FastMCP:
                             "Image editing",
                             "Character consistency",
                             "Multi-image blending",
-                            "World knowledge integration"
+                            "World knowledge integration",
                         ],
-                        "default": True
+                        "default": True,
                     }
                 },
                 "imagen": {
@@ -84,8 +83,8 @@ def create_app() -> FastMCP:
                             "Negative prompts",
                             "Seed-based reproducibility",
                             "Person generation controls",
-                            "Advanced controls"
-                        ]
+                            "Advanced controls",
+                        ],
                     },
                     "imagen-4-fast": {
                         "name": "Imagen 4 Fast",
@@ -96,8 +95,8 @@ def create_app() -> FastMCP:
                             "Negative prompts",
                             "Seed-based reproducibility",
                             "Person generation controls",
-                            "Cost-effective"
-                        ]
+                            "Cost-effective",
+                        ],
                     },
                     "imagen-4-ultra": {
                         "name": "Imagen 4 Ultra",
@@ -107,10 +106,10 @@ def create_app() -> FastMCP:
                             "Best prompt adherence",
                             "Professional results",
                             "Enhanced text rendering",
-                            "Advanced controls"
-                        ]
-                    }
-                }
+                            "Advanced controls",
+                        ],
+                    },
+                },
             }
 
             return json.dumps(models_info, indent=2)

src/services/__init__.py

@@ -1,8 +1,8 @@
 """Services module for Ultimate Gemini MCP."""
 
 from .gemini_client import GeminiClient
-from .imagen_client import ImagenClient
 from .image_service import ImageResult, ImageService
+from .imagen_client import ImagenClient
 from .prompt_enhancer import PromptEnhancer, create_prompt_enhancer
 
 __all__ = [

src/services/gemini_client.py

@@ -3,7 +3,6 @@ Gemini API client for Gemini 2.5 Flash Image generation.
 Uses the generateContent API endpoint per Google's documentation.
 """
 
-import base64
 import logging
 from typing import Any
 
@@ -69,12 +68,7 @@ class GeminiClient:
 
         # Add input image if provided (for editing)
         if input_image:
-            parts.append({
-                "inline_data": {
-                    "mime_type": "image/png",
-                    "data": input_image
-                }
-            })
+            parts.append({"inline_data": {"mime_type": "image/png", "data": input_image}})
 
         # Add text prompt (include aspect ratio hint if specified)
         prompt_text = prompt
@@ -84,20 +78,13 @@ class GeminiClient:
         parts.append({"text": prompt_text})
 
         # Build generation config for image generation
-        generation_config = {
-            "responseModalities": ["Image"]
-        }
+        generation_config = {"responseModalities": ["Image"]}
 
         # Add aspect ratio to image config if specified
         if aspect_ratio:
-            generation_config["imageConfig"] = {
-                "aspectRatio": aspect_ratio
-            }
+            generation_config["imageConfig"] = {"aspectRatio": aspect_ratio}
 
-        request_body = {
-            "contents": [{"parts": parts}],
-            "generationConfig": generation_config
-        }
+        request_body = {"contents": [{"parts": parts}], "generationConfig": generation_config}
 
         headers = {
             "x-goog-api-key": self.api_key,
@@ -118,7 +105,9 @@ class GeminiClient:
             images = self._extract_images(data)
 
             if not images:
-                logger.error(f"No images extracted from response. Response structure: {list(data.keys())}")
+                logger.error(
+                    f"No images extracted from response. Response structure: {list(data.keys())}"
+                )
                 if "candidates" in data:
                     logger.error(f"Candidates: {data['candidates']}")
                 raise APIError("No image data found in Gemini API response")
@@ -129,7 +118,7 @@ class GeminiClient:
             self._handle_http_error(e)
         except Exception as e:
             logger.error(f"Gemini API request failed: {e}")
-            raise APIError(f"Gemini API request failed: {e}")
+            raise APIError(f"Gemini API request failed: {e}") from e
 
     async def generate_text(
         self,
@@ -152,14 +141,10 @@ class GeminiClient:
         model_id = GEMINI_MODELS.get(model, model)
         url = f"{self.base_url}/models/{model_id}:generateContent"
 
-        request_body = {
-            "contents": [{"parts": [{"text": prompt}]}]
-        }
+        request_body = {"contents": [{"parts": [{"text": prompt}]}]}
 
         if system_instruction:
-            request_body["system_instruction"] = {
-                "parts": [{"text": system_instruction}]
-            }
+            request_body["system_instruction"] = {"parts": [{"text": system_instruction}]}
 
         headers = {
             "x-goog-api-key": self.api_key,
@@ -179,7 +164,7 @@ class GeminiClient:
             self._handle_http_error(e)
         except Exception as e:
             logger.error(f"Gemini text generation failed: {e}")
-            raise APIError(f"Gemini text generation failed: {e}")
+            raise APIError(f"Gemini text generation failed: {e}") from e
 
     def _extract_images(self, response_data: dict[str, Any]) -> list[str]:
         """Extract base64 image data from Gemini API response."""
@@ -230,23 +215,23 @@ class GeminiClient:
 
         if status_code == 401 or status_code == 403:
             raise AuthenticationError(
-                "Authentication failed. Please check your Gemini API key.",
-                status_code=status_code
+                "Authentication failed. Please check your Gemini API key.", status_code=status_code
             )
         elif status_code == 429:
             raise RateLimitError(
-                "Rate limit exceeded. Please try again later.",
-                status_code=status_code
+                "Rate limit exceeded. Please try again later.", status_code=status_code
             )
-        elif status_code == 400 and ("SAFETY" in error_text.upper() or "BLOCKED" in error_text.upper()):
+        elif status_code == 400 and (
+            "SAFETY" in error_text.upper() or "BLOCKED" in error_text.upper()
+        ):
             raise ContentPolicyError(
                 "Content was blocked by safety filters. Please modify your prompt.",
-                status_code=status_code
+                status_code=status_code,
             )
         else:
             raise APIError(
                 f"API request failed with status {status_code}: {error_text}",
-                status_code=status_code
+                status_code=status_code,
             )
 
     async def close(self) -> None:

src/services/image_service.py

@@ -9,8 +9,6 @@ from datetime import datetime
 from pathlib import Path
 from typing import Any
 
-from PIL import Image
-
 from ..config.constants import GEMINI_MODELS, IMAGEN_MODELS
 from ..core import sanitize_filename
 from ..core.exceptions import ImageProcessingError
@@ -30,7 +28,7 @@ class ImageResult:
         prompt: str,
         model: str,
         index: int = 0,
-        metadata: dict[str, Any] | None = None
+        metadata: dict[str, Any] | None = None,
     ):
         self.image_data = image_data  # Base64-encoded
         self.prompt = prompt
@@ -53,7 +51,7 @@ class ImageResult:
             logger.info(f"Saved image to {output_path}")
             return output_path
         except Exception as e:
-            raise ImageProcessingError(f"Failed to save image: {e}")
+            raise ImageProcessingError(f"Failed to save image: {e}") from e
 
     def _generate_filename(self) -> str:
         """Generate descriptive filename."""
@@ -71,13 +69,7 @@ class ImageResult:
 class ImageService:
     """Unified service for image generation using Gemini or Imagen."""
 
-    def __init__(
-        self,
-        api_key: str,
-        *,
-        enable_enhancement: bool = True,
-        timeout: int = 60
-    ):
+    def __init__(self, api_key: str, *, enable_enhancement: bool = True, timeout: int = 60):
         """
         Initialize image service.
 
@@ -100,12 +92,7 @@ class ImageService:
             self.prompt_enhancer = PromptEnhancer(self.gemini_client)
 
     async def generate(
-        self,
-        prompt: str,
-        *,
-        model: str | None = None,
-        enhance_prompt: bool = True,
-        **kwargs: Any
+        self, prompt: str, *, model: str | None = None, enhance_prompt: bool = True, **kwargs: Any
     ) -> list[ImageResult]:
         """
         Generate images using the appropriate API.
@@ -136,8 +123,7 @@ class ImageService:
         if enhance_prompt and self.enable_enhancement and self.prompt_enhancer:
             try:
                 result = await self.prompt_enhancer.enhance_prompt(
-                    prompt,
-                    context=enhancement_context
+                    prompt, context=enhancement_context
                 )
                 prompt = result["enhanced_prompt"]
                 logger.info(f"Prompt enhanced: {len(original_prompt)} -> {len(prompt)} chars")
@@ -151,18 +137,10 @@ class ImageService:
             return await self._generate_with_imagen(prompt, model, original_prompt, kwargs)
 
     async def _generate_with_gemini(
-        self,
-        prompt: str,
-        model: str,
-        original_prompt: str,
-        params: dict[str, Any]
+        self, prompt: str, model: str, original_prompt: str, params: dict[str, Any]
     ) -> list[ImageResult]:
         """Generate images using Gemini API."""
-        response = await self.gemini_client.generate_image(
-            prompt=prompt,
-            model=model,
-            **params
-        )
+        response = await self.gemini_client.generate_image(prompt=prompt, model=model, **params)
 
         images = response["images"]
         results = []
@@ -173,29 +151,17 @@ class ImageService:
                 prompt=original_prompt,
                 model=model,
                 index=i,
-                metadata={
-                    "enhanced_prompt": prompt,
-                    "api": "gemini",
-                    **params
-                }
+                metadata={"enhanced_prompt": prompt, "api": "gemini", **params},
             )
             results.append(result)
 
         return results
 
     async def _generate_with_imagen(
-        self,
-        prompt: str,
-        model: str,
-        original_prompt: str,
-        params: dict[str, Any]
+        self, prompt: str, model: str, original_prompt: str, params: dict[str, Any]
     ) -> list[ImageResult]:
         """Generate images using Imagen API."""
-        response = await self.imagen_client.generate_image(
-            prompt=prompt,
-            model=model,
-            **params
-        )
+        response = await self.imagen_client.generate_image(prompt=prompt, model=model, **params)
 
         images = response["images"]
         results = []
@@ -206,11 +172,7 @@ class ImageService:
                 prompt=original_prompt,
                 model=model,
                 index=i,
-                metadata={
-                    "enhanced_prompt": prompt,
-                    "api": "imagen",
-                    **params
-                }
+                metadata={"enhanced_prompt": prompt, "api": "imagen", **params},
             )
             results.append(result)
 

src/services/imagen_client.py

@@ -73,19 +73,22 @@ class ImagenClient:
 
         # Build request body according to Imagen API
         request_body: dict[str, Any] = {
-            "instances": [
-                {
-                    "prompt": prompt
-                }
-            ],
+            "instances": [{"prompt": prompt}],
             "parameters": {
                 "outputMimeType": output_format,
                 "sampleCount": number_of_images,
                 "personGeneration": person_generation,
-                "aspectRatio": aspect_ratio
-            }
+                "aspectRatio": aspect_ratio,
+            },
         }
 
+        # Add imageSize for models that support it (Standard and Ultra, not Fast)
+        # Fast model only supports 1K, while Standard and Ultra support up to 2K
+        if model in ["imagen-4", "imagen-4-ultra"]:
+            request_body["parameters"]["imageSize"] = "2K"
+        elif model == "imagen-4-fast":
+            request_body["parameters"]["imageSize"] = "1K"
+
         # Add optional parameters
         if negative_prompt:
             request_body["instances"][0]["negativePrompt"] = negative_prompt
@@ -106,9 +109,7 @@ class ImagenClient:
             logger.debug(f"Sending request to {url}")
             # Add API key as query parameter
             response = await self.client.post(
-                f"{url}?key={self.api_key}",
-                json=request_body,
-                headers=headers
+                f"{url}?key={self.api_key}", json=request_body, headers=headers
             )
             response.raise_for_status()
             data = response.json()
@@ -119,17 +120,13 @@ class ImagenClient:
             if not images:
                 raise APIError("No image data found in Imagen API response")
 
-            return {
-                "images": images,
-                "model": model,
-                "response": data
-            }
+            return {"images": images, "model": model, "response": data}
 
         except httpx.HTTPStatusError as e:
             self._handle_http_error(e)
         except Exception as e:
             logger.error(f"Imagen API request failed: {e}")
-            raise APIError(f"Imagen API request failed: {e}")
+            raise APIError(f"Imagen API request failed: {e}") from e
 
     def _extract_images(self, response_data: dict[str, Any]) -> list[str]:
         """Extract base64 image data from Imagen API response."""
@@ -156,23 +153,21 @@ class ImagenClient:
 
         if status_code == 401 or status_code == 403:
             raise AuthenticationError(
-                "Authentication failed. Please check your API key.",
-                status_code=status_code
+                "Authentication failed. Please check your API key.", status_code=status_code
             )
         elif status_code == 429:
             raise RateLimitError(
-                "Rate limit exceeded. Please try again later.",
-                status_code=status_code
+                "Rate limit exceeded. Please try again later.", status_code=status_code
             )
         elif status_code == 400 and "SAFETY" in error_text.upper():
             raise ContentPolicyError(
                 "Content was blocked by safety filters. Please modify your prompt.",
-                status_code=status_code
+                status_code=status_code,
             )
         else:
             raise APIError(
                 f"API request failed with status {status_code}: {error_text}",
-                status_code=status_code
+                status_code=status_code,
             )
 
     async def close(self) -> None:

src/services/prompt_enhancer.py

@@ -64,7 +64,7 @@ class PromptEnhancer:
             enhanced = await self.gemini_client.generate_text(
                 prompt=instruction,
                 system_instruction=PROMPT_ENHANCEMENT_SYSTEM_INSTRUCTION,
-                model="gemini-flash-latest"
+                model="gemini-flash-latest",
             )
 
             # Clean up the enhanced prompt
@@ -84,11 +84,7 @@ class PromptEnhancer:
                 "enhanced_prompt": original_prompt,
             }
 
-    def _build_enhancement_instruction(
-        self,
-        prompt: str,
-        context: dict[str, Any] | None
-    ) -> str:
+    def _build_enhancement_instruction(self, prompt: str, context: dict[str, Any] | None) -> str:
         """Build the instruction for prompt enhancement."""
         instruction_parts = [f"Enhance this image generation prompt:\n\n{prompt}"]
 

src/tools/batch_generate.py

@@ -54,12 +54,12 @@ async def batch_generate_images(
         "batch_size": batch_size,
         "completed": 0,
         "failed": 0,
-        "results": []
+        "results": [],
     }
 
     # Process prompts in batches
     for i in range(0, len(prompts), batch_size):
-        batch = prompts[i:i + batch_size]
+        batch = prompts[i : i + batch_size]
         logger.info(f"Processing batch {i // batch_size + 1}: {len(batch)} prompts")
 
         # Create tasks for parallel processing
@@ -71,7 +71,7 @@ async def batch_generate_images(
                 aspect_ratio=aspect_ratio,
                 output_format=output_format,
                 number_of_images=1,
-                **shared_params
+                **shared_params,
             )
             for prompt in batch
         ]
@@ -86,19 +86,19 @@ async def batch_generate_images(
             if isinstance(result, Exception):
                 logger.error(f"Failed to generate image for prompt {prompt_index}: {result}")
                 results["failed"] += 1
-                results["results"].append({
-                    "prompt_index": prompt_index,
-                    "prompt": batch[j],
-                    "success": False,
-                    "error": str(result)
-                })
+                results["results"].append(
+                    {
+                        "prompt_index": prompt_index,
+                        "prompt": batch[j],
+                        "success": False,
+                        "error": str(result),
+                    }
+                )
             else:
                 results["completed"] += 1
-                results["results"].append({
-                    "prompt_index": prompt_index,
-                    "prompt": batch[j],
-                    **result
-                })
+                results["results"].append(
+                    {"prompt_index": prompt_index, "prompt": batch[j], **result}
+                )
 
     return results
 
@@ -114,7 +114,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
         aspect_ratio: str = "1:1",
         output_format: str = "png",
         batch_size: int | None = None,
-        person_generation: str = "allow_adult",
         negative_prompt: str | None = None,
     ) -> str:
         """
@@ -130,7 +129,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
             aspect_ratio: Aspect ratio for all images (default: 1:1)
             output_format: Image format for all images (default: png)
             batch_size: Parallel batch size (default: from config)
-            person_generation: Person policy for Imagen models (default: allow_adult)
             negative_prompt: Negative prompt for Imagen models (optional)
 
         Returns:
@@ -144,7 +142,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
                 aspect_ratio=aspect_ratio,
                 output_format=output_format,
                 batch_size=batch_size,
-                person_generation=person_generation,
                 negative_prompt=negative_prompt,
             )
 
@@ -152,8 +149,6 @@ def register_batch_generate_tool(mcp_server: Any) -> None:
 
         except Exception as e:
             logger.error(f"Batch generation error: {e}")
-            return json.dumps({
-                "success": False,
-                "error": str(e),
-                "error_type": type(e).__name__
-            }, indent=2)
+            return json.dumps(
+                {"success": False, "error": str(e), "error_type": type(e).__name__}, indent=2
+            )

src/tools/generate_image.py

@@ -8,15 +8,12 @@ import logging
 from pathlib import Path
 from typing import Any
 
-from fastmcp import Image
-
-from ..config import ALL_MODELS, ASPECT_RATIOS, get_settings
+from ..config import get_settings
 from ..core import (
     validate_aspect_ratio,
     validate_image_format,
     validate_model,
     validate_number_of_images,
-    validate_person_generation,
     validate_prompt,
     validate_seed,
 )
@@ -38,7 +35,6 @@ async def generate_image_tool(
     blend_images: bool = False,
     use_world_knowledge: bool = False,
     # Imagen-specific options
-    person_generation: str = "allow_adult",
     negative_prompt: str | None = None,
     seed: int | None = None,
     # Output options
@@ -59,7 +55,6 @@ async def generate_image_tool(
         maintain_character_consistency: Maintain character features across generations (Gemini)
         blend_images: Enable multi-image blending (Gemini)
         use_world_knowledge: Use real-world knowledge for context (Gemini)
-        person_generation: Person generation policy (Imagen: dont_allow, allow_adult, allow_all)
         negative_prompt: What to avoid in the image (Imagen)
         seed: Random seed for reproducibility (Imagen)
         save_to_disk: Save images to output directory
@@ -75,18 +70,6 @@ async def generate_image_tool(
     validate_aspect_ratio(aspect_ratio)
     validate_image_format(output_format)
 
-    if person_generation:
-        validate_person_generation(person_generation)
-
-        # Warn if prompt may conflict with person_generation policy
-        if person_generation == "dont_allow":
-            person_keywords = ["person", "people", "man", "woman", "child", "human", "face", "portrait", "crowd"]
-            if any(keyword in prompt.lower() for keyword in person_keywords):
-                logger.warning(
-                    f"Prompt contains person-related keywords but person_generation is set to 'dont_allow'. "
-                    f"This may result in the API blocking image generation."
-                )
-
     if seed is not None:
         validate_seed(seed)
         logger.warning(
@@ -134,7 +117,7 @@ async def generate_image_tool(
         if model.startswith("imagen"):
             params["number_of_images"] = number_of_images
             params["output_format"] = f"image/{output_format}"
-            params["person_generation"] = person_generation
+            params["person_generation"] = "allow_all"  # Hard-coded to allow all people
             if negative_prompt:
                 params["negative_prompt"] = negative_prompt
             if seed is not None:
@@ -145,7 +128,7 @@ async def generate_image_tool(
             prompt=prompt,
             model=model,
             enhance_prompt=enhance_prompt and settings.api.enable_prompt_enhancement,
-            **params
+            **params,
         )
 
         # Prepare response
@@ -158,7 +141,7 @@ async def generate_image_tool(
             "metadata": {
                 "enhance_prompt": enhance_prompt,
                 "aspect_ratio": aspect_ratio,
-            }
+            },
         }
 
         # Save images and prepare for MCP response
@@ -202,7 +185,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
         maintain_character_consistency: bool = False,
         blend_images: bool = False,
         use_world_knowledge: bool = False,
-        person_generation: str = "allow_adult",
         negative_prompt: str | None = None,
         seed: int | None = None,
     ) -> str:
@@ -224,7 +206,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
             maintain_character_consistency: Maintain character features (Gemini only)
             blend_images: Enable multi-image blending (Gemini only)
             use_world_knowledge: Use real-world knowledge (Gemini only)
-            person_generation: Person policy: dont_allow, allow_adult, allow_all (Imagen only)
             negative_prompt: What to avoid in the image (Imagen only)
             seed: Random seed for reproducibility (NOT SUPPORTED - will be ignored)
 
@@ -249,7 +230,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
                 maintain_character_consistency=maintain_character_consistency,
                 blend_images=blend_images,
                 use_world_knowledge=use_world_knowledge,
-                person_generation=person_generation,
                 negative_prompt=negative_prompt,
                 seed=seed,
             )
@@ -258,8 +238,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
 
         except Exception as e:
             logger.error(f"Error generating image: {e}")
-            return json.dumps({
-                "success": False,
-                "error": str(e),
-                "error_type": type(e).__name__
-            }, indent=2)
+            return json.dumps(
+                {"success": False, "error": str(e), "error_type": type(e).__name__}, indent=2
+            )

{ultimate_gemini_mcp-1.0.3.dist-info → ultimate_gemini_mcp-1.0.5.dist-info}/METADATA

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

@@ -0,0 +1,21 @@
+src/__init__.py,sha256=T9HBR_iOl-3bM_AG1NyK9PT1li5t_kuhswn-oVWOJWI,435
+src/server.py,sha256=nZI63qIDL3JWv3dyyFk6lIXStZuHe1MTdZqZSrqK56k,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.0.5.dist-info/METADATA,sha256=G2FWqdrQVbJaMDpi5TpLshSnoDrY6fp4UrREpBFoc6M,11646
+ultimate_gemini_mcp-1.0.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+ultimate_gemini_mcp-1.0.5.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
+ultimate_gemini_mcp-1.0.5.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
+ultimate_gemini_mcp-1.0.5.dist-info/RECORD,,

ultimate_gemini_mcp-1.0.3.dist-info/RECORD

@@ -1,21 +0,0 @@
-src/__init__.py,sha256=fXeeHRer3O8akAlzdXxD5fmz35ToF8aNHQ7eUXE6-xA,435
-src/server.py,sha256=vVJ96VMl-Z7y756qjASywZaD2QYj5v8nIQ8r504Mg3g,5877
-src/config/__init__.py,sha256=hL0recV_ycXBEGCym7BqwyaPCnQHy8o429pBirnBeiA,704
-src/config/constants.py,sha256=iLHMFVJzWRkSMkG6gXHBWepVLbcDvlgqZtkgfEuaaUQ,1877
-src/config/settings.py,sha256=IxxvETiW-CK5VLeTOegx3hAfd5gIxBSVAUyKbc9GBpU,4332
-src/core/__init__.py,sha256=h8ik58bFn0g0XFVtMEI4MDoFapi4FlQywgd3S0eR2Oo,1266
-src/core/exceptions.py,sha256=U09aPGmlDTAGbt6PbF-PKbmT1ee9LHqghkOXGn094UI,1197
-src/core/validation.py,sha256=NzF01GVhmBefn_bo4cGDMREuE6kc3RujbfOtfx9eLyk,5186
-src/services/__init__.py,sha256=5iCkCasXmsBcJ0zDIu3qCSdTZtmKPDmWuz14dMK0N7I,395
-src/services/gemini_client.py,sha256=6xZWNIRbqtXUP7hs-LygHXKfMvkhOlPaR36nKJkM74Y,8520
-src/services/image_service.py,sha256=UcErTYMjG6ihKeT2w0UaGiiVIoUwf2Kt7oSafEjx28w,7526
-src/services/imagen_client.py,sha256=TWtordh22auxI19ZdWxyXpfSgrWFnM28I1WzTqMiVn4,5932
-src/services/prompt_enhancer.py,sha256=JFTKqwWafngq37yGjiYT5-FJbIj4Buvt0js2nU_9gsw,4805
-src/tools/__init__.py,sha256=zBfAjFT51LvvD7WXTfDYiyJstRdphr2ChddAmGMZxkI,346
-src/tools/batch_generate.py,sha256=tcilmaOWXMFnVBMdMbIUXxHtbpnBLeIkJe2LDXnE7qU,5200
-src/tools/generate_image.py,sha256=woTKSXUE12wX1vCQ3bMzLHduXgU7Z3hCvhoDekadzP0,9506
-ultimate_gemini_mcp-1.0.3.dist-info/METADATA,sha256=LPy5QqjDXamsbIA6oWn2n3g3V1tCDSpVaam0alT5H-0,11646
-ultimate_gemini_mcp-1.0.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ultimate_gemini_mcp-1.0.3.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
-ultimate_gemini_mcp-1.0.3.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
-ultimate_gemini_mcp-1.0.3.dist-info/RECORD,,