# -*- coding: utf-8 -*-
"""app.py

Automatically generated by Colab.

Original file is located at
    https://colab.research.google.com/drive/1EUSf7Q4En-DqvH6-G5zg4q7I7oGOYPO_
"""

import gradio as gr
import joblib
import numpy as np

# Load model and scaler
model = joblib.load("wine_quality_model.pkl")
scaler = joblib.load("wine_scaler.pkl")

def predict_quality(fixed_acidity, volatile_acidity, citric_acid, residual_sugar,
                    chlorides, free_sulfur_dioxide, total_sulfur_dioxide,
                    density, pH, sulphates, alcohol):

    sample = np.array([[fixed_acidity, volatile_acidity, citric_acid, residual_sugar,
                        chlorides, free_sulfur_dioxide, total_sulfur_dioxide,
                        density, pH, sulphates, alcohol]])

    scaled = scaler.transform(sample)
    pred = model.predict(scaled)[0]

    return "GOOD 🍷" if pred == 1 else "BAD ❌"

inputs = [
    gr.Number(label="Fixed Acidity"),
    gr.Number(label="Volatile Acidity"),
    gr.Number(label="Citric Acid"),
    gr.Number(label="Residual Sugar"),
    gr.Number(label="Chlorides"),
    gr.Number(label="Free Sulfur Dioxide"),
    gr.Number(label="Total Sulfur Dioxide"),
    gr.Number(label="Density"),
    gr.Number(label="pH"),
    gr.Number(label="Sulphates"),
    gr.Number(label="Alcohol"),
]

output = gr.Textbox(label="Prediction")

app = gr.Interface(
    fn=predict_quality,
    inputs=inputs,
    outputs=output,
    title="Wine Quality Prediction App",
    description="Enter wine chemistry values to predict if wine is GOOD or BAD"
)

app.launch()