Update app.py

app.py

@@ -13,14 +13,17 @@ from predict import predict_transaction
 
 # Gradio UI
 def predict_ui(check_id, employee_id, total, discount_amount, item_count, time, terminal_id):
+    if not employee_id or not time or not terminal_id:
+        return "Please provide all required fields."
+
     return predict_transaction({
         "check_id": check_id,
-        "employee_id": employee_id,
+        "employee_id": employee_id.strip(),
         "total": total,
         "discount_amount": discount_amount,
         "item_count": item_count,
-        "time": time,
-        "terminal_id": terminal_id
+        "time": time.strip(),
+        "terminal_id": terminal_id.strip()
     })
 
 demo = gr.Interface(