Update app.py

app.py

@@ -1,13 +1,26 @@
 import gradio as gr
+import subprocess
 
-def status():
-    return "DuckDB Fraud Detection Pipeline is ready. Run src/train.py to train the model."
+def run_pipeline():
+    try:
+        result = subprocess.run(
+            ["python", "src/train.py"],
+            capture_output=True,
+            text=True
+        )
+        if result.returncode == 0:
+            return "Pipeline executed successfully:\n\n" + result.stdout
+        else:
+            return "Pipeline failed:\n\n" + result.stderr
+    except Exception as e:
+        return str(e)
 
 demo = gr.Interface(
-    fn=status,
+    fn=run_pipeline,
     inputs=[],
     outputs="text",
     title="Credit Card Fraud Detection with DuckDB",
+    description="Click Generate to run the DuckDB Medallion pipeline and train the fraud model."
 )
 
 demo.launch()