File size: 2,331 Bytes
1189adc
99c5cc5
1189adc
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
import os
os.environ["OPENAI_API_KEY"] = "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"

import openai
import os
import PyPDF2
from dotenv import load_dotenv, find_dotenv

_ = load_dotenv(find_dotenv())

openai.api_key = os.getenv('OPENAI_API_KEY')

def get_completion(prompt, model="gpt-3.5-turbo", temperature=0, max_tokens=500):
    messages = [{"role": "user", "content": prompt}]

    response = openai.ChatCompletion.create(
        model=model,
        temperature=temperature,
        max_tokens=max_tokens,
        messages=messages,
    )

    return response.choices[0].message["content"]

def generate_prompt(text, format="text"):
    prompt = f"""Play as an AI HR recruiter specialist and extract all the jobs as a list with :
                  1. Job Title
                  2. Location
                  3. Educations as list
                  4. Experiences as list
                  5. Skills and Competences as list
                  6. Functions and Tasks as list
                  7. Return the result in {format} format \
                  here the text \
                  ``` {text}```


                   """
    return prompt

import PyPDF2

def read_pdf(file_path):
    pdf_file = open(file_path, 'rb')
    pdf_reader = PyPDF2.PdfReader(pdf_file)
    text = ""
    for page in pdf_reader.pages:
        text += page.extract_text()
    pdf_file.close()
    return text

import PyPDF2
def summarize(input):
    text = read_pdf(input.name)
    prompt = generate_prompt(text)
    response = get_completion(prompt)
    return response

import gradio as gr
gr.close_all()
demo = gr.Interface(
    fn=summarize, 
    inputs=gr.components.File(label="Importez votre document en format PDF ici"), 
    outputs=gr.components.Textbox(label="Voici le résultat"),
    title="Téléchargement de CV\n(version anglaise)",
    description="Outil d’importation de CV")
demo.launch()

#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
#++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++

import gradio as gr

def colorize_text(text):
    return "<span style='color:red;background-color:orange'>" + text + "</span>" "<span style='color:white;background-color:red'>" + "Experiences" + "</span>"

iface = gr.Interface(fn=colorize_text, inputs="text", outputs="html")
iface.launch()