Baskar2005's picture
Update app.py
215fe64 verified
raw
history blame
3.25 kB
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_core.messages import HumanMessage
from langchain_core.messages import AIMessage
from langchain.memory import ChatMessageHistory
from langchain_openai import AzureChatOpenAI
from pypdf import PdfReader
import os
import gradio as gr
chat = AzureChatOpenAI(azure_deployment = "GPT-3")
def extract_text( pdf_path):
# creating a pdf reader object
reader = PdfReader(pdf_path)
all_text = ""
for page in reader.pages:
all_text += page.extract_text()
return all_text
def get_response( candidate, chat_history, resume, jd):
resume = extract_text(resume.name)
jd = extract_text(jd.name)
prompt = ChatPromptTemplate.from_messages(
[
(
"system",
"""Your Task is Perform as intelligent interviewer, Your Task is ask question to the resume's candidate by following candidate Answer.
at the end exit with greeting to the candidate.
**Ask question follow up on the candidate response. get chat history.**
""",
),
MessagesPlaceholder(variable_name="messages"),
]
)
chain = prompt | chat
# chat_histroy_prompt = chat_history
answer = chain.invoke(
{
"messages": [
HumanMessage(
content=f" job description :{jd}\n Resume :{resume}"
),
AIMessage(content=f"""Perform as intelligent interviewer, Your Task is ask question to the resume's candidate by following candidate Answer.
chat history : {chat_history}"""),
HumanMessage(content=candidate),
],
}
)
# print("INTERVIEWER :", answer.content)
# chat_history.append({"candidate":candidate,"interviewer":answer.content })
result = answer.content
chat_history.append((candidate, result))
print("chat_history", chat_history)
return "", chat_history
def gradio_interface() -> None:
"""Create a Gradio interface for the chatbot."""
with gr.Blocks(css = "style.css" ,theme="HaleyCH/HaleyCH_Theme") as demo:
gr.HTML("""<center class="darkblue" text-align:center;padding:30px;'></center>
<img class="leftimage" align="center" src="https://encrypted-tbn0.gstatic.com/images?q=tbn:ANd9GcQbH6dyllkVQ-qz6QiN7KL4gunt61DvVtjD3D2_fF7RhQ&s" alt="Image" width="60" height="210">
<h1 class ="center" style="color:#fff">Athmick</h1>
<br><center><h1 style="color:#fff">Screening Assistant Chatbot</h1></center>""")
chatbot = gr.Chatbot()
with gr.Row():
with gr.Column(scale=1):
msg = gr.Textbox(label="Question")
with gr.Row():
with gr.Column(scale=0.15):
resume = gr.File(label="Resume")
with gr.Column(scale=0.15):
jd = gr.File(label="Job Description")
with gr.Column(scale=0.85):
clear = gr.ClearButton([msg, chatbot])
msg.submit(get_response, [msg, chatbot, resume, jd], [msg, chatbot])
demo.launch(debug =True, share=True)
gradio_interface()