import gradio as gr
import pickle
import numpy as np

# Load model and scaler
model = pickle.load(open('model.pkl', 'rb'))
scaler = pickle.load(open('scaler.pkl', 'rb'))

def predict_price(area, bedrooms, bathrooms, stories, parking, mainroad, guestroom, 
                  basement, hotwaterheating, airconditioning, prefarea, furnishingstatus):
    
    # Encode inputs
    mainroad = 1.0 if mainroad == "Yes" else 0.0
    guestroom = 1.0 if guestroom == "Yes" else 0.0
    basement = 1.0 if basement == "Yes" else 0.0
    hotwaterheating = 1.0 if hotwaterheating == "Yes" else 0.0
    airconditioning = 1.0 if airconditioning == "Yes" else 0.0
    prefarea = 1.0 if prefarea == "Yes" else 0.0
    
    furnishing_map = {"furnished": 0.0, "semi-furnished": 1.0, "unfurnished": 2.0}
    furnishingstatus = furnishing_map[furnishingstatus]
    
    # Create input array
    input_data = np.array([[area, bedrooms, bathrooms, stories, 
                           mainroad, guestroom, basement,
                           hotwaterheating, airconditioning,
                           parking, prefarea, furnishingstatus]])
    
    # Scale and predict
    input_scaled = scaler.transform(input_data)
    prediction = model.predict(input_scaled)
    
    return f" {prediction[0]:,.2f}"

# Create Gradio interface
demo = gr.Interface(
    fn=predict_price,
    inputs=[
        gr.Number(label="Area (sq ft)", value=5000),
        gr.Dropdown(choices=[1, 2, 3, 4, 5, 6], label="Bedrooms", value=3),
        gr.Dropdown(choices=[1, 2, 3, 4], label="Bathrooms", value=2),
        gr.Dropdown(choices=[1, 2, 3, 4], label="Stories", value=2),
        gr.Dropdown(choices=[0, 1, 2, 3], label="Parking", value=1),
        gr.Dropdown(choices=["Yes", "No"], label="Main Road", value="Yes"),
        gr.Dropdown(choices=["Yes", "No"], label="Guest Room", value="No"),
        gr.Dropdown(choices=["Yes", "No"], label="Basement", value="No"),
        gr.Dropdown(choices=["Yes", "No"], label="Hot Water Heating", value="No"),
        gr.Dropdown(choices=["Yes", "No"], label="Air Conditioning", value="Yes"),
        gr.Dropdown(choices=["Yes", "No"], label="Preferred Area", value="Yes"),
        gr.Dropdown(choices=["furnished", "semi-furnished", "unfurnished"], label="Furnishing", value="furnished")
    ],
    outputs=gr.Textbox(label="Predicted Price"),
    title="🏠 House Price Prediction",
    description="Enter house features to predict the price"
)

demo.launch(share=True)