import gradio as gr
from datetime import datetime
from apscheduler.schedulers.background import BackgroundScheduler
from huggingface_hub import restart_space
import os

HF_TOKEN = os.getenv("HF_TOKEN")

time_start = datetime.now()

def restart():
    gr.Info("Restarting space...")
    restart_space("davanstrien/restart", token=HF_TOKEN)


with gr.Blocks() as demo:
    gr.Markdown(f"Time started: {time_start}")

scheduler = BackgroundScheduler()
job = scheduler.add_job(restart, "interval", minutes=2)
scheduler.start()

demo.launch()