Update app.py

app.py

@@ -1,7 +1,3 @@
-!wget https://dl.google.com/linux/direct/google-chrome-stable_current_amd64.deb
-!sudo apt install ./google-chrome-stable_current_amd64.deb
-
-
 import gradio as gr
 from bs4 import BeautifulSoup
 import requests
@@ -12,6 +8,8 @@ from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
 from webdriver_manager.chrome import ChromeDriverManager
 
+from weasyprint import HTML
+
 import base64
 from urllib.parse import urljoin
 import time
@@ -162,34 +160,7 @@ def generate_html(chapter_num, conceptual_input, problem_input,path):
 
 
 def generate_pdf(input_path, output_path):
-    chrome_options = Options()
-    chrome_options.add_argument('--headless')
-    chrome_options.add_argument('--disable-gpu')
-    chrome_options.add_argument('--no-sandbox')
-    
-    driver = None
-    try:
-        driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()), options=chrome_options)
-        html_path = 'file:///' + input_path
-        driver.get(html_path)
-
-        # Add a delay to allow all elements to load (tune this time as needed)
-        time.sleep(2)
-        
-        pdf_data = driver.execute_cdp_cmd('Page.printToPDF', {'printBackground': True})
-        
-        if 'data' in pdf_data:
-            pdf_content = pdf_data['data']
-            pdf_bytes = base64.b64decode(pdf_content)
-            with open(output_path, 'wb') as f:
-                f.write(pdf_bytes)
-        else:
-            print("PDF could not be generated.")
-    except Exception as e:
-        print(f"An error occurred: {e}")
-    finally:
-        if driver:
-            driver.quit()
+    HTML(input_path).write_pdf(output_path)
 
 def main_function(unit_num, conceptual_input, problem_input):
     if not conceptual_input and not problem_input: