app.py

import gradio as gr
import spaces
import subprocess
import requests
import datetime
import os

@spaces.GPU
def run_peft_eval():
    print("Starting evaluation via run_eval.py", flush=True)
    try:
        result = subprocess.run(["python3", "run_eval.py"], capture_output=True, text=True, check=True)
        print(result.stdout, flush=True)
    except subprocess.CalledProcessError as e:
        print("Evaluation script failed:", flush=True)
        print(e.stderr, flush=True)
        return "Evaluation failed. Check logs."

    try:
        shutdown_url = os.environ.get("HF_ENDPOINT_SHUTDOWN")
        if shutdown_url:
            print("Evaluation done. Shutting down...", flush=True)
            requests.post(shutdown_url)
    except Exception as e:
        print("Failed to shutdown:", e, flush=True)

    return f"PEFT-Bench completed at {datetime.datetime.utcnow():%Y-%m-%d %H:%M UTC}"

with gr.Blocks() as demo:
    gr.Markdown("## PEFT-Bench Evaluation")
    gr.Markdown("This Space runs evaluation and shuts down when finished.")
    status = gr.Textbox(label="Status", lines=2)
    btn = gr.Button("Start Evaluation Now")
    btn.click(run_peft_eval, outputs=status)

demo.launch()