Update app.py

app.py

@@ -2,18 +2,24 @@ import gradio as gr
 import random
 
 # Function to simulate the prediction
-def predict_equipment_usage(hours_used):
-    # Example logic: Predict 'Repair' if usage is greater than 8 hours
-    prediction = "Repair" if hours_used > 8 else "Normal"
-    confidence = random.uniform(0.7, 1.0)  # Random confidence
+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
+# Create the Gradio interface with inputs: hours used and idle hours
 iface = gr.Interface(
-    fn=predict_equipment_usage, 
-    inputs="number",  # Input is a number (hours used)
-    outputs=["text", "number"],  # Output is prediction (text) and confidence (number)
-    live=True,  # Makes it responsive
+    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"
 )