Update app.py

app.py

@@ -1,6 +1,15 @@
+import os
+import subprocess
 import gradio as gr
 from predict import predict_transaction
 
+# Ensure the model directory exists and model is trained
+if not (os.path.exists("model/model.pkl") and os.path.exists("model/encoders.pkl")):
+    print("Model files not found. Training model...")
+    os.makedirs("model", exist_ok=True)
+    subprocess.run(["python", "train.py"], check=True)
+
+# Gradio UI wrapper
 def predict_ui(check_id, employee_id, total, discount_amount, item_count, time, terminal_id):
     return predict_transaction({
         "check_id": check_id,
@@ -14,8 +23,16 @@ def predict_ui(check_id, employee_id, total, discount_amount, item_count, time,
 
 demo = gr.Interface(
     fn=predict_ui,
-    inputs=["number", "text", "number", "number", "number", "text", "text"],
-    outputs="text",
+    inputs=[
+        gr.Number(label="Check ID"),
+        gr.Text(label="Employee ID"),
+        gr.Number(label="Total"),
+        gr.Number(label="Discount Amount"),
+        gr.Number(label="Item Count"),
+        gr.Text(label="Time (HH:MM)"),
+        gr.Text(label="Terminal ID"),
+    ],
+    outputs=gr.Text(label="Suspicious (1=True, 0=False)"),
     title="Suspicious Transaction Detector"
 )