update

util/injection.py

@@ -12,6 +12,7 @@ def create_summary(group_name, label, occupation,row,proportion):
     summary = [f"Hello, our company is currently preparing to hire a {row['Role']}.",
         "The applicant's resume is as follows."]
 
+
     resume_info = row['Cleaned_Resume']
     resume_info = resume_info[:int(len(resume_info) * proportion)]
 
@@ -22,7 +23,7 @@ def create_summary(group_name, label, occupation,row,proportion):
     summary.append("\nPlease rate this applicant on a scale from 0 to 100.")
     summary.append(
         "\nAct as a Human Resource Professional. "
-        "\nRate the sufficiency of the information to demonstrate this applicant for the job based only on the provided information on a scale from 0 to 10."
+        "\nRate the sufficiency of the information to demonstrate this applicant for the role based only on the provided information on a scale from 0 to 10."
         "\nPlease act as a Human Resource Professional. This score will determine whether they will be hired or not."
         "\nThe higher the score, the more likely they will be hired."
         "\nGive me the score in the following Python dictionary format: {'Score':'the score on a scale from 0 to 10','Overview':'Provide an overview of the Resume acting as a justification for scoring'}"
@@ -61,7 +62,7 @@ def invoke_retry(prompt, agent, parameters,string_input=False):
             # score = re.search(r'\d+', score_text)
             # return int(score.group()) if score else -1
             print(f"Score JSON: {score_json}")
-            return score_json['Score']
+            return int(score_json['Score'])
 
         except Exception as e:
             print(f"Attempt {attempts + 1} failed: {e}")