Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from bs4 import BeautifulSoup
 import requests
 from jinja2 import Template
-from weasyprint import HTML
 from urllib.parse import urljoin
 import os
 import warnings
@@ -132,12 +131,6 @@ def generate_html(chapter_num, conceptual_input, problem_input, path):
     with open(path, 'w', encoding='utf-8') as f:
         f.write(rendered_html)
 
-def generate_pdf(input_path, output_path):
-    try:
-        HTML(input_path).write_pdf(output_path)
-    except Exception as e:
-        logging.error(f"An error occurred when generating PDF: {e}")
-
 def main_function(unit_num, conceptual_input, problem_input):
     try:
         if not conceptual_input and not problem_input:
@@ -148,20 +141,18 @@ def main_function(unit_num, conceptual_input, problem_input):
         if not os.path.exists(abs_path):
             os.makedirs(abs_path)
 
-        # Delete all existing PDFs in the directory
+        # Delete all existing HTML files in the directory
         for file_name in os.listdir(abs_path):
-            if file_name.endswith(".pdf"):
+            if file_name.endswith(".html"):
                 os.remove(os.path.join(abs_path, file_name))
 
         file_name = "questions_" + str(int(unit_num))
         html_path = os.path.join(abs_path, file_name + ".html")
-        pdf_path = os.path.join(abs_path, file_name + ".pdf")
 
         generate_html(unit_num, conceptual_input, problem_input, html_path)
-        generate_pdf(html_path, pdf_path)
 
-        logging.info("PDF generated successfully")
-        return pdf_path
+        logging.info("HTML generated successfully")
+        return html_path
     except Exception as e:
         logging.error(f"An error occurred in main_function: {e}")
         return "An error occurred. Please check the logs for more details."
@@ -173,7 +164,7 @@ iface = gr.Interface(
         gr.Textbox(label="Conceptual Problems List (comma-separated)"),
         gr.Textbox(label="Problems & Exercises List (comma-separated)")
     ],
-    outputs=gr.File(label="Generated PDF"),
+    outputs=gr.File(label="Generated HTML"),
     live=False
 )