wehub-resource-sync
91 lines
3.2 KiB
Python
91 lines
3.2 KiB
Python
"""
|
|
Template Manager for Section Extraction
|
|
|
|
This module provides functionality to load and render Jinja templates for
|
|
section-specific resume extraction prompts.
|
|
"""
|
|
|
|
import os
|
|
from typing import Dict, Optional
|
|
from jinja2 import Environment, FileSystemLoader, Template
|
|
|
|
|
|
class TemplateManager:
|
|
"""
|
|
Manages Jinja templates for section-specific resume extraction.
|
|
|
|
This class provides functionality to load and render templates for
|
|
different resume sections (basics, work, education, skills, projects, awards).
|
|
"""
|
|
|
|
def __init__(self, template_dir: str = "prompts/templates"):
|
|
"""
|
|
Initialize the template manager.
|
|
|
|
Args:
|
|
template_dir (str): Directory containing Jinja templates
|
|
"""
|
|
self.template_dir = template_dir
|
|
self.env = Environment(
|
|
loader=FileSystemLoader(template_dir), trim_blocks=True, lstrip_blocks=True
|
|
)
|
|
self._templates: Dict[str, Template] = {}
|
|
self._load_templates()
|
|
|
|
def _load_templates(self):
|
|
"""Load all available templates."""
|
|
template_files = {
|
|
"basics": "basics.jinja",
|
|
"work": "work.jinja",
|
|
"education": "education.jinja",
|
|
"skills": "skills.jinja",
|
|
"projects": "projects.jinja",
|
|
"awards": "awards.jinja",
|
|
"system_message": "system_message.jinja",
|
|
"github_project_selection": "github_project_selection.jinja",
|
|
"resume_evaluation_criteria": "resume_evaluation_criteria.jinja",
|
|
"resume_evaluation_system_message": "resume_evaluation_system_message.jinja",
|
|
}
|
|
|
|
for section_name, filename in template_files.items():
|
|
try:
|
|
template_path = os.path.join(self.template_dir, filename)
|
|
if os.path.exists(template_path):
|
|
self._templates[section_name] = self.env.get_template(filename)
|
|
else:
|
|
print(f"⚠️ Template file not found: {template_path}")
|
|
except Exception as e:
|
|
print(f"❌ Error loading template {filename}: {e}")
|
|
|
|
def get_available_sections(self) -> list:
|
|
"""
|
|
Get list of available section names.
|
|
|
|
Returns:
|
|
list: List of available section names
|
|
"""
|
|
return list(self._templates.keys())
|
|
|
|
def render_template(self, section_name: str, **kwargs) -> Optional[str]:
|
|
"""
|
|
Render a template for a specific section.
|
|
|
|
Args:
|
|
section_name (str): Name of the section (basics, work, education, etc.)
|
|
**kwargs: Template variables (e.g., text_content)
|
|
|
|
Returns:
|
|
Optional[str]: Rendered template string, or None if template not found
|
|
"""
|
|
if section_name not in self._templates:
|
|
print(f"❌ Template not found for section: {section_name}")
|
|
print(f"Available sections: {self.get_available_sections()}")
|
|
return None
|
|
|
|
try:
|
|
template = self._templates[section_name]
|
|
return template.render(**kwargs)
|
|
except Exception as e:
|
|
print(f"❌ Error rendering template for {section_name}: {e}")
|
|
return None
|