Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,109 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+from resume_parsing import parse_resume_with_llm
+from database_integration import save_candidate, save_job, delete_candidate, delete_job, session, Candidate, Job
+from chroma_utils import add_to_job_chroma, get_all_jobs_from_chroma, delete_resume_from_chroma, delete_job_from_chroma
+from embedding_utils import generate_embedding
+from job_matching import calculate_ats_score
+import pandas as pd
+import numpy as np
+
+st.set_page_config(page_title="ATS System", layout="wide")
+
+st.title("📄 ATS System Dashboard")
+
+tabs = st.tabs(["Upload Resume", "Post Job", "Match Candidates", "Delete Resume/Job"])
+
+# ---------------- Upload Resume ---------------- #
+with tabs[0]:
+    st.header("Upload Candidate Resume")
+    name = st.text_input("Candidate Name")
+    location = st.text_input("Location")
+    uploaded_file = st.file_uploader("Upload PDF/DOCX Resume", type=["pdf", "docx"])
+
+    if st.button("Submit Resume"):
+        if not uploaded_file:
+            st.error("Please upload a resume file.")
+        elif not name or not location:
+            st.error("Please provide both name and location.")
+        else:
+            file_type = uploaded_file.name.split(".")[-1]
+            resume_content = uploaded_file.read()
+            result, embedding = parse_resume_with_llm(resume_content, name, location, file_type)
+            if "error" in result:
+                st.error(result["error"])
+            else:
+                save_candidate({
+                    "name": name,
+                    "location": location,
+                    "experience": result.get("experience"),
+                    "education": result.get("education"),
+                    "skills": result.get("skills")
+                }, result["unique_id"])
+                st.success(f"Resume uploaded successfully! Candidate ID: {result['unique_id']}")
+
+# ---------------- Post Job ---------------- #
+with tabs[1]:
+    st.header("Post a New Job")
+    job_title = st.text_input("Job Title", key="job_title")
+    job_description = st.text_area("Job Description", key="job_description")
+
+    if st.button("Post Job"):
+        if not job_title or not job_description:
+            st.error("Both job title and description are required.")
+        else:
+            job_embedding = generate_embedding(job_description)
+            metadata = {"title": job_title, "description": job_description}
+            unique_id = add_to_job_chroma(job_embedding, metadata)
+            save_job(job_title, job_description, unique_id)
+            st.success(f"Job posted successfully! Job ID: {unique_id}")
+
+# ---------------- Match Candidates ---------------- #
+with tabs[2]:
+    st.header("Match Candidates for All Jobs")
+    if st.button("Run Matching"):
+        job_ids, job_embeddings, job_metadatas = get_all_jobs_from_chroma()
+        if not job_ids:
+            st.warning("No jobs found in database.")
+        else:
+            all_results = []
+            for i, job_embedding in enumerate(job_embeddings):
+                job_embedding = np.array(job_embedding)
+                matched_candidates = calculate_ats_score(job_embedding)
+                for candidate in matched_candidates:
+                    all_results.append({
+                        "Job ID": job_ids[i],
+                        "Job Title": job_metadatas[i].get("title"),
+                        "Candidate ID": candidate["candidate_id"],
+                        "Candidate Name": candidate["metadata"]["name"],
+                        "Score": round(candidate["score"], 3)
+                    })
+            if all_results:
+                df = pd.DataFrame(all_results)
+                st.dataframe(df)
+            else:
+                st.info("No candidates matched any jobs yet.")
+
+# ---------------- Delete Resume/Job ---------------- #
+with tabs[3]:
+    st.header("Delete Resume or Job")
+    delete_option = st.radio("Select Type to Delete", ["Resume", "Job"])
+    unique_id = st.text_input("Enter Unique ID to Delete")
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+    if st.button("Delete"):
+        if not unique_id:
+            st.error("Please provide a unique ID.")
+        else:
+            if delete_option == "Resume":
+                delete_resume_from_chroma(unique_id)
+                success = delete_candidate(unique_id)
+                if success:
+                    st.success(f"Resume {unique_id} deleted successfully!")
+                else:
+                    st.warning("Resume not found.")
+            elif delete_option == "Job":
+                delete_job_from_chroma(unique_id)
+                success = delete_job(unique_id)
+                if success:
+                    st.success(f"Job {unique_id} deleted successfully!")
+                else:
+                    st.warning("Job not found.")