sejalkishan commited on
Commit
409245b
·
verified ·
1 Parent(s): e557e58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -1,8 +1,9 @@
1
  import gradio as gr
2
  import fitz # PyMuPDF
 
3
  import docx
4
  import re
5
- from fastapi import FastAPI
6
 
7
  def extract_text_from_pdf(file):
8
  doc = fitz.open(stream=file.read(), filetype="pdf")
@@ -26,7 +27,8 @@ def extract_info(text):
26
  data["name"] = name_match.group(1).strip() if name_match else ""
27
  data["email"] = email_match.group(0) if email_match else ""
28
  data["phone"] = phone_match.group(0) if phone_match else ""
29
-
 
30
  skill_keywords = ["Python", "Java", "C++", "NLP", "Machine Learning", "Data Science", "SQL", "React"]
31
  found_skills = [skill for skill in skill_keywords if skill.lower() in text.lower()]
32
  data["skills"] = found_skills
@@ -56,4 +58,9 @@ with gr.Blocks(css="body { max-width: 600px; margin: auto; }") as demo:
56
 
57
  submit_btn.click(fn=process_resume, inputs=file_input, outputs=output)
58
 
59
- app = gr.mount_gradio_app(app=FastAPI(), blocks=demo, path="/")
 
 
 
 
 
 
1
  import gradio as gr
2
  import fitz # PyMuPDF
3
+ from fastapi import FastAPI
4
  import docx
5
  import re
6
+ import os
7
 
8
  def extract_text_from_pdf(file):
9
  doc = fitz.open(stream=file.read(), filetype="pdf")
 
27
  data["name"] = name_match.group(1).strip() if name_match else ""
28
  data["email"] = email_match.group(0) if email_match else ""
29
  data["phone"] = phone_match.group(0) if phone_match else ""
30
+
31
+ # Rough extraction for keywords
32
  skill_keywords = ["Python", "Java", "C++", "NLP", "Machine Learning", "Data Science", "SQL", "React"]
33
  found_skills = [skill for skill in skill_keywords if skill.lower() in text.lower()]
34
  data["skills"] = found_skills
 
58
 
59
  submit_btn.click(fn=process_resume, inputs=file_input, outputs=output)
60
 
61
+ app = gr.mount_gradio_app(app=FastAPI(), blocks=iface, path="/")
62
+
63
+ # For local testing
64
+ if __name__ == "__main__":
65
+ import uvicorn
66
+ uvicorn.run("app:app", host="0.0.0.0", port=7860)