from smolagents import tool
from typing import List, Dict
import re

@tool
def generate_blog_section(topic: str, section_title: str) -> str:
    """Generates content for a specific section of the blog post
    Args:
        topic: The main topic of the blog post
        section_title: The title of the section to generate
    """
    return f"Generated content for section '{section_title}' about {topic}"

@tool
def improve_writing_style(text: str, style: str = "professional") -> str:
    """Improves the writing style of the given text
    Args:
        text: The text to improve
        style: The desired writing style (e.g., professional, casual, academic)
    """
    return f"Improved version of the text in {style} style"

@tool
def check_readability(text: str) -> Dict:
    """Analyzes the readability of the text
    Args:
        text: The text to analyze
    """
    # Simple implementation - you can make this more sophisticated
    words = len(text.split())
    sentences = len(re.split(r'[.!?]+', text))
    avg_words_per_sentence = words / max(sentences, 1)
    
    return {
        "word_count": words,
        "sentence_count": sentences,
        "avg_words_per_sentence": avg_words_per_sentence,
        "readability_score": "Good" if avg_words_per_sentence < 20 else "Complex"
    }

@tool
def generate_seo_metadata(title: str, content: str) -> Dict:
    """Generates SEO metadata for the blog post
    Args:
        title: The blog post title
        content: The blog post content
    """
    return {
        "meta_description": f"An informative article about {title}",
        "keywords": [word.lower() for word in title.split()],
        "suggested_title_tags": [f"<h1>{title}</h1>"]
    }