Create app.py

app.py

@@ -0,0 +1,74 @@
+import gradio as gr
+import torch
+from transformers import pipeline
+import pdfplumber
+import re
+
+# Load pre-trained model for Named Entity Recognition (NER) to extract details
+nlp = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", framework="pt")
+
+def extract_text_from_pdf(pdf_file):
+    """Extract text from the uploaded PDF resume."""
+    with pdfplumber.open(pdf_file) as pdf:
+        text = ""
+        for page in pdf.pages:
+            text += page.extract_text()
+    return text
+
+def parse_resume(resume_text):
+    """Parse the resume and extract details like name, email, phone, skills, etc."""
+    # Define regex for phone and email extraction
+    phone_pattern = r'\(?\d{3}\)?[-.\s]?\d{3}[-.\s]?\d{4}'
+    email_pattern = r'[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}'
+
+    # Extract phone and email using regex
+    phone = re.findall(phone_pattern, resume_text)
+    email = re.findall(email_pattern, resume_text)
+
+    # Extract named entities (e.g., skills, education, and experience)
+    entities = nlp(resume_text)
+    
+    # For simplicity, we just list out the entities here
+    skills = []
+    experience = []
+    education = []
+    certifications = []
+
+    # Iterate through recognized entities and categorize them
+    for entity in entities:
+        if 'ORG' in entity['entity']:
+            experience.append(entity['word'])
+        elif 'MISC' in entity['entity']:
+            skills.append(entity['word'])
+        elif 'LOC' in entity['entity']:
+            education.append(entity['word'])
+        else:
+            certifications.append(entity['word'])
+
+    # Create a JSON-like output
+    parsed_data = {
+        "Phone": phone[0] if phone else "Not found",
+        "Email": email[0] if email else "Not found",
+        "Skills": ", ".join(skills),
+        "Experience": ", ".join(experience),
+        "Education": ", ".join(education),
+        "Certifications": ", ".join(certifications)
+    }
+
+    return parsed_data
+
+# Define Gradio interface
+def resume_parser(pdf_file):
+    """Main function for resume parsing"""
+    resume_text = extract_text_from_pdf(pdf_file)
+    parsed_info = parse_resume(resume_text)
+    return parsed_info
+
+# Create the Gradio interface
+gr.Interface(
+    fn=resume_parser,
+    inputs=gr.inputs.File(label="Upload Resume (PDF)"),
+    outputs=gr.outputs.JSON(label="Parsed Information"),
+    title="AI Resume Parser",
+    description="Upload a resume (PDF) to extract details like Name, Email, Phone, Skills, Experience, Education, and Certifications."
+).launch()