feat: added the change format function, and a visulize image function

mcp_server.py

@@ -1,5 +1,8 @@
 from mcp.server.fastmcp import FastMCP
-from src.securty.prompt_injection import check_prompt_injection
+from src.utils.change_format import change_format
+from src.utils.image_helpers import remove_background_from_url
+from src.utils.visualize_image import visualize_base64_image
+
 mcp = FastMCP("Youtube Service")
 
 @mcp.tool()
@@ -15,7 +18,9 @@ def say_hello(name: str) -> str:
   """
   return f"Hello, {name}!"
 
-mcp.add_tool(check_prompt_injection)
+mcp.add_tool(remove_background_from_url)
+mcp.add_tool(change_format)
+mcp.add_tool(visualize_base64_image)
 
 if __name__ == "__main__":
     mcp.run()

src/utils/change_format.py

@@ -0,0 +1,32 @@
+from PIL import Image
+from io import BytesIO
+import requests
+import base64
+
+def change_format(image_url: str, target_format: str) -> str:
+    """
+    Change the format of an image from a URL to the specified target format.
+    
+    Args:
+        image_url: The URL of the input image.
+        target_format: The desired output format (e.g., 'JPEG', 'PNG').
+        
+    Returns:
+        The image converted to the target format as a base64-encoded string.
+    """
+    
+    response = requests.get(image_url, timeout=30)
+    response.raise_for_status()
+    
+    # Open the image from bytes
+    img = Image.open(BytesIO(response.content))
+    
+    # Convert the image to the target format
+    output = BytesIO()
+    img.save(output, format=target_format)
+    output.seek(0)
+    
+    # Convert to base64 string for JSON serialization
+    encoded_image = base64.b64encode(output.getvalue()).decode('utf-8')
+    
+    return encoded_image  # Return base64 encoded string that can be serialized to JSON

src/utils/visualize_image.py

@@ -0,0 +1,18 @@
+import base64
+from PIL import Image
+from io import BytesIO
+
+def visualize_base64_image(base64_string:str):
+    """
+    Visualize a base64-encoded image string.
+    
+    Args:
+        base64_string: The base64-encoded image string.
+    """
+    # Decode the base64 string back to binary
+    image_data = base64.b64decode(base64_string)
+    
+    # Create an image from the binary data
+    img = Image.open(BytesIO(image_data))
+    
+    img.show()