import gradio as gr
import pickle

# Load the model and vectorizer
with open("model.pkl", "rb") as model_file:
    model = pickle.load(model_file)

with open("vectorizer.pkl", "rb") as vectorizer_file:
    vectorizer = pickle.load(vectorizer_file)

# Define the prediction function
def predict_sentiment(text):
    text_vectorized = vectorizer.transform([text])
    prediction = model.predict(text_vectorized)
    sentiment = "Positive" if prediction == 1 else "Negative"
    return sentiment


with gr.Blocks() as app:
    gr.Markdown("<h1 style='text-align:center; color:black;'>Sentiment Analysis 💬</h1>")
    gr.Markdown("<p style='text-align:center; color:black;'>Enter a sentence to check if it's positive or negative.</p>")


    with gr.Row():
        input_text = gr.Textbox(label="Your Text", placeholder="Type something here...", lines=2)
        predict_button = gr.Button("Analyze Sentiment 🚀")

    output_text = gr.Textbox(label="Prediction", interactive=False)

    predict_button.click(fn=predict_sentiment, inputs=input_text, outputs=output_text)

    app.launch(mcp_server=True,ssr_mode=False,share=True)