Create promt_generator.py

promt_generator.py

@@ -0,0 +1,73 @@
+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