Update app.py

app.py

@@ -2,53 +2,50 @@ import gradio as gr
 import requests
 import xml.etree.ElementTree as ET
 
-
 def collect_urls_from_sitemap(sitemap_content):
     root = ET.fromstring(sitemap_content)
-
+    
     urls = []
     for url_element in root.findall(".//{http://www.sitemaps.org/schemas/sitemap/0.9}url"):
         loc_element = url_element.find("{http://www.sitemaps.org/schemas/sitemap/0.9}loc")
         if loc_element is not None:
             urls.append(loc_element.text)
-
+    
     return urls
 
-
 def compare_sitemaps(old_urls, new_urls):
     added_urls = list(set(new_urls) - set(old_urls))
     removed_urls = list(set(old_urls) - set(new_urls))
     return added_urls, removed_urls
 
-
 def compare_sitemaps_ui(old_sitemap_url, new_sitemap_url):
     old_response = requests.get(old_sitemap_url)
     new_response = requests.get(new_sitemap_url)
-
+    
     if old_response.status_code == 200 and new_response.status_code == 200:
         old_sitemap_content = old_response.content
         new_sitemap_content = new_response.content
-
+        
         old_urls = collect_urls_from_sitemap(old_sitemap_content)
         new_urls = collect_urls_from_sitemap(new_sitemap_content)
-
+        
         added_urls, removed_urls = compare_sitemaps(old_urls, new_urls)
-
+        
         added_urls_text = "\n".join(added_urls)
         removed_urls_text = "\n".join(removed_urls)
-
-        return f"Newly Added URLs:\n{added_urls_text}\n\nRemoved URLs:\n{removed_urls_text}"
+        
+        return {"Newly Added URLs": added_urls_text, "Removed URLs": removed_urls_text}
     else:
-        return "Failed to fetch sitemaps."
-
+        return {"Newly Added URLs": "Failed to fetch sitemaps.", "Removed URLs": "Failed to fetch sitemaps."}
 
 iface = gr.Interface(
     fn=compare_sitemaps_ui,
     inputs=["text", "text"],
-    outputs="text",
+    outputs=[gr.outputs.Textbox(label="Newly Added URLs"), gr.outputs.Textbox(label="Removed URLs")],
     title="Sitemap Comparator",
     description="Enter the URLs of the old and new sitemaps to compare.",
     examples=[["http://bonrix.net/old-sitemap.xml", "http://bonrix.net/sitemap.xml"]]
 )
 
 iface.launch()
+