|
|
import gradio as gr |
|
|
import openai |
|
|
import os |
|
|
import PyPDF2 |
|
|
from docx import Document |
|
|
|
|
|
openai.api_key = os.getenv("OPENAI_API_KEY") |
|
|
|
|
|
def extract_text_from_file(file_path): |
|
|
file_extension = os.path.splitext(file_path)[1] |
|
|
if file_extension == '.pdf': |
|
|
with open(file_path, 'rb') as file: |
|
|
reader = PyPDF2.PdfFileReader(file) |
|
|
text = "" |
|
|
for page_number in range(reader.getNumPages()): |
|
|
page = reader.getPage(page_number) |
|
|
text += page.extractText() |
|
|
return text |
|
|
|
|
|
elif file_extension == '.txt': |
|
|
with open(file_path, 'r') as file: |
|
|
text = file.read() |
|
|
return text |
|
|
|
|
|
elif file_extension == '.docx': |
|
|
doc = Document(file_path) |
|
|
text = [] |
|
|
for paragraph in doc.paragraphs: |
|
|
text.append(paragraph.text) |
|
|
text='\n'.join(text) |
|
|
return text |
|
|
|
|
|
else: |
|
|
return "Unsupported file type" |
|
|
|
|
|
def gpt3_prompt(prompt): |
|
|
response = openai.Completion.create( |
|
|
engine="text-davinci-003", |
|
|
prompt=prompt, |
|
|
temperature=0, |
|
|
max_tokens=200 |
|
|
) |
|
|
return response.choices[0].text.strip() |
|
|
|
|
|
def respond(message, chat_history, file_path): |
|
|
try: |
|
|
if file_path is not None: |
|
|
text = extract_text_from_file(file_path.name) |
|
|
text = text + " " + message |
|
|
|
|
|
|
|
|
prompt=f"""Your are HR chatbot. Answer for user Queries with user friendly and proper output following Instructions if . |
|
|
Instructions: |
|
|
1. Aanalyse the text to Generate Job Description For Given Job title and Experience. if User ask this ask to user Job Title and Experience and return Response. |
|
|
2. Aanalyse the text to Prepare questions from job Description. If User ask this ask to user Upload file and generate and return responce from that file |
|
|
3. Aanalyse the text to Find Education Gaps in given resume. Find Skills in resume. If User ask this ask to user Upload file and generate and return responce from that file |
|
|
4. Aanalyse the text to to generate online courses with website links. If User ask this ask to user Upload file and generate and return responce from that file Generate atmost five courses. |
|
|
5. If the text not available continue normal HR Chatbot |
|
|
text : {text} |
|
|
""" |
|
|
bot_message = gpt3_prompt(prompt) |
|
|
chat_history.append((message, bot_message)) |
|
|
return "", chat_history |
|
|
except: |
|
|
bot_message = "Please Upload Resume!!" |
|
|
chat_history.append((message, bot_message)) |
|
|
return "", chat_history |
|
|
|
|
|
with gr.Blocks() as demo: |
|
|
chatbot = gr.Chatbot() |
|
|
msg = gr.Textbox() |
|
|
clear = gr.ClearButton([msg, chatbot]) |
|
|
uploader = gr.inputs.File() |
|
|
|
|
|
msg.submit(respond, [msg, chatbot, uploader], [msg, chatbot]) |
|
|
|
|
|
demo.launch() |
