Spaces:
Runtime error
Runtime error
| import openai | |
| import streamlit as st | |
| import fitz | |
| # environment setup | |
| key=st.text_input('Please input your OpenAI key') | |
| openai.api_key=key | |
| def prompt1(company,position,round,intervier_level): | |
| prompt=f""" | |
| I want you to act as a {intervier_level} for {round} interviews on {company} company for the position of {position}. You'll suggest the characteristics of this job interview and the characteristics of the {company}'s interview. | |
| The return format should bullet point be like this: | |
| Characteristics of this job interview: | |
| - name: detail explanation | |
| Characteristics of {company}: | |
| - name: detail explanation | |
| """ | |
| return prompt | |
| def followup1(company,position,round,bullet_point): | |
| prompt=f""" | |
| Can you talk the detail about {bullet_point} for {round} interviews on {company} company for the position of {position}? | |
| The return format should bullet point be like this: | |
| - name: detail explanation | |
| """ | |
| return prompt | |
| def prompt2(position): | |
| prompt=f""" | |
| I want you to act as a talent recruiter for the position of {position}'s interviews. You'll suggest the characteristics of this job interview for both behavior requirement and technical requirement. | |
| The return format should bullet point be like this: | |
| Characteristics of this behavior requirement: | |
| - name: detail explanation | |
| Characteristics of technical requirement: | |
| - name: detail explanation | |
| """ | |
| return prompt | |
| def followup2(position,bullet_point): | |
| prompt=f""" | |
| Can you talk the detail about {bullet_point} for the interview of {position}? | |
| The return format should bullet point be like this: | |
| - name: detail impovement advice | |
| """ | |
| return prompt | |
| def prompt3(position,resume,dis_num,ad_num): | |
| prompt=f""" | |
| I want you to act as a talent recruiter for hiring {position}. I will give you a resume and you'll suggest what are {dis_num}+ disadvantages and {ad_num}+ advantages of the {position} position. Please remember, the return should be highly related to the {position} position. | |
| The return format should bullet point be like this: | |
| Disadvatage: | |
| - name: Suggestions for improvements to make the resume more appropriate for the {position} position and link to specific sentences on the resume. | |
| Advantage: | |
| - name: use one sentence to explain why it is advantageous for the {position} position. | |
| Here is the resume: | |
| {resume}. | |
| """ | |
| return prompt | |
| def followup3(position,resume,bullet_point): | |
| prompt=f""" | |
| Can you talk the detail about {bullet_point} for the position of {position} based on {resume}? | |
| The return format should bullet point be like this: | |
| - name: detail impovement advice | |
| """ | |
| return prompt | |
| def ask(prompt): | |
| rsp = openai.ChatCompletion.create( | |
| model="gpt-3.5-turbo", | |
| messages=[ | |
| {"role": "system", "content": prompt}, | |
| ] | |
| ) | |
| output=rsp.choices[0].message.content | |
| return output | |
| #################### | |
| st.title('InterviewPrepGPT') | |
| st.subheader('A tool using to help people prepare their interview by using gpt3.5') | |
| option = st.selectbox( | |
| 'How would you like to prepare for the interview?', | |
| ('Prepare for a specific interview', | |
| 'Understand the requirement of a specific position', | |
| 'Analyze resume', | |
| )) | |
| if "submit" not in st.session_state: | |
| st.session_state["submit"] = False | |
| st.write('') | |
| ###### option 1 | |
| if option=='Prepare for a specific interview': | |
| col1, col2= st.columns(2) | |
| with col1: | |
| company=st.text_input('Input your company','Amazon') | |
| with col2: | |
| position=st.text_input('Input your position','Data engineer') | |
| col3, col4 = st.columns(2) | |
| with col3: | |
| intervier_level=st.text_input('Input your interviewer level: ','Talent Recuriter') | |
| with col4: | |
| round=st.radio('Select your round: ',('Phone Screen','Behavior Interview','Technical Interview', 'General')) | |
| if round=='General': | |
| round='' | |
| st.write('Click button again if you want to regenerate a new ansewer.') | |
| submit_button=st.button('Submit') | |
| if st.session_state.get('button') != True: | |
| st.session_state['button'] = submit_button | |
| if st.session_state['button'] == True: | |
| prompt=prompt1(company,position,round,intervier_level) | |
| output=ask(prompt) | |
| st.write(output) | |
| followup_time=0 | |
| followup='' | |
| while True: | |
| output_list=output.split('\n') | |
| indexes = [i for i, word in enumerate(output_list) if '- ' in word] | |
| new_list = [output_list[i] for i in indexes] | |
| Cq=[i.split(':')[0].strip('- ') for i in new_list] | |
| Cq = ['None']+ Cq | |
| followup_radio = st.radio('I want to follow up:', tuple(Cq),key='0') | |
| if followup_radio: | |
| followup_time +=1 | |
| if followup_radio == 'None': | |
| break | |
| else: | |
| if followup_radio != 'None': | |
| followup = followup1(company, position, round, followup_radio) | |
| output = ask(followup) | |
| st.write(output) | |
| if followup_time>5: | |
| break | |
| ###### option 2 | |
| if option =='Understand the requirement of a specific position': | |
| position=st.text_input('Input your position','Data engineer') | |
| st.write('Click button again if you want to regenerate a new ansewer.') | |
| submit=st.checkbox('submit') | |
| if submit: | |
| prompt=prompt2(position) | |
| output=ask(prompt) | |
| st.write(output) | |
| followup='' | |
| output_list=output.split('\n') | |
| indexes = [i for i, word in enumerate(output_list) if '- ' in word] | |
| new_list = [output_list[i] for i in indexes] | |
| Cq=[i.split(':')[0].strip('- ') for i in new_list] | |
| Cq = ['None']+ Cq | |
| followup_radio = st.radio('I want to follow up:', tuple(Cq)) | |
| if followup_radio!='None': | |
| followup = followup2(position,followup_radio) | |
| op = ask(followup) | |
| st.write(op) | |
| ###### option 3 | |
| if option =='Analyze resume': | |
| uploaded_file = st.file_uploader("Upload your resume", type=["pdf"]) | |
| if uploaded_file: | |
| if "submit" not in st.session_state: | |
| st.session_state["submit"] = False | |
| doc = fitz.open(stream=uploaded_file.read(), filetype="pdf") | |
| resume = "" | |
| for page in doc: | |
| resume += page.get_text() | |
| col31, col32,col33= st.columns(3) | |
| with col31: | |
| position=st.text_input('Input your position') | |
| with col32: | |
| dis_num=st.text_input('Input your dis_num') | |
| with col33: | |
| ad_num=st.text_input('Input your ad_num') | |
| st.write('Click button again if you want to regenerate a new ansewer.') | |
| submit_button=st.button('Submit') | |
| if st.session_state.get('button') != True: | |
| st.session_state['button'] = submit_button | |
| if st.session_state['button'] == True: | |
| prompt=prompt3(position,resume,dis_num,ad_num) | |
| output=ask(prompt) | |
| st.write(output) | |
| followup_time=0 | |
| while True: | |
| output_list=output.split('\n') | |
| output_list= [element for element in output_list if element != ''] | |
| ind = [i for i, word in enumerate(output_list) if 'Advantage:' in word] | |
| Cdis_list=output_list[1:int(dis_num)+1] | |
| Cdis=[i.split(':')[0].strip('- ') for i in Cdis_list] | |
| Cdis = ['None']+ Cdis | |
| followup_radio = st.radio('I want to follow up:', tuple(Cdis),key=followup_time) | |
| followup_time +=1 | |
| if followup_radio == 'None': | |
| break | |
| else: | |
| followup = followup3(position,resume,followup_radio) | |
| output = ask(followup) | |
| st.write(output) | |
| if followup_time>4: | |
| break | |