import streamlit as st
import joblib

st.title('Fake or Real News Prediction')

# Load the model
try:
    model = joblib.load('src/passive_aggressive_classifier.pkl')
except Exception as e:
    st.error(f"Error loading model: {e}")

# Load the vectorizer
try:
    vectorizer = joblib.load('src/tfidf_vectorizer.joblib')
except Exception as e:
    st.error(f"Error loading vectorizer: {e}")

# Input field for news text
news_text = st.text_area('Enter news article text')

if st.button('Predict'):
    if news_text.strip() == "":
        st.warning("Please enter some text.")
    else:
        # Transform input text
        input_vector = vectorizer.transform([news_text])
        
        # Predict
        prediction = model.predict(input_vector)[0]
        st.success(f'Prediction: {prediction}')