import gradio as gr
import random

# Function to simulate the prediction
def predict_equipment_usage(hours_used, idle_hours):
    # Example logic based on hours of usage and idle time
    if hours_used > 8 and idle_hours > 2:
        prediction = "Repair"
    elif hours_used > 6 and idle_hours < 2:
        prediction = "Move"
    else:
        prediction = "Normal"
    
    confidence = random.uniform(0.7, 1.0)  # Random confidence value for now
    return prediction, confidence

# Create the Gradio interface with inputs: hours used and idle hours
iface = gr.Interface(
    fn=predict_equipment_usage,
    inputs=["number", "number"],  # Inputs: hours used and idle hours
    outputs=["text", "number"],  # Outputs: prediction and confidence
    live=True,
    title="Equipment Usage Prediction"
)

iface.launch()