devopsmarc/wip-space1-en

1import os2os.environ["OPENAI_API_KEY"] = "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"3 4import openai5import os6import PyPDF27from dotenv import load_dotenv, find_dotenv8 9_ = load_dotenv(find_dotenv())10 11openai.api_key = os.getenv('OPENAI_API_KEY')12 13def get_completion(prompt, model="gpt-3.5-turbo", temperature=0, max_tokens=500):14    messages = [{"role": "user", "content": prompt}]15 16    response = openai.ChatCompletion.create(17        model=model,18        temperature=temperature,19        max_tokens=max_tokens,20        messages=messages,21    )22 23    return response.choices[0].message["content"]24 25def generate_prompt(text, format="text"):26    prompt = f"""Play as an AI HR recruiter specialist and extract all the jobs as a list with :27                  1. Job Title28                  2. Location29                  3. Educations as list30                  4. Experiences as list31                  5. Skills and Competences as list32                  6. Functions and Tasks as list33                  7. Return the result in {format} format \34                  here the text \35                  ``` {text}```36 37 38                   """39    return prompt40 41import PyPDF242 43def read_pdf(file_path):44    pdf_file = open(file_path, 'rb')45    pdf_reader = PyPDF2.PdfReader(pdf_file)46    text = ""47    for page in pdf_reader.pages:48        text += page.extract_text()49    pdf_file.close()50    return text51 52import PyPDF253def summarize(input):54    text = read_pdf(input.name)55    prompt = generate_prompt(text)56    response = get_completion(prompt)57    return response58 59import gradio as gr60gr.close_all()61demo = gr.Interface(62    fn=summarize, 63    inputs=gr.components.File(label="Importez votre document en format PDF ici"), 64    outputs=gr.components.Textbox(label="Voici le résultat"),65    title="Téléchargement de CV\n(version anglaise)",66    description="Outil d’importation de CV")67demo.launch()68 69#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++70#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++71 72import gradio as gr73 74def colorize_text(text):75    return "<span style='color:red;background-color:orange'>" + text + "</span>" "<span style='color:white;background-color:red'>" + "Experiences" + "</span>"76 77iface = gr.Interface(fn=colorize_text, inputs="text", outputs="html")78iface.launch()