Update app.py

app.py

@@ -1,42 +1,49 @@
 import os
-import io
 import streamlit as st
 from dotenv import load_dotenv
-import fitz  # PyMuPDF
-import base64
-import google.generativeai as genai
+import fitz  # PyMuPDF for handling PDFs
+import openai
 
-
-# Load environment variables
+# Load environment variables and configure OpenAI API key
 load_dotenv()
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-# Configure your application, for example, with an API key for a service
-# api_key = os.getenv("API_KEY")
-
-def get_gemini_response(input, pdf_content,prompt):
-    model = genai.GenerativeModel('gemini-pro')
-    response = model.generate_content([input,pdf_content[0],prompt])
-    return response.text
-
-def extract_text_from_first_page(pdf_path):
-    """Extract text from the first page of a PDF file using PyMuPDF (fitz)."""
-    doc = fitz.open(pdf_path)
-    first_page_text = doc[0].get_text()
+def extract_text_from_pdf(upload_file):
+    """Extract text from a PDF file using PyMuPDF (fitz)."""
+    doc = fitz.open(stream=upload_file.read(), filetype="pdf")
+    text = ""
+    for page in doc:
+        text += page.get_text("text")
     doc.close()
-    return first_page_text
+    return text
 
-def input_pdf_setup(upload_file):
-    """Save uploaded file to disk and extract text from the first page."""
-    if upload_file is not None:
-        # Save the uploaded PDF to a temporary file
-        with open("temp_uploaded.pdf", "wb") as f:
-            f.write(upload_file.getbuffer())
-        
-        # Extract text from the first page
-        extracted_text = extract_text_from_first_page("temp_uploaded.pdf")
-        return extracted_text
-    else:
-        raise FileNotFoundError("No File uploaded")
+def get_openai_response(job_description, resume_text, prompt_type):
+    """Send a prompt to the OpenAI API including both job description and resume text, and return the response."""
+    if prompt_type == "evaluation":
+        prompt = f"""
+        As an experienced Technical Human Resource Manager, evaluate the following resume against the job description provided:
+        Job Description: {job_description}
+        Resume Text: {resume_text}
+        Based on the job description and resume, provide a professional evaluation on whether the candidate's profile aligns with the role, highlighting the strengths and weaknesses.
+        """
+    elif prompt_type == "match_percentage":
+        prompt = f"""
+        As a skilled ATS scanner with a deep understanding of data science and ATS functionality, evaluate how well the following resume matches the job description provided:
+        Job Description: {job_description}
+        Resume Text: {resume_text}
+        Estimate the percentage match and list any missing keywords or qualifications.
+        """
+
+    response = openai.Completion.create(
+      engine="gpt-3.5-turbo-instruct",  # Adjust as necessary
+      prompt=prompt,
+      temperature=0.7,
+      max_tokens=1024,  # Adjusted to allow for more detailed responses
+      top_p=1.0,
+      frequency_penalty=0.0,
+      presence_penalty=0.0
+    )
+    return response.choices[0].text.strip()
 
 # Streamlit UI setup
 st.set_page_config(page_title="Resume Parser")
@@ -49,36 +56,20 @@ uploaded_file = st.file_uploader("Upload your resume (only PDFs are supported)..
 if uploaded_file is not None:
     st.write("Resume uploaded successfully.")
  
-submit1=st.button("Tell me about the resume")
-
-submit2=st.button("Percentage match with the job description")
+submit1 = st.button("Tell me about the resume")
+submit2 = st.button("Percentage match with the job description")
 
-input_prompt1 = """
- You are an experienced Technical Human Resource Manager,your task is to review the provided resume against the job description. 
-  Please share your professional evaluation on whether the candidate's profile aligns with the role. 
- Highlight the strengths and weaknesses of the applicant in relation to the specified job requirements.
-"""
-
-input_prompt2 = """
-You are an skilled ATS (Applicant Tracking System) scanner with a deep understanding of data science and ATS functionality, 
-your task is to evaluate the resume against the provided job description. give me the percentage of match if the resume matches
-the job description. First the output should come as a percentage and then keywords missing and last final thoughts.
-"""
-
-if submit1:
-    if uploaded_file is not None:
-        pdf_content=input_pdf_setup(uploaded_file)
-        response=get_gemini_response(input_prompt1,pdf_content,input_text)
-        st.subheader("The Response is")
+if uploaded_file is not None:
+    resume_text = extract_text_from_pdf(uploaded_file)
+    
+    if submit1:
+        response = get_openai_response(input_text, resume_text, "evaluation")
+        st.subheader("The Response is:")
         st.write(response)
-    else:
-        st.write("Please upload the resume")
-
-elif submit2:
-    if uploaded_file is not None:
-        pdf_content=input_pdf_setup(uploaded_file)
-        response=get_gemini_response(input_prompt2,pdf_content,input_text)
-        st.subheader("The Response is")
+    elif submit2:
+        response = get_openai_response(input_text, resume_text, "match_percentage")
+        st.subheader("The Response is:")
         st.write(response)
-    else:
-        st.write("Please upload the resume")
+else:
+    if submit1 or submit2:
+        st.error("Please upload the resume.")