import gradio as gr
import subprocess

def run_training():
    try:
        print("✅ Training started.")
        result = subprocess.run(["python", "train_abuse_model.py"], capture_output=True, text=True)
        print("✅ Training finished.")
        print("STDOUT:", result.stdout)
        print("STDERR:", result.stderr)
        return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
    except Exception as e:
        print("❌ Exception occurred:", e)
        return f"Exception occurred:\n{str(e)}"



# Define a simple Gradio interface with one button
demo = gr.Interface(
    fn=run_training,
    inputs=[],
    outputs="text",
    title="Run Model Training",
    description="Click the button to execute train.py. This will use GPU if available."
)

demo.launch()