Jekyll2000 commited on
Commit
04727ee
·
verified ·
1 Parent(s): 6b51515

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -18,10 +18,10 @@ def setup_chromadb():
18
 
19
  def extract_text_from_resume(file):
20
  if file.name.endswith(".pdf"):
21
- doc = fitz.open(stream=file.read(), filetype="pdf")
22
- return "\n".join([page.get_text("text") for page in doc])
23
  elif file.name.endswith(".txt"):
24
- return file.read().decode("utf-8")
25
  return ""
26
 
27
  def extract_candidate_name(resume_text):
@@ -128,7 +128,7 @@ class InterviewCoach:
128
  self.candidate_name = store_resume(resume_text, self.user_id)
129
 
130
  resume_data = retrieve_resume(self.user_id, "background experience")
131
- greeting = zero_agent_greeting(resume_data, self.candidate_name)
132
  self.questions.append(greeting)
133
 
134
  return f"Interview started with {self.candidate_name}", greeting, gr.update(visible=True)
 
18
 
19
  def extract_text_from_resume(file):
20
  if file.name.endswith(".pdf"):
21
+ with fitz.open(stream=file.content, filetype="pdf") as doc:
22
+ return "\n".join([page.get_text("text") for page in doc])
23
  elif file.name.endswith(".txt"):
24
+ return file.content.decode("utf-8")
25
  return ""
26
 
27
  def extract_candidate_name(resume_text):
 
128
  self.candidate_name = store_resume(resume_text, self.user_id)
129
 
130
  resume_data = retrieve_resume(self.user_id, "background experience")
131
+ greeting = self.zero_agent_greeting(resume_data, self.candidate_name)
132
  self.questions.append(greeting)
133
 
134
  return f"Interview started with {self.candidate_name}", greeting, gr.update(visible=True)