Add Generate Resume Route

Signed-off-by: Tal Jacob <taljacob2@gmail.com>

Author: taljacob2

nextstep-backend/src/controllers/resume_controller.ts

@@ -2,7 +2,7 @@ import { Request, Response } from 'express';
 import { config } from '../config/config';
 import fs from 'fs';
 import path from 'path';
-import { scoreResume, streamScoreResume, getResumeTemplates } from '../services/resume_service';
+import { scoreResume, streamScoreResume, getResumeTemplates, generateImprovedResume } from '../services/resume_service';
 import multer from 'multer';
 import { CustomRequest } from "types/customRequest";
 import { handleError } from "../utils/handle_error";
@@ -78,8 +78,19 @@ const getTemplates = async (req: Request, res: Response) => {
     }
 };
 
-export default {
-    getResumeScore,
-    getStreamResumeScore,
-    getTemplates
-};
+const generateResume = async (req: Request, res: Response) => {
+    try {
+        const { feedback, jobDescription, templateName } = req.body;
+        
+        if (!feedback || !jobDescription || !templateName) {
+            return res.status(400).json({ error: 'Missing required fields' });
+        }
+
+        const result = await generateImprovedResume(feedback, jobDescription, templateName);
+        return res.status(200).json(result);
+    } catch (error) {
+        handleError(error, res);
+    }
+};
+
+export default { getResumeScore, getStreamResumeScore, getTemplates, generateResume };

nextstep-backend/src/openapi/swagger.yaml

@@ -987,6 +987,75 @@ paths:
         '500':
           description: Internal server error
 
+  /resume/generate:
+    post:
+      tags:
+        - Resume
+      summary: Generate an improved resume based on a resume template, feedback and job description
+      security:
+        - BearerAuth: []
+      requestBody:
+        required: true
+        content:
+          application/json:
+            schema:
+              type: object
+              required:
+                - feedback
+                - jobDescription
+                - templateName
+              properties:
+                feedback:
+                  type: string
+                  description: The feedback received from the resume scoring process
+                jobDescription:
+                  type: string
+                  description: The job description used for scoring the resume
+                templateName:
+                  type: string
+                  description: The name of the template to use for generating the improved resume
+      responses:
+        '200':
+          description: The generated improved resume
+          content:
+            application/json:
+              schema:
+                type: object
+                properties:
+                  content:
+                    type: string
+                    description: Base64 encoded content of the generated resume file
+                  type:
+                    type: string
+                    description: MIME type of the generated resume file
+                    enum:
+                      - application/msword
+                      - application/vnd.openxmlformats-officedocument.wordprocessingml.document
+        '400':
+          description: Bad request - Missing required fields
+          content:
+            application/json:
+              schema:
+                type: object
+                properties:
+                  error:
+                    type: string
+                    description: Error message indicating missing fields
+        '401':
+          description: Unauthorized - Missing or invalid authentication token
+        '404':
+          description: Template not found
+          content:
+            application/json:
+              schema:
+                type: object
+                properties:
+                  error:
+                    type: string
+                    description: Error message indicating template not found
+        '500':
+          description: Internal server error
+
   /room/user/{receiverUserId}:
     get:
       tags:

nextstep-backend/src/routes/resume_routes.ts

@@ -10,4 +10,6 @@ router.get('/streamScore/:filename', Resume.getStreamResumeScore);
 
 router.get('/templates', Resume.getTemplates);
 
+router.post('/generate', Resume.generateResume);
+
 export default router; 

nextstep-backend/src/services/resume_service.ts

@@ -205,4 +205,53 @@ const getResumeTemplates = async (): Promise<{ name: string; content: string; ty
     }
 };
 
-export { scoreResume, streamScoreResume, getResumeTemplates };
+const generateImprovedResume = async (
+    feedback: string,
+    jobDescription: string,
+    templateName: string
+): Promise<{ content: string; type: string }> => {
+    try {
+        const templatesDir = config.assets.resumeTemplatesDirectoryPath();
+        const templatePath = path.join(templatesDir, `${templateName}.docx`);
+        
+        if (!fs.existsSync(templatePath)) {
+            throw new Error(`Template ${templateName} not found`);
+        }
+
+        const prompt = `You are an expert resume writer. Using the following feedback and job description, modify the resume template to implement all suggested improvements.
+
+Feedback:
+${feedback}
+
+Job Description:
+${jobDescription}
+
+Please modify the template to:
+1. Keep the exact same format and structure as the template
+2. Implement all suggested improvements from the feedback
+3. Ensure the content matches the job description requirements
+4. Maintain professional formatting and style
+
+Return the modified resume in the same format as the template.`;
+
+        if (!config.chatAi.turned_on()) {
+            throw new Error('Chat AI feature is turned off');
+        }
+
+        const modifiedContent = await chatWithAI(prompt, SYSTEM_TEMPLATE);
+        
+        // For now, we'll return the original template content
+        // TODO: Implement actual document modification
+        const content = fs.readFileSync(templatePath);
+        
+        return {
+            content: content.toString('base64'),
+            type: 'application/vnd.openxmlformats-officedocument.wordprocessingml.document'
+        };
+    } catch (error) {
+        console.error('Error generating improved resume:', error);
+        throw error;
+    }
+};
+
+export { scoreResume, streamScoreResume, getResumeTemplates, generateImprovedResume };

nextstep-frontend/src/pages/Resume.tsx

@@ -134,8 +134,11 @@ const Resume: React.FC = () => {
   const [error, setError] = useState('');
   const [templates, setTemplates] = useState<Array<{ name: string; content: string; type: string }>>([]);
   const [selectedTemplate, setSelectedTemplate] = useState<number | null>(null);
+  const [previewOpen, setPreviewOpen] = useState(false);
+  const [previewTemplate, setPreviewTemplate] = useState<{ name: string; content: string; type: string } | null>(null);
   const [previewUrl, setPreviewUrl] = useState<string | null>(null);
   const [previewUrlCache, setPreviewUrlCache] = useState<Record<string, string>>({});
+  const [generatedResume, setGeneratedResume] = useState<{ content: string; type: string } | null>(null);
   const fileInputRef = useRef<HTMLInputElement>(null);
   const feedbackEndRef = useRef<HTMLDivElement>(null);
 
@@ -244,9 +247,30 @@ const Resume: React.FC = () => {
     setActiveStep((prevStep) => prevStep - 1);
   };
 
-  const handleGenerateResume = () => {
-    // TODO: Implement resume generation
-    console.log('Generating resume with template:', selectedTemplate);
+  const handleGenerateResume = async () => {
+    if (selectedTemplate === null) {
+      setError('Please select a template');
+      return;
+    }
+
+    setLoading(true);
+    setError('');
+
+    try {
+      const selectedTemplateData = templates[selectedTemplate];
+      const response = await api.post('/resume/generate', {
+        feedback,
+        jobDescription,
+        templateName: selectedTemplateData.name
+      });
+
+      setGeneratedResume(response.data);
+      setActiveStep(2);
+    } catch (err) {
+      setError(err instanceof Error ? err.message : 'Failed to generate resume');
+    } finally {
+      setLoading(false);
+    }
   };
 
   const uploadToTmpFiles = async (base64Content: string, fileName: string): Promise<string> => {
@@ -515,13 +539,23 @@ const Resume: React.FC = () => {
             <Typography variant="h4" gutterBottom>
               Generate matching resume
             </Typography>
-            <Button
-              variant="contained"
-              onClick={handleGenerateResume}
-              disabled={selectedTemplate === null}
-            >
-              Generate Resume
-            </Button>
+            {generatedResume ? (
+              <Box sx={{ mt: 3 }}>
+                <Button
+                  variant="contained"
+                  href={`data:${generatedResume.type};base64,${generatedResume.content}`}
+                  download={`improved_resume${generatedResume.type.includes('docx') ? '.docx' : '.doc'}`}
+                >
+                  Download Improved Resume
+                </Button>
+              </Box>
+            ) : (
+              <Box sx={{ mt: 3, textAlign: 'center' }}>
+                <Typography variant="body1" color="text.secondary">
+                  Your improved resume will be ready for download here.
+                </Typography>
+              </Box>
+            )}
           </Box>
         );
       default: