# Import dependencies
import gradio as gr
import numpy as np
import joblib 

# Predict function
def predict(region1, region2, region3, region4, region5, region6): 
    model = joblib.load('./localoutlierfactor.joblib')
    yhat = model.predict(np.array([[region1, region2, region3, region4, region5, region6]]))
    return 'Weird Traffic' if yhat == -1 else 'Normal Traffic'

# Building the interface
app = gr.Interface(
    title='Traffic Anomaly Detection 🚦',
    fn=predict, 
    inputs=[gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800),gr.Slider(0,800)], 
    outputs='text'
)

# Launch the app 
app.launch()