Update app.py

app.py

@@ -1,25 +1,30 @@
 import gradio as gr
 import random
 from scraperr import scrape_google
-from pdf_converter import DocumentHandler
+from pdf_converter import create_pdf
 
 
 
-async def scrape_websites(topic, num_links, num_results_per_link=10):
-    output_text = await scrape_google(topic, num_results_per_link)
+async def scrape_websites(topic, num_links, num_results_per_link=10) -> str:
+
+  output_text = await scrape_google(topic, num_results_per_link)
+
     # Ensure output_text is a list
     if not isinstance(output_text, list):
         output_text = [output_text]
+
     # Select random links based on the user's input
     selected_links = random.sample(output_text, min(num_links, len(output_text)))
+
     # Convert the list of strings to a single string
     output_string = "\n".join(selected_links)
+
     return output_string
 
-def convert_to_pdf(url):
-    document_handler = DocumentHandler(url)
-    pdf_path = document_handler.create_pdf()
-    return pdf_path
+def convert_to_pdf(url) -> str:
+   #Call the create_pdf function to convert the webpage to PDF.
+   pdf_path = create_pdf(url)
+   return pdf_path
 
 html = """
 <div style="text-align:center; max-width: 900px; margin: 0 auto; margin-top:5px">