Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# app.py
|
2 |
+
import gradio as gr
|
3 |
+
from text_extractor import extract_text_from_file
|
4 |
+
from embedder import get_embeddings
|
5 |
+
from vector_store import create_faiss_index, search_similar_cvs
|
6 |
+
from groq_api import get_summary
|
7 |
+
|
8 |
+
# Global storage
|
9 |
+
cv_texts = []
|
10 |
+
cv_names = []
|
11 |
+
cv_vectors = []
|
12 |
+
faiss_index = None
|
13 |
+
|
14 |
+
|
15 |
+
def upload_cvs(files):
|
16 |
+
global cv_texts, cv_names, cv_vectors, faiss_index
|
17 |
+
|
18 |
+
cv_texts = [extract_text_from_file(f.name) for f in files]
|
19 |
+
cv_names = [f.name for f in files]
|
20 |
+
cv_vectors = get_embeddings(cv_texts)
|
21 |
+
faiss_index = create_faiss_index(cv_vectors)
|
22 |
+
|
23 |
+
return f"Uploaded and indexed {len(files)} CVs."
|
24 |
+
|
25 |
+
|
26 |
+
def match_jd(jd_text):
|
27 |
+
if not faiss_index:
|
28 |
+
return "Please upload CVs first."
|
29 |
+
|
30 |
+
jd_vector = get_embeddings([jd_text])[0]
|
31 |
+
top_k_indices = search_similar_cvs(jd_vector, faiss_index, k=3)
|
32 |
+
|
33 |
+
matched_names = [cv_names[i] for i in top_k_indices]
|
34 |
+
matched_texts = [cv_texts[i] for i in top_k_indices]
|
35 |
+
summary = get_summary(jd_text, matched_texts)
|
36 |
+
|
37 |
+
return f"Top Matches: {matched_names}\n\nSummary: {summary}"
|
38 |
+
|
39 |
+
|
40 |
+
def clear_data():
|
41 |
+
global cv_texts, cv_names, cv_vectors, faiss_index
|
42 |
+
cv_texts, cv_names, cv_vectors, faiss_index = [], [], [], None
|
43 |
+
return "Data cleared."
|
44 |
+
|
45 |
+
|
46 |
+
iface = gr.Interface(
|
47 |
+
fn=match_jd,
|
48 |
+
inputs=[
|
49 |
+
gr.Textbox(lines=10, label="Paste Job Description"),
|
50 |
+
],
|
51 |
+
outputs="text",
|
52 |
+
title="CV Matcher with Groq",
|
53 |
+
description="Upload CVs, enter a Job Description, and get top matches and summary."
|
54 |
+
)
|
55 |
+
|
56 |
+
upload = gr.Interface(
|
57 |
+
fn=upload_cvs,
|
58 |
+
inputs=gr.File(file_types=[".pdf", ".docx"], file_count="multiple"),
|
59 |
+
outputs="text",
|
60 |
+
title="Upload CVs"
|
61 |
+
)
|
62 |
+
|
63 |
+
clear = gr.Interface(fn=clear_data, inputs=[], outputs="text", title="Reset Data")
|
64 |
+
|
65 |
+
app = gr.TabbedInterface([upload, iface, clear], ["Upload CVs", "Match JD", "Clear Data"])
|
66 |
+
|
67 |
+
if __name__ == "__main__":
|
68 |
+
app.launch()
|