Added interview questions generator chatbot

app.py

@@ -8,6 +8,8 @@ from streamlit_chat import message
 
 from job_description_generator import predict_job_description, get_job_description_conversation
 from job_description_fixer import fix_job_description, get_job_description_fixer_conversation
+from interview_questions_generator import (predict_interview_question,
+                                           get_interview_questions_conversation)
 from cover_letter_generator import get_cover_letter
 
 conversation = get_job_description_conversation()
@@ -17,6 +19,10 @@ if 'generator_conversation' not in st.session_state:
 fixer_conversation = get_job_description_fixer_conversation()
 if 'fixer_conversation' not in st.session_state:
     st.session_state['fixer_conversation'] = fixer_conversation
+    
+interview_questions_conversation = get_interview_questions_conversation()
+if 'interview_questions_conversation' not in st.session_state:
+    st.session_state['interview_questions_conversation'] = interview_questions_conversation
 
 def message_writer(input_text, response):
     '''
@@ -69,7 +75,11 @@ def main():
     st.sidebar.markdown("---")
     #selector
     page = st.sidebar.selectbox(
-        "Select a page", ["Home", "Job Description Generator", "Job Description Fixer", "Cover Letter Generator"])
+        "Select a page", ["Home",
+                          "Job Description Generator",
+                          "Job Description Fixer",
+                          "Cover Letter Generator",
+                          "interview questions generator"])
     if page == "Home":
         st.title("JobGPT")
         st.write("Select a page in the sidebar to get started.")
@@ -77,6 +87,7 @@ def main():
         st.write("1. Job Description Generator")
         st.write("2. Job Description Fixer")
         st.write("3. Cover Letter Generator")
+        st.write("4. Interview Questions Generator")
         st.markdown("---")
 
     elif page == "Job Description Generator":
@@ -143,19 +154,15 @@ def main():
         container_three.markdown( "JobGPT is a chatbot that generates cover letters. \
                                  This is built just for demo purpose.")
         container_three.markdown("---")
-        #upload file
         uploaded_file = container_three.file_uploader("Upload your resume", type=["pdf"])
         if uploaded_file is not None:
-            #store it as a pdf
             with open("resume.pdf", "wb") as file_io:
                 file_io.write(uploaded_file.getbuffer())
-           
-            #show a loading bar
+
             with st.spinner('Uploading...'):
                 time.sleep(1)
             container_three.success('Uploaded!')
             container_three.markdown("---")
-            #add a form to get title, company and more info from the user
             form = container_three.form(key='my_form')
             title = form.text_input("Job Title (required)", placeholder="VP of Engineering")
             company = form.text_input("Company Name (required)", placeholder="Google")
@@ -175,4 +182,34 @@ def main():
                         container_three.markdown("### Cover Letter:")
                         container_three.write(cover_letter)
 
+    elif page == "interview questions generator":
+        container_two = st.container()
+        container_two.title("A Interview Questions Generating Chatbot")
+        container_two.markdown(
+            "JobGPT is a chatbot that generates interview questions.\
+                This is built just for demo purpose."
+        )
+        input_text = container_two.text_area(
+            "Prompt",
+            "Hi, can you please help me generate interview questions?")
+        button = container_two.button("Send")
+
+        st.sidebar.markdown("---")
+        st.sidebar.markdown("Click on `new chat` to start a new chat. \
+                History will be cleared and you'll lose access to current chat."
+                           )
+        clear_session = st.sidebar.button("New Chat")
+
+        if clear_session:
+            st.session_state['interview_questions_conversation'] = interview_questions_conversation
+        container_two.markdown("---")
+
+        initial_message = "Hello, how can I help you?"
+        message(initial_message)
+
+        if button:
+            response = predict_interview_question(
+                input_text, st.session_state['interview_questions_conversation'])
+            message_writer(input_text, response)
+
 main()

interview_questions_generator.py

@@ -0,0 +1,29 @@
+import os
+import openai
+from langchain.chat_models import ChatOpenAI
+from langchain.memory import ConversationBufferMemory
+from llm import generate_prompt, generate_conversation, predict
+
+openai_api_key = os.environ['OPENAI_API_KEY']
+openai.api_key = openai_api_key
+
+
+def get_interview_questions_conversation():
+    '''
+    Generate a conversation object for job description generation
+    '''
+    prompt = generate_prompt(
+        input_variables=['history', 'input'],
+        template_file='templates/interview_question_generator.txt')
+    llm = ChatOpenAI(temperature=0, openai_api_key=openai_api_key)
+    memory = ConversationBufferMemory(ai_prefix="JobGPT")
+    conversation = generate_conversation(memory=memory, llm=llm, prompt=prompt)
+    return conversation
+
+
+def predict_interview_question(input_text: str, conversation: object):
+    '''
+    Predict the next response from the conversation object
+    '''
+    response = predict(input_text=input_text, conversation=conversation)
+    return response

templates/interview_question_generator.txt

@@ -0,0 +1,20 @@
+JobGPT is a friendly and professional chatbot bot that generates interview questions based on the job description of executive positions that is given to it. \
+
+It should get following information from the user, asking only one question at a time to not overwhelm the user: \
+1. Number of questions to be generated. \
+2. Type of questions (Technical, Behavioral, etc.) \
+3, Job description of the position. \
+
+It should strictly follow following guidelines: \
+1. The questions should be open-ended. \
+2. The questions should test the candidate for the skills and requirements mentioned in the job description. \
+3. The questions should be well-suited for the level of job description given. \
+
+Once it has the questions and answers, it returns them as a well formatted markdown text. \
+
+The following is a friendly conversation between a human and JobGPT. JobGPT provides lots of specific details from its context. Your job is to predict what JobGPT says next.
+
+Current conversation:
+{history}
+Human: {input}
+JobGPT: