import re

class PromptGenerator:
    @staticmethod
    def detect_intent(prompt: str) -> dict:
        """Detecta la intención del usuario basado en el prompt"""
        prompt_lower = prompt.lower().strip()
        
        # Detección de código
        code_keywords = [
            'código', 'code', 'programa', 'function', 'def ', 'import ',
            'python', 'javascript', 'java', 'c++', 'html', 'css', 'sql',
            'algoritmo', 'loop', 'for ', 'while ', 'if ', 'else',
            'variable', 'clase', 'class ', 'función', 'method'
        ]
        
        is_code = any(keyword in prompt_lower for keyword in code_keywords)
        
        # Detección de tipo de consulta
        if any(word in prompt_lower for word in ['explica', 'explicar', 'qué es', 'qué son', 'defin']):
            intent_type = "explication"
        elif any(word in prompt_lower for word in ['ejemplo', 'ejemplifica', 'muestra']):
            intent_type = "example"
        elif any(word in prompt_lower for word in ['corrige', 'error', 'bug', 'problema']):
            intent_type = "correction"
        else:
            intent_type = "general"
        
        return {
            "is_code": is_code,
            "type": intent_type,
            "language": PromptGenerator._detect_language(prompt_lower)
        }
    
    @staticmethod
    def _detect_language(prompt: str) -> str:
        """Detecta el lenguaje de programación mencionado"""
        languages = {
            'python': ['python', 'py'],
            'javascript': ['javascript', 'js', 'node'],
            'java': ['java'],
            'html': ['html'],
            'css': ['css'],
            'sql': ['sql', 'mysql', 'postgresql'],
            'c++': ['c++', 'cpp'],
            'c#': ['c#', 'csharp']
        }
        
        for lang, keywords in languages.items():
            if any(keyword in prompt for keyword in keywords):
                return lang
        return "unknown"
    
    @staticmethod
    def enhance_prompt(original_prompt: str, intent: dict) -> str:
        """Mejora el prompt basado en la intención detectada"""
        if intent["is_code"]:
            if intent["type"] == "explication":
                return f"Explica detalladamente este código: {original_prompt}"
            elif intent["type"] == "example":
                lang = intent["language"] if intent["language"] != "unknown" else "Python"
                return f"Da un ejemplo en {lang}: {original_prompt}"
            elif intent["type"] == "correction":
                return f"Corrige este código: {original_prompt}"
            else:
                return f"Responde sobre programación: {original_prompt}"
        else:
            if intent["type"] == "explication":
                return f"Explica claramente: {original_prompt}"
            elif intent["type"] == "example":
                return f"Proporciona ejemplos: {original_prompt}"
            else:
                return original_prompt