initial commit

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -1,8 +1,8 @@
 ---
-title: Wg Cv Evaluation 2
-emoji: 🏆
-colorFrom: pink
-colorTo: indigo
+title: Cv Profiler
+emoji: 📚
+colorFrom: blue
+colorTo: yellow
 sdk: gradio
 sdk_version: 3.44.4
 app_file: app.py

app.py

@@ -0,0 +1,81 @@
+import openai
+import gradio as gr
+from PyPDF2 import PdfReader
+from dotenv import load_dotenv
+
+load_dotenv()
+
+with open("sys_prompt.txt") as f:
+    sys_prompt = f.read()
+
+
+def add_file(file, chat, job, resume):
+    if file.name.endswith(".pdf"):
+        doc = PdfReader(file.name)
+        text = ""
+        for page in doc.pages:
+            text += page.extract_text()
+    else:
+        with open(file.name) as f:
+            text = f.read()
+
+    if job:
+        chat += [["📄 " + file.name.split("/")[-1], None]]
+        resume = text
+    else:
+        chat += [["📄 " + file.name.split("/")[-1], "Thanks. Please upload the resume."]]
+        job = text
+
+    return chat, job, resume
+
+
+def user(message, history):
+    return "", history + [[message, None]]
+
+
+def bot(history, job, resume):
+    if not resume:
+        yield history
+        return
+
+    system = sys_prompt.format(job=job, resume=resume, n=8)
+    messages = [{"role": "system", "content": system}]
+    for user, assistant in history:
+        if user:
+            messages.append({"role": "user", "content": user})
+        if assistant:
+            messages.append({"role": "assistant", "content": assistant})
+
+    response = openai.ChatCompletion.create(
+        model="gpt-4",
+        messages=messages,
+        temperature=0.0,
+        stream=True,
+    )
+
+    history[-1][1] = ""
+    for chunk in response:
+        if len(chunk["choices"][0]["delta"]) != 0:
+            history[-1][1] = history[-1][1] + chunk["choices"][0]["delta"]["content"]
+            yield history
+    return
+
+
+with gr.Blocks() as app:
+    job = gr.State("")
+    resume = gr.State("")
+    chat = gr.Chatbot([[None, "Please upload the job description."]], height=600)
+    with gr.Row():
+        clr = gr.Button("Clear", scale=0)
+        msg = gr.Textbox(lines=1, show_label=False, scale=1)
+        file = gr.UploadButton("Browse", file_types=[".pdf", ".txt"], scale=0)
+    msg.submit(user, [msg, chat], [msg, chat], queue=False).then(
+        bot, [chat, job, resume], chat
+    )
+    file.upload(
+        add_file, [file, chat, job, resume], [chat, job, resume], queue=False
+    ).then(bot, [chat, job, resume], chat)
+    clr.click(lambda: None, None, chat, queue=False)
+
+app.queue()
+app.launch()

sys_prompt.txt

@@ -0,0 +1,22 @@
+You are a professional recruiter specialized in conducting interviews. Your task is to lead through an interview by asking {n} questions and collect as much relevant information from the applicant (the user) as possible. As context you will be given the job description as well as the applicants resume.
+
+Please follow these rules:
+
+- Pose the questions one at a time, waiting for the applicant to respond to each.
+- If a question has been fully addressed, proceed to the next topic.
+- You may ask a follow-up question if the previous response was not satisfactory.
+- Try to ask open ended questions to collect more information from the applicant.
+- Concentrate on questions that the resume alone cannot answer. Aim to fill in the gaps.
+- Don't give feedback, don't summarize and don't explain yourself. Your role is investigative.
+- Use the {n} questions wisely to get an overall impression of the applicant.
+
+
+After {n} questions have been answered fully, you switch roles. Now is your time to analyze and summarize how well the candidate fits the job overall. Use all information available including job description, resume and the answers to your questions. What is positive? What is missing? How are their chances of getting hired?
+
+
+JOB DESCRIPTION:
+{job}
+
+
+RESUME:
+{resume}