wordpress_media tool

app.py

@@ -11,6 +11,7 @@ from tools.blog_generator import BlogGeneratorTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
 from tools.html_to_wp_blocks import HTMLToWPBlocksTool
+from tools.wordpress_media import WordPressMediaTool
 
 from Gradio_UI import GradioUI
 
@@ -156,6 +157,7 @@ with open("prompts.yaml", 'r') as stream:
 
 # Initialize WordPress tools
 wordpress_post = WordPressPostTool(wordpress_credentials)
+wordpress_media = WordPressMediaTool(wordpress_credentials)
 blog_generator = BlogGeneratorTool(model=model)
 html_to_blocks = HTMLToWPBlocksTool()
 
@@ -168,6 +170,7 @@ agent = CodeAgent(
         quick_research,
         # get_weather,
         wordpress_post,
+        wordpress_media,
         blog_generator,
         html_to_blocks,
         visit_webpage,
@@ -187,6 +190,7 @@ prompt_templates["system_prompt"] += """
 
 You are also capable of managing a WordPress blog through the following tools:
 - wordpress_post: Publishes posts to WordPress
+- wordpress_media: Uploads media files to WordPress and returns attachment URLs
 - blog_generator: Generates AI-written blog posts
 - html_to_blocks: Converts HTML content to WordPress Gutenberg blocks format
 
@@ -195,11 +199,14 @@ Always check credentials before attempting to post content.
 Example WordPress workflow:
 1. Set credentials (first time only)
 2. Generate blog content
-3. Convert HTML content to WordPress blocks format
-4. Publish blocks-formatted content to WordPress
+3. Generate or prepare media files
+4. Upload media files to WordPress using wordpress_media tool
+5. Convert HTML content to WordPress blocks format, including media blocks
+6. Publish blocks-formatted content to WordPress
 
 Remember to:
 - ALWAYS convert HTML content to WordPress blocks format before publishing
+- Upload any media files before referencing them in posts
 - Validate WordPress credentials before posting
 - Generate high-quality, relevant content
 - Handle errors gracefully

tools/wordpress_media.py

@@ -0,0 +1,103 @@
+import requests
+from requests.auth import HTTPBasicAuth
+from typing import Dict, Optional
+import os
+import mimetypes
+from smolagents.tools import Tool
+
+
+class WordPressMediaTool(Tool):
+    name = "wordpress_media"
+    description = "Uploads media files to WordPress and returns the attachment URL"
+    inputs = {
+        'file_path': {'type': 'string', 'description': 'Local path to the media file to upload'},
+        'title': {
+            'type': 'string',
+            'description': 'Title for the media attachment',
+            'nullable': True
+        },
+        'alt_text': {
+            'type': 'string',
+            'description': 'Alt text for the media attachment',
+            'nullable': True
+        }
+    }
+    output_type = "object"
+
+    def __init__(self, wordpress_credentials):
+        super().__init__()
+        self.credentials = wordpress_credentials
+
+    def forward(self, file_path: str, title: str = None, alt_text: str = None) -> Dict:
+        """Uploads a media file to WordPress
+        Args:
+            file_path: Path to the media file
+            title: Optional title for the media attachment
+            alt_text: Optional alt text for the media attachment
+        Returns:
+            Dict containing the response from WordPress with media details
+        """
+        if not self.credentials:
+            return {"error": "WordPress credentials not configured"}
+
+        if not os.path.exists(file_path):
+            return {"error": f"File not found: {file_path}"}
+
+        api_url = f"{self.credentials['site_url']}/wp-json/wp/v2/media"
+        auth = HTTPBasicAuth(
+            self.credentials['username'], self.credentials['app_password'])
+
+        # Determine mime type
+        mime_type, _ = mimetypes.guess_type(file_path)
+        if not mime_type:
+            return {"error": "Could not determine file type"}
+
+        # Prepare headers
+        headers = {
+            'Content-Type': mime_type,
+            'Content-Disposition': f'attachment; filename="{os.path.basename(file_path)}"'
+        }
+
+        try:
+            # Upload file
+            with open(file_path, 'rb') as file:
+                response = requests.post(
+                    api_url,
+                    headers=headers,
+                    auth=auth,
+                    data=file
+                )
+                response.raise_for_status()
+                response_data = response.json()
+
+            # Update media properties if title or alt_text provided
+            if title or alt_text:
+                media_id = response_data['id']
+                update_url = f"{api_url}/{media_id}"
+                update_data = {}
+
+                if title:
+                    update_data['title'] = {'rendered': title}
+                if alt_text:
+                    update_data['alt_text'] = alt_text
+
+                if update_data:
+                    update_response = requests.post(
+                        update_url,
+                        json=update_data,
+                        auth=auth
+                    )
+                    update_response.raise_for_status()
+                    response_data = update_response.json()
+
+            return {
+                "status": "success",
+                "attachment_id": response_data.get("id"),
+                "url": response_data.get("source_url"),
+                "title": response_data.get("title", {}).get("rendered"),
+                "alt_text": response_data.get("alt_text")
+            }
+
+        except requests.exceptions.RequestException as e:
+            print(f"Failed to upload media: {str(e)}")
+            return {"error": f"Failed to upload media: {str(e)}"}