import gradio as gr
import subprocess

def run_training():
    try:
        # Run train.py using subprocess and capture output
        result = subprocess.run(["python", "train.py"], capture_output=True, text=True)
        # Return stdout if success, otherwise stderr
        return result.stdout if result.returncode == 0 else f"Error:\n{result.stderr}"
    except Exception as 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()