Update app.py

app.py

@@ -97,10 +97,12 @@ def crawl_website(url):
     xml_str = "\n".join([line for line in xml_str.split("\n") if line.strip()])
 
     # Write the XML string to a file
-    with open("sitemap.xml", "w") as file:
+    sitemap_file = "sitemap.xml"
+    with open(sitemap_file, "w") as file:
         file.write(xml_str)
 
-    return "sitemap.xml"
+    return sitemap_file
+
 
 def extract_text_from_sitemap(sitemap_file):
     with open(sitemap_file, 'r') as file:
@@ -125,18 +127,24 @@ def extract_text_from_sitemap(sitemap_file):
 
     return extracted_text
 
+
 def gradio_interface(url):
     sitemap_file = crawl_website(url)
     extracted_text = extract_text_from_sitemap(sitemap_file)
-    text_file_path = 'extracted_text.txt'
 
+    # Save the extracted text to a file
+    text_file_path = 'extracted_text.txt'
     with open(text_file_path, 'w', encoding='utf-8') as file:
         file.write(extracted_text)
 
     return "\n".join(crawl_website.progress_textbox), text_file_path
 
 
-with gr.Interface(fn=gradio_interface, inputs="text", outputs=["text", "file"],
-                  title="Website Crawler",
-                  description="Enter a website URL to crawl and extract text from web pages.") as iface:
-    iface.launch()
+iface = gr.Interface(
+    fn=gradio_interface,
+    inputs="text",
+    outputs=["text", "file"],
+    title="Website Crawler",
+    description="Enter a website URL to crawl and extract text from web pages."
+)
+iface.launch()