| import openai |
| import os |
| from paperqa import Docs |
| import gradio as gr |
| from langchain.document_loaders import PyPDFLoader |
| from langchain.vectorstores import Chroma |
| from langchain.embeddings.openai import OpenAIEmbeddings |
| from langchain.document_loaders import UnstructuredPDFLoader |
| from langchain.llms import OpenAI |
| from langchain.chains.question_answering import load_qa_chain |
| from langchain.chat_models import ChatOpenAI |
|
|
| css_style = """ |
| |
| .gradio-container { |
| font-family: "IBM Plex Mono"; |
| } |
| |
| .answerText p { |
| font-size: 24px !important; |
| color: #8dbcfe !important; |
| } |
| """ |
|
|
|
|
| def run(uploaded_files): |
| all_files = [] |
| if uploaded_files is None: |
| return all_files |
| for file in uploaded_files: |
| if file.name.endswith('.pdf'): |
| all_files.append(file.name) |
| print(all_files) |
| return all_files |
|
|
|
|
| def createAnswer(files, designation, openaikey): |
| os.environ['OPENAI_API_KEY'] = openaikey.strip() |
| docs = Docs(llm='gpt-3.5-turbo') |
| for d in files: |
| docs.add(d.name) |
| answer = docs.query( |
| f"Who is the best canidate to hire for {designation}. Provide a list with the candidate name. If you don't know, simply say None of the canidates are suited for the Job role.") |
| print(answer.formatted_answer) |
| print(type(answer)) |
| return answer.answer |
|
|
|
|
| with gr.Blocks(css=css_style) as demo: |
| gr.Markdown(f""" |
| # HR-GPT - Filter & Find The Best Candidate for the Job using AI |
| |
| *By Amin Memon ([@AminMemon](https://twitter.com/AminMemon))* |
| |
| This tool will enable asking questions of your uploaded text, PDF documents,. |
| It uses OpenAI's ChatGPT model & OpenAI Embeddings and thus you must enter your API key below. |
| |
| This tool is under active development and currently uses many tokens - up to 10,000 |
| for a single query. That is $0.10-0.20 per query, so please be careful! |
| Porting it to Llama.cpp soon for saved cost. |
| |
| 1. Enter API Key ([What is that?](https://platform.openai.com/account/api-keys)) |
| 2. Upload your Resumes (Try a few resumes/cv to try < 5) |
| 3. Provide Designation for which you are hiring |
| """) |
|
|
| openaikey = gr.Text( |
| label='Your OpenAI Api Key', value="") |
| position = gr.Text( |
| label='Position/Designation for which you are hiring for', value="") |
|
|
| with gr.Tab('File Upload'): |
| uploaded_files = gr.File( |
| label="Resume Upload - ONLY PDF. (Doc File Support Coming Soon)", file_count="multiple", show_progress=True) |
|
|
| uploaded_files.change( |
| fn=run, inputs=[uploaded_files], outputs=[uploaded_files]) |
| ask = gr.Button("Find Top Candidate") |
| answer = gr.Markdown(label="Result", elem_classes='answerText') |
| ask.click(fn=createAnswer, inputs=[ |
| uploaded_files, position, openaikey], outputs=[answer]) |
|
|
| demo.queue(concurrency_count=20) |
| demo.launch(show_error=True) |
|
|
