Update app.py

app.py

@@ -3,45 +3,82 @@ import openai
 import os
 import fitz  # PyMuPDF
 
-# Set OpenAI API key
-api_key = "sk-1E6ExsyFb-cdU8jPNDP1dsEq_ra_bazU-EXQZQ86pJT3BlbkFJ4zURsV0t--3qNM7A-P57NUqZIBosrL7POwzpjR5EQA"
+# Set OpenAI API key from environment variable
+api_key = os.getenv("OPENAI_API_KEY")
+
+if not api_key:
+    raise ValueError("OpenAI API key not found. Please set it in the environment variables.")
 
 openai.api_key = api_key
 
 def extract_text_from_pdf(pdf_file):
-    # Open the PDF file
-    document = fitz.open(pdf_file)
-    text = ""
-    # Extract text from each page
-    for page_num in range(len(document)):
-        page = document.load_page(page_num)
-        text += page.get_text()
-    return text
+    try:
+        # Open the PDF file
+        document = fitz.open(pdf_file.name)
+        text = ""
+        # Extract text from each page
+        for page_num in range(len(document)):
+            page = document.load_page(page_num)
+            text += page.get_text()
+        return text
+    except Exception as e:
+        return f"Error extracting text from PDF: {e}"
+
+def extract_keywords(job_description):
+    prompt = f"""به عنوان یک تحلیلگر حرفه‌ای، لطفا مهم‌ترین کلمات کلیدی را از شرح شغل زیر استخراج کنید:
+    شرح شغل: {job_description}
+    پاسخ را به صورت یک لیست از کلمات کلیدی ارائه دهید.
+    """
+    
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0
+        )
+        keywords = response.choices[0].message['content']
+        return keywords
+    except Exception as e:
+        return f"Error during keyword extraction: {e}"
 
 def evaluate_resume(pdf_file, job_description):
+    # Extract keywords from job description
+    keywords = extract_keywords(job_description)
+    
+    if "Error" in keywords:
+        return keywords
+    
     # Extract text from PDF
     resume_text = extract_text_from_pdf(pdf_file)
-
-    prompt = f"""به عنوان یک تحلیلگر با تجربه سیستم ردیابی متقاضی (ATS)، نقش شما شامل ارزیابی رزومه در برابر شرح شغل است.
-   هدف شما تجزیه و تحلیل رزومه بر اساس شرح شغل داده شده است،
-    یک درصد تطابق را بر اساس معیارهای کلیدی در رزومه فرد  و شرح شغل ذکر شده را به دقت تعیین کنید و تعداد کلمات کلیدی گمشده را  مشخص کنید.
-    رزومه:{resume_text}
-    شرح شغل:{job_description}
-    من پاسخ را در یک رشته با ساختار زیر  می خواهم
     
+    if "Error" in resume_text:
+        return resume_text
+    
+    prompt = f"""به عنوان یک تحلیلگر با تجربه سیستم ردیابی متقاضی (ATS)، نقش شما شامل ارزیابی رزومه در برابر شرح شغل و کلمات کلیدی استخراج شده است.
+    لطفاً رزومه فرد را با کلمات کلیدی زیر مطابقت دهید و درصد تطابق را بر اساس معیارهای کلیدی و همچنین تعداد کلمات کلیدی گمشده و منطبق را محاسبه کنید.
+    
+    کلمات کلیدی: {keywords}
+    رزومه: {resume_text}
+    
+    من پاسخ را در یک رشته با ساختار زیر می‌خواهم:
     {{"تطابق شرح شغل با رزومه فرد ":"%"، "تعداد کلمات کلیدی غیر منطبق  ":""، "تعداد کلمات کلیدی منطبق   ":" }}
     """
     
-    response = openai.ChatCompletion.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": prompt}
-        ],
-        temperature=0
-    )
-    
-    return response.choices[0].message['content']
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt}
+            ],
+            temperature=0
+        )
+        return response.choices[0].message['content']
+    except Exception as e:
+        return f"Error during resume evaluation: {e}"
 
 iface = gr.Interface(
     fn=evaluate_resume,
@@ -53,4 +90,4 @@ iface = gr.Interface(
     title="Resume Evaluator"
 )
 
-iface.launch()
+iface.launch()