Update app.py

app.py

@@ -1,29 +1,27 @@
 import gradio as gr
-import pdfkit
 from datetime import datetime
-from utils import gen_html
+from utils import gen_html, save_html_to_file
 import os
 
-def generate_pdf(num_questions):
+def generate_html(num_questions):
     now = datetime.now()
-    filename = now.strftime("%Y_%m_%d_AMC.pdf")
-    
-    allhtml = gen_html(num_questions)
+    filename = now.strftime("%Y_%m_%d_AMC.html")
 
-    pdfkit.from_string(allhtml, filename)
+    allhtml = gen_html(num_questions)
+    save_html_to_file(allhtml, filename)
     
     return filename
 
 def lucky_button(num_questions):
-    pdf_file = generate_pdf(num_questions)
-    return pdf_file
+    html_file = generate_html(num_questions)
+    return html_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 HTML of AMC/AIME problems. You can specify the number of questions to include in the PDF."
+    outputs=gr.File(label="Generated HTML"), 
+    title="Random AMC/AIME Problem HTML Generator",
+    description="Click the button to generate a random HTML file of AMC/AIME problems. You can specify the number of questions to include in the HTML."
 )
 
-interface.launch()
+interface.launch()