Clean up

Signed-off-by: Aivin V. Solatorio <avsolatorio@gmail.com>

wdi_mcp_server.py

@@ -69,41 +69,6 @@ def get_top_k(query: str, top_k: int = 10, fields: list[str] | None = None):
     return df.iloc[idx][fields].to_dict("records")
 
 
-# @mcp.tool()
-# async def generate_image(prompt: str, width: int = 512, height: int = 512) -> str:
-#     """Generate an image using SanaSprint model.
-
-#     Args:
-#         prompt: Text prompt describing the image to generate
-#         width: Image width (default: 512)
-#         height: Image height (default: 512)
-#     """
-#     client = Client("https://ysharma-sanasprint.hf.space/")
-
-#     try:
-#         result = client.predict(
-#             prompt, "0.6B", 0, True, width, height, 4.0, 2, api_name="/infer"
-#         )
-
-#         if isinstance(result, list) and len(result) >= 1:
-#             image_data = result[0]
-#             if isinstance(image_data, dict) and "url" in image_data:
-#                 return json.dumps(
-#                     {
-#                         "type": "image",
-#                         "url": image_data["url"],
-#                         "message": f"Generated image for prompt: {prompt}",
-#                     }
-#                 )
-
-#         return json.dumps({"type": "error", "message": "Failed to generate image"})
-
-#     except Exception as e:
-#         return json.dumps(
-#             {"type": "error", "message": f"Error generating image: {str(e)}"}
-#         )
-
-
 class SearchOutput(BaseModel):
     idno: str = Field(..., description="The unique identifier of the indicator.")
     name: str = Field(..., description="The name of the indicator.")
@@ -175,7 +140,6 @@ async def get_wdi_data(
     Returns:
         A dictionary with keys `data` and `note`. The `data` key contains a list of indicator data entries requested. The `note` key contains a note about the data returned.
     """
-    print("Hello...")
     MAX_INFO = 20
     note = ""