Upload app.py

app.py

@@ -0,0 +1,232 @@
+
+import openai
+import streamlit as st
+import fitz
+            
+# environment setup
+#key=st.text_input('Please input your OpenAI key')
+key='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'
+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
+