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

src/services/gemini_client.py

@@ -83,7 +83,21 @@ class GeminiClient:
 
         parts.append({"text": prompt_text})
 
-        request_body = {"contents": [{"parts": parts}]}
+        # Build generation config for image generation
+        generation_config = {
+            "responseModalities": ["Image"]
+        }
+
+        # Add aspect ratio to image config if specified
+        if aspect_ratio:
+            generation_config["imageConfig"] = {
+                "aspectRatio": aspect_ratio
+            }
+
+        request_body = {
+            "contents": [{"parts": parts}],
+            "generationConfig": generation_config
+        }
 
         headers = {
             "x-goog-api-key": self.api_key,
@@ -92,14 +106,21 @@ class GeminiClient:
 
         try:
             logger.debug(f"Sending request to {url}")
+            logger.debug(f"Request body: {request_body}")
             response = await self.client.post(url, json=request_body, headers=headers)
             response.raise_for_status()
             data = response.json()
 
+            logger.debug(f"Response status: {response.status_code}")
+            logger.debug(f"Response data: {data}")
+
             # Extract images from response
             images = self._extract_images(data)
 
             if not images:
+                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")
 
             return {"images": images, "model": model, "response": data}
@@ -170,10 +191,13 @@ class GeminiClient:
                 content = candidate.get("content", {})
                 parts = content.get("parts", [])
                 for part in parts:
-                    if "inline_data" in part:
-                        image_data = part["inline_data"].get("data")
+                    # Handle both inline_data and inlineData formats
+                    inline_data = part.get("inline_data") or part.get("inlineData")
+                    if inline_data:
+                        image_data = inline_data.get("data")
                         if image_data:
                             images.append(image_data)
+                            logger.debug(f"Extracted image data of length: {len(image_data)}")
         except Exception as e:
             logger.warning(f"Error extracting images from response: {e}")
 

src/services/imagen_client.py

@@ -90,8 +90,13 @@ class ImagenClient:
         if negative_prompt:
             request_body["instances"][0]["negativePrompt"] = negative_prompt
 
+        # Note: Seed parameter is not supported by Imagen API (as of 2025)
+        # The API returns a 400 error if seed is provided
         if seed is not None:
-            request_body["parameters"]["seed"] = seed
+            logger.warning(
+                "Seed parameter is not supported by Imagen API and will be ignored. "
+                "Images cannot be reproduced with a seed value."
+            )
 
         headers = {
             "Content-Type": "application/json",

src/tools/generate_image.py

@@ -3,6 +3,7 @@ Image generation tool supporting both Gemini and Imagen models.
 """
 
 import base64
+import json
 import logging
 from pathlib import Path
 from typing import Any
@@ -76,8 +77,21 @@ async def generate_image_tool(
 
     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(
+            "Note: The seed parameter is not currently supported by Imagen API and will be ignored."
+        )
 
     # Get settings
     settings = get_settings()
@@ -212,7 +226,7 @@ def register_generate_image_tool(mcp_server: Any) -> None:
             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 (Imagen only)
+            seed: Random seed for reproducibility (NOT SUPPORTED - will be ignored)
 
         Available models:
         - gemini-2.5-flash-image (default)
@@ -240,7 +254,6 @@ def register_generate_image_tool(mcp_server: Any) -> None:
                 seed=seed,
             )
 
-            import json
             return json.dumps(result, indent=2)
 
         except Exception as e:

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

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

@@ -1,4 +1,4 @@
-src/__init__.py,sha256=DNVyeC3hyUv0WJlXtrOXH-Miw4ee8ZDiCW6GwJnjPJo,435
+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
@@ -7,15 +7,15 @@ 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=nVhx2s8sVhizeHa1D6IbxT5TRqGIF23kobMTcBh5vyA,7490
+src/services/gemini_client.py,sha256=6xZWNIRbqtXUP7hs-LygHXKfMvkhOlPaR36nKJkM74Y,8520
 src/services/image_service.py,sha256=UcErTYMjG6ihKeT2w0UaGiiVIoUwf2Kt7oSafEjx28w,7526
-src/services/imagen_client.py,sha256=dvDa0Z6xV-xeF428_arf3JXMf-rYlwOXoDD_6pTZzOE,5660
+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=7X23jAdxzyC4lWCmIu3tIcs807w3duDrZdGMb7cPpZE,8818
-ultimate_gemini_mcp-1.0.1.dist-info/METADATA,sha256=R2TJ6uoVfF0q9541w7RUB5lFfWufVQ5HYvQNbT8ZKnU,11646
-ultimate_gemini_mcp-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ultimate_gemini_mcp-1.0.1.dist-info/entry_points.txt,sha256=-BeRTT4oR05e-YnF1ZNbNxlaekD4LsWhD-Zy_1dyRnc,56
-ultimate_gemini_mcp-1.0.1.dist-info/licenses/LICENSE,sha256=ilyzUnN0QHYtYGJks-NFUwiniNu08IedLmn_muRqa0o,1480
-ultimate_gemini_mcp-1.0.1.dist-info/RECORD,,
+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,,