import gradio as gr
from weasyprint import HTML
from datetime import datetime
from utils import gen_html
import os

def generate_pdf(num_questions):
    now = datetime.now()
    filename = now.strftime("%Y_%m_%d_AMC.pdf")

    allhtml = gen_html(num_questions)

    HTML(string=allhtml).write_pdf(filename)
    
    return filename

def lucky_button(num_questions):
    pdf_file = generate_pdf(num_questions)
    return pdf_file

interface = gr.Interface(
    fn=lucky_button, 
    inputs=gr.Slider(minimum=1, maximum=20, step=1, value=5, label="Number of Questions"), 
    outputs=gr.File(label="Generated PDF"), 
    title="Random AMC/AIME Problem PDF Generator",
    description="Click the button to generate a random PDF of AMC/AIME problems. You can specify the number of questions to include in the PDF."
)

interface.launch()