#!/usr/bin/env python

import os

import gradio as gr

from papers import PaperList, get_df
from update_scheduler import UpdateScheduler

DESCRIPTION = "# [Daily Papers](https://huggingface.co/papers)"
DAILY_PAPERS = "# [pxiaoer's Daily Papers and AI News Tracking](https://aigc.openbot.ai/s/papers)"

paper_list = PaperList(get_df("papers.csv"))

if (SPACE_ID := os.getenv("SPACE_ID")) is not None:
    CRON_HOUR = os.getenv("CRON_HOUR", "*/4")
    CRON_MINUTE = os.getenv("CRON_MINUTE", "0")
    scheduler = UpdateScheduler(space_id=SPACE_ID, cron_hour=CRON_HOUR, cron_minute=CRON_MINUTE)
    scheduler.start()

with gr.Blocks(css="style.css") as demo:
    gr.Markdown(DESCRIPTION)
    gr.Markdown(DAILY_PAPERS)
    df = gr.Dataframe(
        value=paper_list.df_prettified,
        datatype=paper_list.column_datatype,
        type="pandas",
        interactive=False,
        height=1000,
        elem_id="table",
        column_widths=["10%", "10%", "60%", "10%", "10%"],
        wrap=True,
    )

if __name__ == "__main__":
    demo.queue(api_open=False).launch()