awslabs.nova-canvas-mcp-server 0.1.10233__py3-none-any.whl → 0.1.10652__py3-none-any.whl

awslabs/nova_canvas_mcp_server/__init__.py

@@ -1,3 +1,3 @@
 """awslabs.nova-canvas-mcp-server"""
 
-__version__ = '0.1.6'
+__version__ = "0.1.6"

awslabs/nova_canvas_mcp_server/consts.py

@@ -1,11 +1,11 @@
 # Constants
-NOVA_CANVAS_MODEL_ID = 'amazon.nova-canvas-v1:0'
+NOVA_CANVAS_MODEL_ID = "amazon.nova-canvas-v1:0"
 DEFAULT_WIDTH = 1024
 DEFAULT_HEIGHT = 1024
-DEFAULT_QUALITY = 'standard'
+DEFAULT_QUALITY = "standard"
 DEFAULT_CFG_SCALE = 6.5
 DEFAULT_NUMBER_OF_IMAGES = 1
-DEFAULT_OUTPUT_DIR = 'output'  # Default directory inside workspace_dir
+DEFAULT_OUTPUT_DIR = "output"  # Default directory inside workspace_dir
 
 
 # Nova Canvas Prompt Best Practices

awslabs/nova_canvas_mcp_server/models.py

@@ -15,8 +15,8 @@ class Quality(str, Enum):
         PREMIUM: Premium quality image generation with enhanced details.
     """
 
-    STANDARD = 'standard'
-    PREMIUM = 'premium'
+    STANDARD = "standard"
+    PREMIUM = "premium"
 
 
 class TaskType(str, Enum):
@@ -27,8 +27,8 @@ class TaskType(str, Enum):
         COLOR_GUIDED_GENERATION: Generate an image guided by both text and color palette.
     """
 
-    TEXT_IMAGE = 'TEXT_IMAGE'
-    COLOR_GUIDED_GENERATION = 'COLOR_GUIDED_GENERATION'
+    TEXT_IMAGE = "TEXT_IMAGE"
+    COLOR_GUIDED_GENERATION = "COLOR_GUIDED_GENERATION"
 
 
 class ImageGenerationConfig(BaseModel):
@@ -50,10 +50,12 @@ class ImageGenerationConfig(BaseModel):
     height: int = Field(default=1024, ge=320, le=4096)
     quality: Quality = Quality.STANDARD
     cfgScale: float = Field(default=6.5, ge=1.1, le=10.0)
-    seed: int = Field(default_factory=lambda: random.randint(0, 858993459), ge=0, le=858993459)
+    seed: int = Field(
+        default_factory=lambda: random.randint(0, 858993459), ge=0, le=858993459
+    )
     numberOfImages: int = Field(default=1, ge=1, le=5)
 
-    @field_validator('width', 'height')
+    @field_validator("width", "height")
     @classmethod
     def must_be_divisible_by_16(cls, v: int) -> int:
         """Validate that width and height are divisible by 16.
@@ -68,10 +70,10 @@ class ImageGenerationConfig(BaseModel):
             ValueError: If the value is not divisible by 16.
         """
         if v % 16 != 0:
-            raise ValueError('Value must be divisible by 16')
+            raise ValueError("Value must be divisible by 16")
         return v
 
-    @model_validator(mode='after')
+    @model_validator(mode="after")
     def validate_aspect_ratio_and_total_pixels(self):
         """Validate aspect ratio and total pixel count.
 
@@ -91,12 +93,12 @@ class ImageGenerationConfig(BaseModel):
         # Check aspect ratio between 1:4 and 4:1
         aspect_ratio = width / height
         if aspect_ratio < 0.25 or aspect_ratio > 4.0:
-            raise ValueError('Aspect ratio must be between 1:4 and 4:1')
+            raise ValueError("Aspect ratio must be between 1:4 and 4:1")
 
         # Check total pixel count
         total_pixels = width * height
         if total_pixels >= 4194304:
-            raise ValueError('Total pixel count must be less than 4,194,304')
+            raise ValueError("Total pixel count must be less than 4,194,304")
 
         return self
 
@@ -130,7 +132,7 @@ class ColorGuidedGenerationParams(BaseModel):
     text: str = Field(..., min_length=1, max_length=1024)
     negativeText: Optional[str] = Field(default=None, min_length=1, max_length=1024)
 
-    @field_validator('colors')
+    @field_validator("colors")
     @classmethod
     def validate_hex_colors(cls, v: List[str]) -> List[str]:
         """Validate that colors are in the correct hexadecimal format.
@@ -144,7 +146,7 @@ class ColorGuidedGenerationParams(BaseModel):
         Raises:
             ValueError: If any color is not a valid hexadecimal color in the format '#RRGGBB'.
         """
-        hex_pattern = re.compile(r'^#[0-9A-Fa-f]{6}$')
+        hex_pattern = re.compile(r"^#[0-9A-Fa-f]{6}$")
         for color in v:
             if not hex_pattern.match(color):
                 raise ValueError(
@@ -180,13 +182,13 @@ class TextImageRequest(BaseModel):
         """
         text_to_image_params = self.textToImageParams.model_dump()
         # Remove negativeText if it's None
-        if text_to_image_params.get('negativeText') is None:
-            text_to_image_params.pop('negativeText', None)
+        if text_to_image_params.get("negativeText") is None:
+            text_to_image_params.pop("negativeText", None)
 
         return {
-            'taskType': self.taskType,
-            'textToImageParams': text_to_image_params,
-            'imageGenerationConfig': self.imageGenerationConfig.model_dump()
+            "taskType": self.taskType,
+            "textToImageParams": text_to_image_params,
+            "imageGenerationConfig": self.imageGenerationConfig.model_dump()
             if self.imageGenerationConfig
             else None,
         }
@@ -204,7 +206,9 @@ class ColorGuidedRequest(BaseModel):
         imageGenerationConfig: Configuration for image generation.
     """
 
-    taskType: Literal[TaskType.COLOR_GUIDED_GENERATION] = TaskType.COLOR_GUIDED_GENERATION
+    taskType: Literal[
+        TaskType.COLOR_GUIDED_GENERATION
+    ] = TaskType.COLOR_GUIDED_GENERATION
     colorGuidedGenerationParams: ColorGuidedGenerationParams
     imageGenerationConfig: Optional[ImageGenerationConfig] = Field(
         default_factory=ImageGenerationConfig
@@ -219,13 +223,13 @@ class ColorGuidedRequest(BaseModel):
         """
         color_guided_params = self.colorGuidedGenerationParams.model_dump()
         # Remove negativeText if it's None
-        if color_guided_params.get('negativeText') is None:
-            color_guided_params.pop('negativeText', None)
+        if color_guided_params.get("negativeText") is None:
+            color_guided_params.pop("negativeText", None)
 
         return {
-            'taskType': self.taskType,
-            'colorGuidedGenerationParams': color_guided_params,
-            'imageGenerationConfig': self.imageGenerationConfig.model_dump()
+            "taskType": self.taskType,
+            "colorGuidedGenerationParams": color_guided_params,
+            "imageGenerationConfig": self.imageGenerationConfig.model_dump()
             if self.imageGenerationConfig
             else None,
         }

awslabs/nova_canvas_mcp_server/novacanvas.py

@@ -41,7 +41,7 @@ def save_generated_images(
     base64_images: List[str],
     filename: Optional[str] = None,
     number_of_images: int = DEFAULT_NUMBER_OF_IMAGES,
-    prefix: str = 'nova_canvas',
+    prefix: str = "nova_canvas",
     workspace_dir: Optional[str] = None,
 ) -> Dict[str, List]:
     """Save base64-encoded images to files.
@@ -56,7 +56,7 @@ def save_generated_images(
     Returns:
         Dictionary with lists of paths to the saved image files and PIL Image objects.
     """
-    logger.debug(f'Saving {len(base64_images)} images')
+    logger.debug(f"Saving {len(base64_images)} images")
     # Determine the output directory
     if workspace_dir:
         output_dir = os.path.join(workspace_dir, DEFAULT_OUTPUT_DIR)
@@ -73,25 +73,27 @@ def save_generated_images(
         # Generate filename if not provided
         if filename:
             image_filename = (
-                f'{filename}_{i + 1}.png' if number_of_images > 1 else f'{filename}.png'
+                f"{filename}_{i + 1}.png" if number_of_images > 1 else f"{filename}.png"
             )
         else:
             # Generate a random filename
-            random_id = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=8))
-            image_filename = f'{prefix}_{random_id}_{i + 1}.png'
+            random_id = "".join(
+                random.choices("abcdefghijklmnopqrstuvwxyz0123456789", k=8)
+            )
+            image_filename = f"{prefix}_{random_id}_{i + 1}.png"
 
         # Decode the base64 image data
         image_data = base64.b64decode(base64_image_data)
 
         # Save the image
         image_path = os.path.join(output_dir, image_filename)
-        with open(image_path, 'wb') as file:
+        with open(image_path, "wb") as file:
             file.write(image_data)
         # Convert to absolute path
         abs_image_path = os.path.abspath(image_path)
         saved_paths.append(abs_image_path)
 
-    return {'paths': saved_paths}
+    return {"paths": saved_paths}
 
 
 async def invoke_nova_canvas(
@@ -110,22 +112,24 @@ async def invoke_nova_canvas(
     Raises:
         Exception: If the API call fails.
     """
-    logger.debug('Invoking Nova Canvas API')
+    logger.debug("Invoking Nova Canvas API")
 
     # Convert the request payload to JSON
     request = json.dumps(request_model_dict)
 
     try:
         # Invoke the model
-        logger.info(f'Sending request to Nova Canvas model: {NOVA_CANVAS_MODEL_ID}')
-        response = bedrock_runtime_client.invoke_model(modelId=NOVA_CANVAS_MODEL_ID, body=request)
+        logger.info(f"Sending request to Nova Canvas model: {NOVA_CANVAS_MODEL_ID}")
+        response = bedrock_runtime_client.invoke_model(
+            modelId=NOVA_CANVAS_MODEL_ID, body=request
+        )
 
         # Decode the response body
-        result = json.loads(response['body'].read())
-        logger.info('Nova Canvas API call successful')
+        result = json.loads(response["body"].read())
+        logger.info("Nova Canvas API call successful")
         return result
     except Exception as e:
-        logger.error(f'Nova Canvas API call failed: {str(e)}')
+        logger.error(f"Nova Canvas API call failed: {str(e)}")
         raise
 
 
@@ -165,18 +169,22 @@ async def generate_image_with_text(
         ImageGenerationResponse: An object containing the paths to the generated images,
         PIL Image objects, and status information.
     """
-    logger.debug(f"Generating text-to-image with prompt: '{prompt[:30]}...' ({width}x{height})")
+    logger.debug(
+        f"Generating text-to-image with prompt: '{prompt[:30]}...' ({width}x{height})"
+    )
 
     try:
         # Validate input parameters using Pydantic
         try:
-            logger.debug('Validating parameters and creating request model')
+            logger.debug("Validating parameters and creating request model")
 
             # Create image generation config
             config = ImageGenerationConfig(
                 width=width,
                 height=height,
-                quality=Quality.STANDARD if quality == DEFAULT_QUALITY else Quality.PREMIUM,
+                quality=Quality.STANDARD
+                if quality == DEFAULT_QUALITY
+                else Quality.PREMIUM,
                 cfgScale=cfg_scale,
                 seed=seed if seed is not None else random.randint(0, 858993459),
                 numberOfImages=number_of_images,
@@ -185,7 +193,9 @@ async def generate_image_with_text(
             # Create text-to-image params
             # The Nova Canvas API doesn't accept null for negativeText
             if negative_prompt is not None:
-                text_params = TextToImageParams(text=prompt, negativeText=negative_prompt)
+                text_params = TextToImageParams(
+                    text=prompt, negativeText=negative_prompt
+                )
             else:
                 text_params = TextToImageParams(text=prompt)
 
@@ -196,13 +206,13 @@ async def generate_image_with_text(
 
             # Convert model to dictionary
             request_model_dict = request_model.to_api_dict()
-            logger.info('Request validation successful')
+            logger.info("Request validation successful")
 
         except Exception as e:
-            logger.error(f'Parameter validation failed: {str(e)}')
+            logger.error(f"Parameter validation failed: {str(e)}")
             return ImageGenerationResponse(
-                status='error',
-                message=f'Validation error: {str(e)}',
+                status="error",
+                message=f"Validation error: {str(e)}",
                 paths=[],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
@@ -210,34 +220,36 @@ async def generate_image_with_text(
 
         try:
             # Invoke the Nova Canvas API
-            logger.debug('Sending request to Nova Canvas API')
-            model_response = await invoke_nova_canvas(request_model_dict, bedrock_runtime_client)
+            logger.debug("Sending request to Nova Canvas API")
+            model_response = await invoke_nova_canvas(
+                request_model_dict, bedrock_runtime_client
+            )
 
             # Extract the image data
-            base64_images = model_response['images']
-            logger.info(f'Received {len(base64_images)} images from Nova Canvas API')
+            base64_images = model_response["images"]
+            logger.info(f"Received {len(base64_images)} images from Nova Canvas API")
 
             # Save the generated images
             result = save_generated_images(
                 base64_images,
                 filename,
                 number_of_images,
-                prefix='nova_canvas',
+                prefix="nova_canvas",
                 workspace_dir=workspace_dir,
             )
 
             logger.info(f'Successfully generated {len(result["paths"])} image(s)')
             return ImageGenerationResponse(
-                status='success',
+                status="success",
                 message=f'Generated {len(result["paths"])} image(s)',
-                paths=result['paths'],
+                paths=result["paths"],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
             )
         except Exception as e:
-            logger.error(f'Image generation failed: {str(e)}')
+            logger.error(f"Image generation failed: {str(e)}")
             return ImageGenerationResponse(
-                status='error',
+                status="error",
                 message=str(e),
                 paths=[],
                 prompt=prompt,
@@ -245,9 +257,9 @@ async def generate_image_with_text(
             )
 
     except Exception as e:
-        logger.error(f'Unexpected error in generate_image_with_text: {str(e)}')
+        logger.error(f"Unexpected error in generate_image_with_text: {str(e)}")
         return ImageGenerationResponse(
-            status='error',
+            status="error",
             message=str(e),
             paths=[],
             prompt=prompt,
@@ -300,13 +312,17 @@ async def generate_image_with_colors(
     try:
         # Validate input parameters using Pydantic
         try:
-            logger.debug('Validating parameters and creating color-guided request model')
+            logger.debug(
+                "Validating parameters and creating color-guided request model"
+            )
 
             # Create image generation config
             config = ImageGenerationConfig(
                 width=width,
                 height=height,
-                quality=Quality.STANDARD if quality == DEFAULT_QUALITY else Quality.PREMIUM,
+                quality=Quality.STANDARD
+                if quality == DEFAULT_QUALITY
+                else Quality.PREMIUM,
                 cfgScale=cfg_scale,
                 seed=seed if seed is not None else random.randint(0, 858993459),
                 numberOfImages=number_of_images,
@@ -333,13 +349,13 @@ async def generate_image_with_colors(
 
             # Convert model to dictionary
             request_model_dict = request_model.to_api_dict()
-            logger.info('Color-guided request validation successful')
+            logger.info("Color-guided request validation successful")
 
         except Exception as e:
-            logger.error(f'Color-guided parameter validation failed: {str(e)}')
+            logger.error(f"Color-guided parameter validation failed: {str(e)}")
             return ImageGenerationResponse(
-                status='error',
-                message=f'Validation error: {str(e)}',
+                status="error",
+                message=f"Validation error: {str(e)}",
                 paths=[],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
@@ -348,35 +364,39 @@ async def generate_image_with_colors(
 
         try:
             # Invoke the Nova Canvas API
-            logger.debug('Sending color-guided request to Nova Canvas API')
-            model_response = await invoke_nova_canvas(request_model_dict, bedrock_runtime_client)
+            logger.debug("Sending color-guided request to Nova Canvas API")
+            model_response = await invoke_nova_canvas(
+                request_model_dict, bedrock_runtime_client
+            )
 
             # Extract the image data
-            base64_images = model_response['images']
-            logger.info(f'Received {len(base64_images)} images from Nova Canvas API')
+            base64_images = model_response["images"]
+            logger.info(f"Received {len(base64_images)} images from Nova Canvas API")
 
             # Save the generated images
             result = save_generated_images(
                 base64_images,
                 filename,
                 number_of_images,
-                prefix='nova_canvas_color',
+                prefix="nova_canvas_color",
                 workspace_dir=workspace_dir,
             )
 
-            logger.info(f'Successfully generated {len(result["paths"])} color-guided image(s)')
+            logger.info(
+                f'Successfully generated {len(result["paths"])} color-guided image(s)'
+            )
             return ImageGenerationResponse(
-                status='success',
+                status="success",
                 message=f'Generated {len(result["paths"])} image(s)',
-                paths=result['paths'],
+                paths=result["paths"],
                 prompt=prompt,
                 negative_prompt=negative_prompt,
                 colors=colors,
             )
         except Exception as e:
-            logger.error(f'Color-guided image generation failed: {str(e)}')
+            logger.error(f"Color-guided image generation failed: {str(e)}")
             return ImageGenerationResponse(
-                status='error',
+                status="error",
                 message=str(e),
                 paths=[],
                 prompt=prompt,
@@ -385,9 +405,9 @@ async def generate_image_with_colors(
             )
 
     except Exception as e:
-        logger.error(f'Unexpected error in generate_image_with_colors: {str(e)}')
+        logger.error(f"Unexpected error in generate_image_with_colors: {str(e)}")
         return ImageGenerationResponse(
-            status='error',
+            status="error",
             message=str(e),
             paths=[],
             prompt=prompt,

awslabs/nova_canvas_mcp_server/server.py

@@ -25,7 +25,7 @@ from typing import TYPE_CHECKING, List, Optional
 
 # Logging
 logger.remove()
-logger.add(sys.stderr, level=os.getenv('FASTMCP_LOG_LEVEL', 'WARNING'))
+logger.add(sys.stderr, level=os.getenv("FASTMCP_LOG_LEVEL", "WARNING"))
 
 # Bedrock Runtime Client typing
 if TYPE_CHECKING:
@@ -36,23 +36,25 @@ else:
 
 # Bedrock Runtime Client
 bedrock_runtime_client: BedrockRuntimeClient
-aws_region: str = os.environ.get('AWS_REGION', 'us-east-1')
+aws_region: str = os.environ.get("AWS_REGION", "us-east-1")
 
 try:
-    if aws_profile := os.environ.get('AWS_PROFILE'):
+    if aws_profile := os.environ.get("AWS_PROFILE"):
         bedrock_runtime_client = boto3.Session(
             profile_name=aws_profile, region_name=aws_region
-        ).client('bedrock-runtime')
+        ).client("bedrock-runtime")
     else:
-        bedrock_runtime_client = boto3.Session(region_name=aws_region).client('bedrock-runtime')
+        bedrock_runtime_client = boto3.Session(region_name=aws_region).client(
+            "bedrock-runtime"
+        )
 except Exception as e:
-    logger.error(f'Error creating bedrock runtime client: {str(e)}')
+    logger.error(f"Error creating bedrock runtime client: {str(e)}")
     raise
 
 
 # Create the MCP server Pwith detailed instructions
 mcp = FastMCP(
-    'awslabs-nova-canvas-mcp-server',
+    "awslabs-nova-canvas-mcp-server",
     instructions=f"""
 # Amazon Nova Canvas Image Generation
 
@@ -71,32 +73,33 @@ Generate an image from a text prompt and color palette using Amazon Nova Canvas.
 {PROMPT_INSTRUCTIONS}
 """,
     dependencies=[
-        'pydantic',
-        'boto3',
+        "pydantic",
+        "boto3",
     ],
 )
 
 
-@mcp.tool(name='generate_image')
+@mcp.tool(name="generate_image")
 async def mcp_generate_image(
     ctx: Context,
     prompt: str = Field(
-        description='The text description of the image to generate (1-1024 characters)'
+        description="The text description of the image to generate (1-1024 characters)"
     ),
     negative_prompt: Optional[str] = Field(
         default=None,
-        description='Text to define what not to include in the image (1-1024 characters)',
+        description="Text to define what not to include in the image (1-1024 characters)",
     ),
     filename: Optional[str] = Field(
-        default=None, description='The name of the file to save the image to (without extension)'
+        default=None,
+        description="The name of the file to save the image to (without extension)",
     ),
     width: int = Field(
         default=DEFAULT_WIDTH,
-        description='The width of the generated image (320-4096, divisible by 16)',
+        description="The width of the generated image (320-4096, divisible by 16)",
     ),
     height: int = Field(
         default=DEFAULT_HEIGHT,
-        description='The height of the generated image (320-4096, divisible by 16)',
+        description="The height of the generated image (320-4096, divisible by 16)",
     ),
     quality: str = Field(
         default=DEFAULT_QUALITY,
@@ -104,11 +107,14 @@ async def mcp_generate_image(
     ),
     cfg_scale: float = Field(
         default=DEFAULT_CFG_SCALE,
-        description='How strongly the image adheres to the prompt (1.1-10.0)',
+        description="How strongly the image adheres to the prompt (1.1-10.0)",
+    ),
+    seed: Optional[int] = Field(
+        default=None, description="Seed for generation (0-858,993,459)"
     ),
-    seed: Optional[int] = Field(default=None, description='Seed for generation (0-858,993,459)'),
     number_of_images: int = Field(
-        default=DEFAULT_NUMBER_OF_IMAGES, description='The number of images to generate (1-5)'
+        default=DEFAULT_NUMBER_OF_IMAGES,
+        description="The number of images to generate (1-5)",
     ),
     workspace_dir: Optional[str] = Field(
         default=None,
@@ -155,7 +161,7 @@ async def mcp_generate_image(
 
     try:
         logger.info(
-            f'Generating image with text prompt, quality: {quality}, cfg_scale: {cfg_scale}'
+            f"Generating image with text prompt, quality: {quality}, cfg_scale: {cfg_scale}"
         )
         response = await generate_image_with_text(
             prompt=prompt,
@@ -171,54 +177,62 @@ async def mcp_generate_image(
             workspace_dir=workspace_dir,
         )
 
-        if response.status == 'success':
+        if response.status == "success":
             # return response.paths
             return McpImageGenerationResponse(
-                status='success',
-                paths=[f'file://{path}' for path in response.paths],
+                status="success",
+                paths=[f"file://{path}" for path in response.paths],
             )
         else:
-            logger.error(f'Image generation returned error status: {response.message}')
-            await ctx.error(f'Failed to generate image: {response.message}')  # type: ignore
+            logger.error(f"Image generation returned error status: {response.message}")
+            await ctx.error(f"Failed to generate image: {response.message}")  # type: ignore
             # Return empty image or raise exception based on requirements
-            raise Exception(f'Failed to generate image: {response.message}')
+            raise Exception(f"Failed to generate image: {response.message}")
     except Exception as e:
-        logger.error(f'Error in mcp_generate_image: {str(e)}')
-        await ctx.error(f'Error generating image: {str(e)}')  # type: ignore
+        logger.error(f"Error in mcp_generate_image: {str(e)}")
+        await ctx.error(f"Error generating image: {str(e)}")  # type: ignore
         raise
 
 
-@mcp.tool(name='generate_image_with_colors')
+@mcp.tool(name="generate_image_with_colors")
 async def mcp_generate_image_with_colors(
     ctx: Context,
     prompt: str = Field(
-        description='The text description of the image to generate (1-1024 characters)'
+        description="The text description of the image to generate (1-1024 characters)"
     ),
     colors: List[str] = Field(
         description='List of up to 10 hexadecimal color values (e.g., "#FF9800")'
     ),
     negative_prompt: Optional[str] = Field(
         default=None,
-        description='Text to define what not to include in the image (1-1024 characters)',
+        description="Text to define what not to include in the image (1-1024 characters)",
     ),
     filename: Optional[str] = Field(
-        default=None, description='The name of the file to save the image to (without extension)'
+        default=None,
+        description="The name of the file to save the image to (without extension)",
     ),
     width: int = Field(
-        default=1024, description='The width of the generated image (320-4096, divisible by 16)'
+        default=1024,
+        description="The width of the generated image (320-4096, divisible by 16)",
     ),
     height: int = Field(
-        default=1024, description='The height of the generated image (320-4096, divisible by 16)'
+        default=1024,
+        description="The height of the generated image (320-4096, divisible by 16)",
     ),
     quality: str = Field(
-        default='standard',
+        default="standard",
         description='The quality of the generated image ("standard" or "premium")',
     ),
     cfg_scale: float = Field(
-        default=6.5, description='How strongly the image adheres to the prompt (1.1-10.0)'
+        default=6.5,
+        description="How strongly the image adheres to the prompt (1.1-10.0)",
+    ),
+    seed: Optional[int] = Field(
+        default=None, description="Seed for generation (0-858,993,459)"
+    ),
+    number_of_images: int = Field(
+        default=1, description="The number of images to generate (1-5)"
     ),
-    seed: Optional[int] = Field(default=None, description='Seed for generation (0-858,993,459)'),
-    number_of_images: int = Field(default=1, description='The number of images to generate (1-5)'),
     workspace_dir: Optional[str] = Field(
         default=None,
         description="The current workspace directory where the image should be saved. CRITICAL: Assistant must always provide this parameter to save images to the user's current project.",
@@ -260,9 +274,9 @@ async def mcp_generate_image_with_colors(
     )
 
     try:
-        color_hex_list = ', '.join(colors[:3]) + (', ...' if len(colors) > 3 else '')
+        color_hex_list = ", ".join(colors[:3]) + (", ..." if len(colors) > 3 else "")
         logger.info(
-            f'Generating color-guided image with colors: [{color_hex_list}], quality: {quality}'
+            f"Generating color-guided image with colors: [{color_hex_list}], quality: {quality}"
         )
 
         response = await generate_image_with_colors(
@@ -280,45 +294,51 @@ async def mcp_generate_image_with_colors(
             workspace_dir=workspace_dir,
         )
 
-        if response.status == 'success':
+        if response.status == "success":
             return McpImageGenerationResponse(
-                status='success',
-                paths=[f'file://{path}' for path in response.paths],
+                status="success",
+                paths=[f"file://{path}" for path in response.paths],
             )
         else:
             logger.error(
-                f'Color-guided image generation returned error status: {response.message}'
+                f"Color-guided image generation returned error status: {response.message}"
+            )
+            await ctx.error(
+                f"Failed to generate color-guided image: {response.message}"
+            )
+            raise Exception(
+                f"Failed to generate color-guided image: {response.message}"
             )
-            await ctx.error(f'Failed to generate color-guided image: {response.message}')
-            raise Exception(f'Failed to generate color-guided image: {response.message}')
     except Exception as e:
-        logger.error(f'Error in mcp_generate_image_with_colors: {str(e)}')
-        await ctx.error(f'Error generating color-guided image: {str(e)}')
+        logger.error(f"Error in mcp_generate_image_with_colors: {str(e)}")
+        await ctx.error(f"Error generating color-guided image: {str(e)}")
         raise
 
 
 def main():
     """Run the MCP server with CLI argument support."""
-    logger.info('Starting nova-canvas-mcp-server MCP server')
+    logger.info("Starting nova-canvas-mcp-server MCP server")
 
     parser = argparse.ArgumentParser(
-        description='MCP server for generating images using Amazon Nova Canvas'
+        description="MCP server for generating images using Amazon Nova Canvas"
+    )
+    parser.add_argument("--sse", action="store_true", help="Use SSE transport")
+    parser.add_argument(
+        "--port", type=int, default=8888, help="Port to run the server on"
     )
-    parser.add_argument('--sse', action='store_true', help='Use SSE transport')
-    parser.add_argument('--port', type=int, default=8888, help='Port to run the server on')
 
     args = parser.parse_args()
-    logger.debug(f'Parsed arguments: sse={args.sse}, port={args.port}')
+    logger.debug(f"Parsed arguments: sse={args.sse}, port={args.port}")
 
     # Run server with appropriate transport
     if args.sse:
-        logger.info(f'Using SSE transport on port {args.port}')
+        logger.info(f"Using SSE transport on port {args.port}")
         mcp.settings.port = args.port
-        mcp.run(transport='sse')
+        mcp.run(transport="sse")
     else:
-        logger.info('Using standard stdio transport')
+        logger.info("Using standard stdio transport")
         mcp.run()
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

{awslabs_nova_canvas_mcp_server-0.1.10233.dist-info → awslabs_nova_canvas_mcp_server-0.1.10652.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: awslabs.nova-canvas-mcp-server
-Version: 0.1.10233
+Version: 0.1.10652
 Summary: An AWS Labs Model Context Protocol (MCP) server for Amazon Nova Canvas
 Requires-Python: >=3.13
 Requires-Dist: boto3>=1.37.24
@@ -9,7 +9,7 @@ Requires-Dist: mcp[cli]>=1.6.0
 Requires-Dist: pydantic>=2.11.1
 Description-Content-Type: text/markdown
 
-# awslabs MCP Nova Canvas Server
+# Nova Canvas MCP Server
 
 MCP server for generating images using Amazon Nova Canvas
 

awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/RECORD

@@ -0,0 +1,10 @@
+awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
+awslabs/nova_canvas_mcp_server/__init__.py,sha256=W7bzpkX1o3FURvzhQujITA3csYdIPA41UrCBnl99m8A,60
+awslabs/nova_canvas_mcp_server/consts.py,sha256=jWKxZvmHi_5l2IULP5mHJLI3FLTx_4Y_KmC390ySWvA,2590
+awslabs/nova_canvas_mcp_server/models.py,sha256=VxqfLSAfe3MiNWDP61tKu3UUOzgwqE2ko0U5n4Fxz-E,9963
+awslabs/nova_canvas_mcp_server/novacanvas.py,sha256=pLnEB4qmQ3_IO2VgatmTi164g1NEhS11UkTL0uVjiEk,15305
+awslabs/nova_canvas_mcp_server/server.py,sha256=mNN8MHLnpz9ZrGYZFTismDM0pRWsoQ9lknTmaPapgN0,12464
+awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/METADATA,sha256=EtSFRGpmUQrtJgGMmTNdyNRNrVIUbh7VTHPXyytglbI,2981
+awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/entry_points.txt,sha256=v8V4vn8YuugOSL7w_sUxz-M0EDZNZU2_ydJZDd31pGI,94
+awslabs_nova_canvas_mcp_server-0.1.10652.dist-info/RECORD,,

awslabs_nova_canvas_mcp_server-0.1.10233.dist-info/RECORD

@@ -1,10 +0,0 @@
-awslabs/__init__.py,sha256=4zfFn3N0BkvQmMTAIvV_QAbKp6GWzrwaUN17YeRoChM,115
-awslabs/nova_canvas_mcp_server/__init__.py,sha256=D1JLDjoqRWgJm28RaKjBtIzAsuc31Ilg8r5-hv8I0ZU,60
-awslabs/nova_canvas_mcp_server/consts.py,sha256=1qnIsWXKsg7R8JpWalgns0vPmBAHu6f9oI8hylhBuuo,2590
-awslabs/nova_canvas_mcp_server/models.py,sha256=tYJeeTKhU_6OJxKJBvKyAzgC46-Dexx_o5up539jqi4,9935
-awslabs/nova_canvas_mcp_server/novacanvas.py,sha256=ltHqnH5slEfq52jF69xE59pam1pEXmIRaGYnWFtfC04,15017
-awslabs/nova_canvas_mcp_server/server.py,sha256=pZBwGwOU_3CoSfi9PHnfBmqI7g6cYqRGCTeF3Fg-sIk,12272
-awslabs_nova_canvas_mcp_server-0.1.10233.dist-info/METADATA,sha256=QsFHmvWwYAiuaPC3UNzBKI_6nJ0C0cjNT50KgDHmB-4,2989
-awslabs_nova_canvas_mcp_server-0.1.10233.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-awslabs_nova_canvas_mcp_server-0.1.10233.dist-info/entry_points.txt,sha256=v8V4vn8YuugOSL7w_sUxz-M0EDZNZU2_ydJZDd31pGI,94
-awslabs_nova_canvas_mcp_server-0.1.10233.dist-info/RECORD,,