Create app.py

app.py

@@ -0,0 +1,29 @@
+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.inputs.Slider(minimum=1, maximum=20, step=1, default=5, label="Number of Questions"), 
+    outputs="file", 
+    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()