// Load Modules
const express = require('express');
const multer = require('multer');
const pdf = require('pdf-parse');
const { Document, Packer, Paragraph, TextRun } = require('docx');
const Anthropic = require('@anthropic-ai/sdk');
// Load Environment Variables
require('dotenv').config();
const router = express.Router();
const upload = multer({ storage: multer.memoryStorage() });
// initialize LLM
//TODO Update to have User Provide their own key
const anthropic = new Anthropic({
apiKey: process.env.ANTHROPIC_API_KEY,
});
// Handle Resume upload and user input for Job Description
router.post('/', upload.single('resume'), async (req, res) =>{
try{
if (!req.file || !req.body.jobDescription) {
return res.status(400).json({ error: 'Resume and Job Description are required' });
}
let resumeText = req.file.buffer.toString('utf-8');
const jobDescription = req.body.jobDescription;
if (req.file.mimetype === 'application/pdf') {
const pdfData = await pdf(req.file.buffer);
resumeText = pdfData.text;
}
// Load the LLM APi Messages
// These include placeholders to be replaced later in function
const resume_parser_api = require('../data/resume_parser_api.json');
const cover_letter_api = require('../data/cover_letter_api.json');
// Replace placeholder in resume api with extracted text
resume_parser_api.messages[0].content[0].text = resume_parser_api.messages[0].content[0].text.replace("{{resume}}", resumeText);
// Send resume to LLM
const resumeResponse = await anthropic.messages.create(resume_parser_api);
const candidateProfile = resumeResponse.content[0].text.split('```json')[1].split('```')[0].trim();
// Replace variables in cover letter API prompt
cover_letter_api.messages[0].content[0].text = cover_letter_api.messages[0].content[0].text
.replace('{{resume_json}}', candidateProfile)
.replace('{{job_description}}', jobDescription)
.replace('{{date}}', new Date().toDateString());
//Send data to LLM to generate Cover Letter
const coverLetterResponse = await anthropic.messages.create(cover_letter_api);
const coverLetterRawText = coverLetterResponse.content[0].text.split('')[1].split('')[0].trim();
res.json({ coverLetter: coverLetterRawText });
} catch (error) {
console.error(error);
res.status(500).json({ error: 'Error generating cover letter' });
}
});
router.post('/download', async (req, res) => {
try {
const { coverLetterText } = req.body;
const outputFilename = path.join(__dirname, '../uploads/cover_letter.docx');
generateCoverLetter(coverLetterText, outputFilename);
res.json({downloadLink: '/uploads/cover_letter.docx'});
} catch (error) {
console.error(error);
res.status(500).json({ error: 'Error generating document' });
}
});
function generateCoverLetter(rawText, outputFilename) {
const header = rawText.match(/(.*?)<\/header>/s)[1].trim();
const greeting = rawText.match(/(.*?)<\/greeting>/s)[1].trim();
const introduction = rawText.match(/(.*?)<\/introduction>/s)[1].trim();
const body = rawText.match(/(.*?)<\/body>/s)[1].trim();
const conclusion = rawText.match(/(.*?)<\/conclusion>/s)[1].trim();
const signature = rawText.match(/(.*?)<\/signature>/s)[1].trim();
// Create a new document using docx
const doc = new Document({
sections: [
{
properties: {},
children: [
new Paragraph({
children: [
new TextRun(header),
new TextRun("\n\n"), // Add line breaks between sections
],
}),
new Paragraph({
children: [
new TextRun(greeting),
new TextRun("\n\n"),
],
}),
new Paragraph({
children: [
new TextRun(introduction),
new TextRun("\n\n"),
],
}),
new Paragraph({
children: [
new TextRun(body),
new TextRun("\n\n"),
],
}),
new Paragraph({
children: [
new TextRun(conclusion),
new TextRun("\n\n"),
],
}),
new Paragraph({
children: [
new TextRun(signature),
new TextRun("\n\n"),
],
}),
],
},
],
});
Packer.toBuffer(doc).then((buffer) => {
fs.writeFileSync(outputFilename, buffer);
console.log(`Cover letter saved to ${outputFilename}`);
}).catch(error => console.error("Error generating document:", error));
}
module.exports = router;