| 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) | |