NLP_pkg/__init__.py

import openai
import os
openai.api_key = os.environ['OPENAI_API_KEY']

def request_api(prompt):
    response = openai.Completion.create(
        model="text-davinci-003", 
        prompt=prompt,
        temperature=0.5, 
        max_tokens=200
    )
    res = response.to_dict_recursive()['choices'][0]['text']
    return res

def create_prompt(resume, jd):
    prompt = f"""Write a small summary for a Job application for the following 
resume based on Job requirements structuring the summary based on skills, 
tools excluding repititive technologies.
    Job Requirements:
"{jd}
"
    Resume Text:
"{resume}
"
Start the response with
"Hi, I am a ..."
"""
    return prompt

def generate_summary(resume, jd):
    return request_api(create_prompt(resume, jd))