"""
Utility functions for the vibe-reader application
"""

import json
import re
from typing import Dict, Any, Optional


def parse_json_response(response: str) -> Optional[Dict[str, Any]]:
    """
    Parse JSON from LLM response, handling various formats

    Args:
        response: Raw LLM response that may contain JSON

    Returns:
        Parsed JSON dict, or None if parsing fails
    """
    # Remove markdown code blocks if present
    cleaned = re.sub(r'```json\s*|\s*```', '', response, flags=re.IGNORECASE)
    cleaned = cleaned.strip()

    # Try to find JSON object in the response
    json_match = re.search(r'\{.*\}', cleaned, re.DOTALL)
    if json_match:
        try:
            return json.loads(json_match.group(0))
        except json.JSONDecodeError as e:
            print(f"JSON parsing error: {e}")
            return None

    return None


def extract_vibe_components(vibe_json: Dict[str, Any]) -> Dict[str, Any]:
    """
    Extract and validate vibe components from parsed JSON

    Args:
        vibe_json: Parsed JSON from vibe extraction

    Returns:
        Dictionary with validated vibe components
    """
    return {
        "aesthetic_genre_keywords": vibe_json.get("aesthetic_genre_keywords", []),
        "mood_atmosphere": vibe_json.get("mood_atmosphere", []),
        "core_themes": vibe_json.get("core_themes", []),
        "tropes": vibe_json.get("tropes", []),
        "feels_like": vibe_json.get("feels_like", "")
    }


def strip_thinking_tags(text: str) -> str:
    """
    Remove <think>...</think> tags and any reasoning content from text
    Qwen3 uses standard XML format: <think>...</think>

    Args:
        text: Text that may contain thinking tags

    Returns:
        Clean text without thinking tags
    """
    # Remove <think>...</think> blocks
    cleaned = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL | re.IGNORECASE)
    # Remove any leftover tags
    cleaned = re.sub(r'</?think>', '', cleaned, flags=re.IGNORECASE)
    return cleaned.strip()