import streamlit as st
import joblib

# Load model
@st.cache_resource
def load_model():
    vectorizer = joblib.load('vectorizer.pkl')
    model = joblib.load('model.pkl')
    return vectorizer, model

vectorizer, model = load_model()

# UI
st.title("Skill Extraction from Job Descriptions")
job_text = st.text_area("Enter job description:", height=200)

if st.button("Extract Skills"):
    if job_text:
        # Vectorize and predict
        text_vec = vectorizer.transform([job_text])
        prediction = model.predict(text_vec)
        
        st.success("Extracted Skills:")
        st.write(prediction[0])