Create resume_gen.py

backend/agents/resume_gen.py

@@ -0,0 +1,113 @@
+from groq import Groq
+
+def generate_custom_resume(resume_text: str, job_description: str, api_key: str) -> str:
+    """
+    Generates a tailored resume using a Groq LLM.
+
+    Args:
+        resume_text (str): The user's original resume text.
+        job_description (str): The job description to tailor the resume for.
+        api_key (str): The Groq API key.
+
+    Returns:
+        str: The content of the tailored resume.
+    """
+    if not api_key:
+        raise ValueError("Groq API key is required.")
+
+    client = Groq(api_key=api_key)
+    
+    prompt = f"""
+    You are an expert resume writer. Your task is to rewrite the following resume to be perfectly tailored for the provided job description.
+    Focus on highlighting the most relevant skills and experiences from the original resume that match the job requirements.
+    Structure the output cleanly and professionally. Do not invent new experiences.
+
+    **Original Resume:**
+    ---
+    {resume_text}
+    ---
+
+    **Job Description:**
+    ---
+    {job_description}
+    ---
+
+    **Your Task:**
+    Rewrite the resume, emphasizing the skills and experiences that align with the job description.
+    The output should be the full text of the new, tailored resume.
+    """
+
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are a professional resume writing assistant.",
+                },
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama3-8b-8192",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        print(f"Error generating custom resume: {e}")
+        return "Error: Could not generate tailored resume."
+
+def generate_cover_letter(resume_text: str, job_description: str, api_key: str) -> str:
+    """
+    Generates a compelling cover letter using a Groq LLM.
+
+    Args:
+        resume_text (str): The user's resume text to extract context from.
+        job_description (str): The job description for the target role.
+        api_key (str): The Groq API key.
+
+    Returns:
+        str: The generated cover letter.
+    """
+    if not api_key:
+        raise ValueError("Groq API key is required.")
+
+    client = Groq(api_key=api_key)
+    
+    prompt = f"""
+    You are an expert career coach. Your task is to write a concise and compelling cover letter for the given job description, based on the provided resume.
+    The cover letter should be professional, enthusiastic, and highlight 2-3 key qualifications from the resume that directly match the job's requirements.
+    Start with a strong opening, explain the value the candidate brings in the body, and end with a clear call to action.
+
+    **Candidate's Resume:**
+    ---
+    {resume_text}
+    ---
+
+    **Target Job Description:**
+    ---
+    {job_description}
+    ---
+
+    **Your Task:**
+    Write a cover letter that is tailored to this specific job. Keep it to about 3-4 paragraphs.
+    Address it to "Hiring Manager" if no specific name is available.
+    """
+
+    try:
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "system",
+                    "content": "You are a professional career coaching assistant.",
+                },
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama3-8b-8192",
+        )
+        return chat_completion.choices[0].message.content
+    except Exception as e:
+        print(f"Error generating cover letter: {e}")
+        return "Error: Could not generate cover letter."