import gradio as gr
from joblib import load

# Load the vectorizer and model
vectorizer = load('tfidf_vectorizer.pkl')
model = load('fake_review_detector.pkl')

# Prediction function
def predict_review(review):
    # Transform the input text into TF-IDF
    review_tfidf = vectorizer.transform([review])
    # Make a prediction
    prediction = model.predict(review_tfidf)[0]
    # Return the result
    return "THIS IS NOT FAKE" if prediction == 0 else "THIS IS FAKE"

# Gradio interface
interface = gr.Interface(
    fn=predict_review,
    inputs="text",
    outputs="text",
    title="Fake Review Detector",
    description="Type a review below to determine if it is fake or not."
)

# Launch the interface
interface.launch()