Update app.py

app.py

@@ -1,15 +1,21 @@
 import gradio as gr
 import re
 from docx import Document
-from PyPDF2 import PdfReader  # Use PdfReader from PyPDF2
+from PyPDF2 import PdfFileReader
 
 # Function to extract text from a PDF file
 def extract_text_from_pdf(pdf_file):
     text = ""
-    pdf = PdfReader(pdf_file)
-    for page in pdf.pages:
-        text += page.extract_text()
-    return text
+    try:
+        pdf = PdfFileReader(pdf_file)
+        if pdf.numPages == 0:
+            raise ValueError("The PDF file is empty.")
+        for page_num in range(pdf.getNumPages()):
+            page = pdf.getPage(page_num)
+            text += page.extractText()
+        return text
+    except Exception as e:
+        raise ValueError("Error reading the PDF file: " + str(e)
 
 # Function to extract text from a DOCX file
 def extract_text_from_docx(docx_file):
@@ -35,23 +41,10 @@ def extract_info_from_resume(resume_path):
     email = re.search(email_pattern, text)
     phone = re.search(phone_pattern, text)
 
-    if name:
-        name = name.group()
-    else:
-        name = "Name not found"
-    if email:
-        email = email.group()
-    else:
-        email = "Email not found"
-    if phone:
-        phone = phone.group()
-    else:
-        phone = "Phone number not found"
-
     extracted_info = {
-        "Name": name,
-        "Email": email,
-        "Phone": phone,
+        "Name": name.group() if name else "Name not found",
+        "Email": email.group() if email else "Email not found",
+        "Phone": phone.group() if phone else "Phone number not found",
     }
 
     return extracted_info